# coding:utf-8 from django.shortcuts import render,redirect from django.http.response import JsonResponse,Http404,HttpResponseNotAllowed,HttpResponse from django.http import QueryDict from django.http import HttpResponseForbidden from django.contrib.auth.decorators import login_required # 登录需求装饰器 from django.views.decorators.csrf import csrf_exempt from django.views.decorators.http import require_http_methods,require_GET,require_POST # 视图请求方法装饰器 from django.core.paginator import Paginator,PageNotAnInteger,EmptyPage,InvalidPage # 后端分页 from django.core.exceptions import PermissionDenied,ObjectDoesNotExist from django.core.serializers import serialize from app_doc.models import Project,Doc,DocTemp from django.contrib.auth.models import User from django.db.models import Q from django.db import transaction from loguru import logger import datetime import traceback import re import json import random from app_doc.report_utils import * from app_admin.models import UserOptions,SysSetting from app_admin.decorators import check_headers,allow_report_file import os.path import base64 import hashlib from django.utils.html import strip_tags import markdown # 替换前端传来的非法字符 def validateTitle(title): rstr = r"[\/\\\:\*\?\"\<\>\|\[\]]" # '/ \ : * ? " < > |' new_title = re.sub(rstr, "_", title) # 替换为下划线 return new_title # markdown文本生成摘要（不带markdown标记） def remove_markdown_tag(docs): for doc in docs: try: doc.pre_content = strip_tags(markdown.markdown(doc.pre_content)) except Exception as e: doc.pre_content = doc.pre_content # 获取文集的文档目录 def get_pro_toc(pro_id): # try: # project = Project.objects.get(id=pro_id) # pro_toc = ProjectToc.objects.get(project=project) # doc_list = json.loads(pro_toc.value) # print("使用缓存") # except: # print("重新生成") # 查询存在上级文档的文档 parent_id_list = Doc.objects.filter(top_doc=pro_id, status=1).exclude(parent_doc=0).values_list('parent_doc', flat=True) # 获取存在上级文档的上级文档ID # print(parent_id_list) doc_list = [] n = 0 # 获取一级文档 top_docs = Doc.objects.filter(top_doc=pro_id, parent_doc=0, status=1).values('id', 'name','open_children').order_by('sort') # 遍历一级文档 for doc in top_docs: top_item = { 'id': doc['id'], 'name': doc['name'], 'open_children':doc['open_children'] # 'spread': True, # 'level': 1 } # 如果一级文档存在下级文档，查询其二级文档 if doc['id'] in parent_id_list: # 获取二级文档 sec_docs = Doc.objects.filter( top_doc=pro_id, parent_doc=doc['id'], status=1).values('id', 'name','open_children').order_by('sort') top_item['children'] = [] for doc in sec_docs: sec_item = { 'id': doc['id'], 'name': doc['name'], 'open_children': doc['open_children'] # 'level': 2 } # 如果二级文档存在下级文档，查询第三级文档 if doc['id'] in parent_id_list: # 获取三级文档 thr_docs = Doc.objects.filter( top_doc=pro_id, parent_doc=doc['id'], status=1).values('id','name').order_by('sort') sec_item['children'] = [] for doc in thr_docs: item = { 'id': doc['id'], 'name': doc['name'], # 'level': 3 } sec_item['children'].append(item) n += 1 top_item['children'].append(sec_item) n += 1 else: top_item['children'].append(sec_item) n += 1 doc_list.append(top_item) n += 1 # 如果一级文档没有下级文档，直接保存 else: doc_list.append(top_item) n += 1 # 将文集的大纲目录写入数据库 # ProjectToc.objects.create( # project = project, # value = json.dumps(doc_list) # ) # print(doc_list,n) # if n > 999: # return ([],n) # else: return (doc_list,n) # 文集列表（首页） @logger.catch() def project_list(request): kw = request.GET.get('kw','') # 搜索词 sort = request.GET.get('sort','') # 排序,0表示按时间升序排序，1表示按时间降序排序，''表示按后台配置排序，默认为'' role = request.GET.get('role',-1) # 筛选文集权限，默认为显示所有可显示的文集 # 是否排序 if sort in [0,'0']: sort_str = '' elif sort == '': try: index_project_sort = SysSetting.objects.get(name='index_project_sort') if index_project_sort.value == '-1': sort_str = '-' else: sort_str = '' except: sort_str = '' else: sort_str = '-' # 是否搜索 if kw == '': is_kw = False else: is_kw = True # 是否认证 if request.user.is_authenticated: is_auth = True else: is_auth = False # 是否筛选 if role in ['',-1,'-1']: is_role = False role_list = [0,3] else: is_role = True # 没有搜索 and 认证用户 and 没有筛选 if (is_kw is False) and (is_auth) and (is_role is False): colla_list = [i.project.id for i in ProjectCollaborator.objects.filter(user=request.user)] # 用户的协作文集列表 project_list = Project.objects.filter( Q(role__in=role_list) | \ Q(role=2,role_value__contains=str(request.user.username)) | \ Q(create_user=request.user) | \ Q(id__in=colla_list) ).order_by("{}create_time".format(sort_str)) # 没有搜索 and 认证用户 and 有筛选 elif (is_kw is False ) and (is_auth) and (is_role): if role in ['0',0]: project_list = Project.objects.filter(role=0).order_by("{}create_time".format(sort_str)) elif role in ['1',1]: project_list = Project.objects.filter(create_user=request.user,role=1).order_by("{}create_time".format(sort_str)) elif role in ['2',2]: project_list = Project.objects.filter(role=2,role_value__contains=str(request.user.username)).order_by("{}create_time".format(sort_str)) elif role in ['3',3]: project_list = Project.objects.filter(role=3).order_by("{}create_time".format(sort_str)) elif role in ['99',99]: colla_list = [i.project.id for i in ProjectCollaborator.objects.filter(user=request.user)] # 用户的协作文集列表 project_list = Project.objects.filter(id__in=colla_list).order_by("{}create_time".format(sort_str)) else: return render(request,'404.html') # 没有搜索 and 游客 and 没有筛选 elif (is_kw is False) and (is_auth is False) and (is_role is False): project_list = Project.objects.filter(role__in=[0,3]).order_by("{}create_time".format(sort_str)) # 没有搜索 and 游客 and 有筛选 elif (is_kw is False) and (is_auth is False) and (is_role): if role in ['0',0]: project_list = Project.objects.filter(role=0).order_by("{}create_time".format(sort_str)) elif role in ['3',3]: project_list = Project.objects.filter(role=3).order_by("{}create_time".format(sort_str)) else: return render(request,'404.html') # 有搜索 and 认证用户 and 没有筛选 elif (is_kw) and (is_auth) and (is_role is False): colla_list = [i.project.id for i in ProjectCollaborator.objects.filter(user=request.user)] # 用户的协作文集 # 查询所有可显示的文集 project_list = Project.objects.filter( Q(role__in=[0, 3]) | \ Q(role=2, role_value__contains=str(request.user.username)) | \ Q(create_user=request.user) | \ Q(id__in=colla_list), Q(name__icontains=kw) | Q(intro__icontains=kw) ).order_by('{}create_time'.format(sort_str)) # 有搜索 and 认证用户 and 有筛选 elif (is_kw) and (is_auth) and (is_role): if role in ['0',0]: project_list = Project.objects.filter( Q(name__icontains=kw)|Q(intro__icontains=kw), role=0 ).order_by("{}create_time".format(sort_str)) elif role in ['1',1]: project_list = Project.objects.filter( Q(name__icontains=kw) | Q(intro__icontains=kw), create_user=request.user ).order_by("{}create_time".format(sort_str)) elif role in ['2',2]: project_list = Project.objects.filter( Q(name__icontains=kw) | Q(intro__icontains=kw), role=2, role_value__contains=str(request.user.username) ).order_by("{}create_time".format(sort_str)) elif role in ['3',3]: project_list = Project.objects.filter( Q(name__icontains=kw) | Q(intro__icontains=kw), role=3 ).order_by("{}create_time".format(sort_str)) elif role in ['99',99]: colla_list = [i.project.id for i in ProjectCollaborator.objects.filter(user=request.user)] # 用户的协作文集列表 project_list = Project.objects.filter( Q(name__icontains=kw) | Q(intro__icontains=kw), id__in=colla_list ).order_by("{}create_time".format(sort_str)) else: return render(request,'404.html') # 有搜索 and 游客 and 没有筛选 elif (is_kw) and (is_auth is False) and (is_role is False): project_list = Project.objects.filter( Q(name__icontains=kw) | Q(intro__icontains=kw), role__in=[0, 3] ).order_by("{}create_time".format(sort_str)) # 有搜索 and 游客 and 有筛选 elif (is_kw) and (is_auth is False) and (is_role): if role in ['0',0]: project_list = Project.objects.filter( Q(name__icontains=kw) | Q(intro__icontains=kw), role=0 ).order_by("{}create_time".format(sort_str)) elif role in ['3',3]: project_list = Project.objects.filter( Q(name__icontains=kw) | Q(intro__icontains=kw), role=3 ).order_by("{}create_time".format(sort_str)) else: return render(request,'404.html') # 分页处理 paginator = Paginator(project_list, 12) page = request.GET.get('page', 1) try: projects = paginator.page(page) except PageNotAnInteger: projects = paginator.page(1) except EmptyPage: projects = paginator.page(paginator.num_pages) return render(request, 'app_doc/pro_list.html', locals()) # 创建文集 @login_required() @require_http_methods(['POST']) def create_project(request): try: name = request.POST.get('pname','') name = validateTitle(name) icon = request.POST.get('picon',None) desc = request.POST.get('desc','') role = request.POST.get('role',0) role_list = ['0','1','2','3',0,1,2,3] if name != '': project = Project.objects.create( name=validateTitle(name), icon = icon, intro=desc[:100], create_user=request.user, role = int(role) if role in role_list else 0 ) project.save() return JsonResponse({'status':True,'data':{'id':project.id,'name':project.name,'role':project.role}}) else: return JsonResponse({'status':False,'data':'文集名称不能为空！'}) except Exception as e: logger.exception("创建文集出错") return JsonResponse({'status':False,'data':'出现异常,请检查输入值！'}) # 文集页 @require_http_methods(['GET']) @check_headers def project_index(request,pro_id): # 获取文集 try: # 获取文集信息 project = Project.objects.get(id=int(pro_id)) # 获取文集最新的5篇文档 new_docs = Doc.objects.filter(top_doc=pro_id,status=1).order_by('-modify_time')[:5] # markdown文本生成摘要（不带markdown标记） remove_markdown_tag(new_docs) # 获取文集的文档目录 toc_list,toc_cnt = get_pro_toc(pro_id) # toc_list,toc_cnt = ([],1000) # 获取文集的协作成员 colla_user_list = ProjectCollaborator.objects.filter(project=project) # 获取文集收藏状态 if request.user.is_authenticated: is_collect_pro = MyCollect.objects.filter( collect_type=2,collect_id=pro_id,create_user=request.user).exists() else: is_collect_pro = False # 获取文集的协作用户信息 if request.user.is_authenticated: # 对登陆用户查询其协作文档信息 colla_user = ProjectCollaborator.objects.filter(project=project,user=request.user).count() else: colla_user = 0 # 获取文集前台下载权限 try: allow_download = ProjectReport.objects.get(project=project) except ObjectDoesNotExist: allow_download = False # 私密文集并且访问者非创建者非协作者 if (project.role == 1) and (request.user != project.create_user) and (colla_user == 0): return render(request,'404.html') # 指定用户可见文集 elif project.role == 2: user_list = project.role_value if request.user.is_authenticated: # 认证用户判断是否在许可用户列表中 if (request.user.username not in user_list) and \ (request.user != project.create_user) and \ (colla_user == 0): # 访问者不在指定用户之中 return render(request, '404.html') else:# 游客直接返回404 return render(request, '404.html') # 访问码可见 elif project.role == 3: # 浏览用户不为创建者、协作者 if request.user != project.create_user and colla_user == 0: viewcode = project.role_value viewcode_name = 'viewcode-{}'.format(project.id) r_viewcode = request.COOKIES[viewcode_name] if viewcode_name in request.COOKIES.keys() else 0 # 从cookie中获取访问码 if viewcode != r_viewcode: # cookie中的访问码不等于文集访问码，跳转到访问码认证界面 return redirect('/check_viewcode/?to={}'.format(request.path)) # 获取搜索词 kw = request.GET.get('kw','') # 获取文集下所有一级文档 # project_docs = Doc.objects.filter( # top_doc=int(pro_id), # parent_doc=0, # status=1 # ).values('id','name','top_doc').order_by('sort') if kw != '': search_result = Doc.objects.filter(top_doc=int(pro_id),pre_content__icontains=kw) return render(request,'app_doc/project_doc_search.html',locals()) return render(request, 'app_doc/project.html', locals()) except Exception as e: logger.exception("文集页访问异常") return render(request,'404.html') # 修改文集 @login_required() @require_http_methods(['GET','POST']) def modify_project(request): if request.method == 'GET': pro_id = request.GET.get('pro_id', None) pro = Project.objects.get(id=pro_id) project_files = ProjectReportFile.objects.filter(project=pro) # 文集的导出文件列表 # 验证用户有权限修改文集 if (request.user == pro.create_user) or request.user.is_superuser: return render(request,'app_doc/manage/manage_project_options.html',locals()) else: return Http404 elif request.method == 'POST': try: pro_id = request.POST.get('pro_id',None) project = Project.objects.get(id=pro_id) # 验证用户有权限修改文集 if (request.user == project.create_user) or request.user.is_superuser: name = request.POST.get('name',None) icon = request.POST.get('picon', None) content = request.POST.get('desc',None) is_watermark = request.POST.get('is_watermark',False) if is_watermark == 'true': is_watermark = True else: is_watermark = False watermark_value = request.POST.get('watermark_value','') project.name = validateTitle(name) project.intro = content project.icon = icon project.is_watermark = is_watermark project.watermark_value = watermark_value project.save() return JsonResponse({'status':True,'data':'修改成功'}) else: return JsonResponse({'status':False,'data':'非法请求'}) except Exception as e: logger.exception("修改文集出错") return JsonResponse({'status':False,'data':'请求出错'}) # 修改文集权限 @login_required() @require_http_methods(['GET',"POST"]) @logger.catch() def modify_project_role(request,pro_id): try: pro = Project.objects.get(id=pro_id) except ObjectDoesNotExist: return Http404 if (pro.create_user != request.user) and (request.user.is_superuser is False): return render(request,'403.html') else: if request.method == 'GET': return render(request,'app_doc/manage/manage_project_role.html',locals()) elif request.method == 'POST': role_type = request.POST.get('role','') if role_type != '': try: if int(role_type) in [0,1]:# 公开或私密 Project.objects.filter(id=int(pro_id)).update( role = role_type, modify_time = datetime.datetime.now() ) if int(role_type) == 2: # 指定用户可见 role_value = request.POST.get('tagsinput','') Project.objects.filter(id=int(pro_id)).update( role=role_type, role_value = role_value, modify_time = datetime.datetime.now() ) if int(role_type) == 3: # 访问码可见 role_value = request.POST.get('viewcode','') Project.objects.filter(id=int(pro_id)).update( role=role_type, role_value=role_value, modify_time=datetime.datetime.now() ) pro = Project.objects.get(id=int(pro_id)) # return render(request, 'app_doc/manage/manage_project_role.html', locals()) return JsonResponse({'status':True,'data':'ok'}) except: return JsonResponse({'status':False,'data':'出错'}) else: return Http404 # 验证文集访问码 @require_http_methods(['GET',"POST"]) def check_viewcode(request): try: if request.method == 'GET': project_id = request.GET.get('to','').split("/")[1].split('-')[1] project = Project.objects.get(id=int(project_id)) return render(request,'app_doc/check_viewcode.html',locals()) else: viewcode = request.POST.get('viewcode','') project_id = request.POST.get('project_id','') project = Project.objects.get(id=int(project_id)) if project.role == 3 and project.role_value == viewcode: obj = redirect("pro_index",pro_id=project_id) obj.set_cookie('viewcode-{}'.format(project_id),viewcode) return obj else: errormsg = "访问码错误" return render(request, 'app_doc/check_viewcode.html', locals()) except Exception as e: logger.exception("验证文集访问码出错") return render(request,'404.html') # 删除文集 @login_required() @require_http_methods(["POST"]) def del_project(request): try: range = request.POST.get('range','single') pro_id = request.POST.get('pro_id','') if pro_id != '': if range == 'single': pro = Project.objects.get(id=pro_id) if (request.user == pro.create_user) or (request.user.is_superuser): # 删除文集下的文档、文档历史、文档分享、文档标签 pro_doc_list = Doc.objects.filter(top_doc=int(pro_id)) for doc in pro_doc_list: DocHistory.objects.filter(doc=doc).delete() DocShare.objects.filter(doc=doc).delete() DocTag.objects.filter(doc=doc).delete() pro_doc_list.delete() # 删除文集 pro.delete() return JsonResponse({'status':True}) else: return JsonResponse({'status':False,'data':'非法请求'}) elif range == 'multi': pros = pro_id.split(",") try: projects = Project.objects.filter(id__in=pros, create_user=request.user) # 删除文集下的文档、文档历史、文档分享、文档标签 pro_doc_list = Doc.objects.filter(top_doc__in=[i.id for i in projects]) for doc in pro_doc_list: DocHistory.objects.filter(doc=doc).delete() DocShare.objects.filter(doc=doc).delete() DocTag.objects.filter(doc=doc).delete() pro_doc_list.delete() projects.delete() return JsonResponse({'status': True, 'data': 'ok'}) except Exception: logger.exception("异常") return JsonResponse({'status': False, 'data': '无指定内容'}) else: return JsonResponse({'status': False, 'data': '类型错误'}) else: return JsonResponse({'status':False,'data':'参数错误'}) except Exception as e: logger.exception("删除文集出错") return JsonResponse({'status':False,'data':'请求出错'}) # 管理文集 @login_required() @require_http_methods(['GET','POST']) def manage_project(request): if request.method == 'GET': return render(request,'app_doc/manage/manage_project.html',locals()) else: kw = request.POST.get('kw','') page = request.POST.get('page', 1) limit = request.POST.get('limit', 10) # 获取文集列表 if kw == '': project_list = Project.objects.filter(create_user=request.user).order_by('-create_time') else: project_list = Project.objects.filter( Q(intro__icontains=kw) | Q(name__icontains=kw), create_user=request.user, ).order_by('-create_time') paginator = Paginator(project_list, limit) try: pros = paginator.page(page) except PageNotAnInteger: pros = paginator.page(1) except EmptyPage: pros = paginator.page(paginator.num_pages) table_data = [] for project in pros: item = { 'id':project.id, 'name':project.name, 'intro':project.intro, 'doc_total':Doc.objects.filter(top_doc=project.id).count(), 'role':project.role, 'role_value':project.role_value, 'colla_total':ProjectCollaborator.objects.filter(project=project).count(), 'create_time':project.create_time, 'modify_time':project.modify_time } table_data.append(item) resp_data = { "code": 0, "msg": "ok", "count": project_list.count(), "data": table_data } return JsonResponse(resp_data) # 管理文集 - 文集文档排序 @login_required() @require_http_methods(['GET','POST']) def manage_project_doc_sort(request,pro_id): if request.method == 'GET': try: # 获取文集 pro = Project.objects.get(id=pro_id) except ObjectDoesNotExist: return render(request, '404.html') # 查询文集的协作者 pro_colla = ProjectCollaborator.objects.filter(project=pro,user=request.user,role=1) # 文集的创建者和文集高级权限协作者允许操作 if (pro.create_user == request.user) or pro_colla.count() > 0: # 查询存在上级文档的文档 parent_id_list = Doc.objects.filter(top_doc=pro_id, status=1).exclude(parent_doc=0).values_list( 'parent_doc', flat=True) # 获取存在上级文档的上级文档ID doc_list = [] # 获取一级文档 top_docs = Doc.objects.filter(top_doc=pro_id, parent_doc=0, status=1).values('id', 'name').order_by('sort') # 遍历一级文档 for doc in top_docs: top_item = { 'id': doc['id'], 'field': doc['name'], 'title': doc['name'], 'spread': True, 'level': 1 } # 如果一级文档存在下级文档，查询其二级文档 if doc['id'] in parent_id_list: # 获取二级文档 sec_docs = Doc.objects.filter(top_doc=pro_id, parent_doc=doc['id'], status=1).values('id', 'name').order_by( 'sort') top_item['children'] = [] for doc in sec_docs: sec_item = { 'id': doc['id'], 'field': doc['name'], 'title': doc['name'], 'level': 2 } # 如果二级文档存在下级文档，查询第三级文档 if doc['id'] in parent_id_list: # 获取三级文档 thr_docs = Doc.objects.filter(top_doc=pro_id, parent_doc=doc['id'], status=1).values('id', 'name').order_by( 'sort') sec_item['children'] = [] for doc in thr_docs: item = { 'id': doc['id'], 'field': doc['name'], 'title': doc['name'], 'level': 3 } sec_item['children'].append(item) top_item['children'].append(sec_item) else: top_item['children'].append(sec_item) doc_list.append(top_item) # 如果一级文档没有下级文档，直接保存 else: doc_list.append(top_item) return render(request,'app_doc/manage/manage_project_doc_sort.html',locals()) else: return render(request, '403.html') else: project_id = request.POST.get('pid', None) # 文集ID sort_data = request.POST.get('sort_data', '[]') # 文档排序列表 try: sort_data = json.loads(sort_data) except Exception: return JsonResponse({'status': False, 'data': '文档参数错误'}) try: pro = Project.objects.get(id=project_id) except ObjectDoesNotExist: return JsonResponse({'status': False, 'data': '没有匹配的文集'}) # 查询文集的协作者 pro_colla = ProjectCollaborator.objects.filter(project=pro, user=request.user, role=1) # 文集的创建者和文集高级权限协作者允许操作 if (pro.create_user == request.user) or pro_colla.count() > 0: # 文档排序 n = 10 # 第一级文档 for data in sort_data: # print(data) Doc.objects.filter(id=data['id']).update(sort=n,parent_doc=0) n += 10 # 存在第二级文档 if 'children' in data.keys(): n1 = 10 for c1 in data['children']: Doc.objects.filter(id=c1['id']).update(sort=n1, parent_doc=data['id']) n1 += 10 # 存在第三级文档 if 'children' in c1.keys(): n2 = 10 for c2 in c1['children']: Doc.objects.filter(id=c2['id']).update(sort=n2, parent_doc=c1['id']) return JsonResponse({'status': True, 'data': 'ok'}) else: return JsonResponse({'status':False,'data':'无权操作'}) # 修改文集前台下载权限 @login_required() @require_http_methods(['GET',"POST"]) @logger.catch() def modify_project_download(request,pro_id): try: pro = Project.objects.get(id=pro_id) except ObjectDoesNotExist: return Http404 if (pro.create_user != request.user) and (request.user.is_superuser is False): return render(request,'403.html') else: project_files = ProjectReportFile.objects.filter(project=pro) if request.method == 'GET': return render(request,'app_doc/manage/manage_project_download.html',locals()) elif request.method == 'POST': download_epub = request.POST.get('download_epub',None) download_pdf = request.POST.get('download_pdf', None) # print("epub状态:",download_epub) # EPUB下载权限 if download_epub == 'on': epub_status = 1 else: epub_status = 0 # PDF下载权限 if download_pdf == 'on': pdf_status = 1 else: pdf_status = 0 # 写入数据库 ProjectReport.objects.update_or_create( project = pro,defaults={'allow_epub':epub_status} ) ProjectReport.objects.update_or_create( project=pro, defaults={'allow_pdf': pdf_status} ) # return render(request,'app_doc/manage/manage_project_download.html',locals()) return JsonResponse({'status':True,'data':'ok'}) # 文集协作管理 @login_required() @require_http_methods(['GET',"POST"]) @logger.catch() def manage_project_collaborator(request,pro_id): if request.user.is_superuser: project = Project.objects.filter(id=pro_id) else: project = Project.objects.filter(id=pro_id, create_user=request.user) if project.exists() is False: return render(request, '404.html') if request.method == 'GET': user_list = User.objects.filter(~Q(username=request.user.username)) # 获取用户列表 pro = project[0] collaborator = ProjectCollaborator.objects.filter(project=pro) # 获取文集的协作者 colla_user_list = [i.user for i in collaborator] # 文集协作用户的ID colla_docs = Doc.objects.filter(top_doc=pro.id,create_user__in=colla_user_list) # 获取文集协作用户创建的文档 return render(request, 'app_doc/manage/manage_project_collaborator.html', locals()) elif request.method == 'POST': # type类型：0表示新增协作者、1表示删除协作者、2表示修改协作者 types = request.POST.get('types','') try: types = int(types) except: return JsonResponse({'status':False,'data':'参数错误'}) # 添加文集协作者 if int(types) == 0: colla_user = request.POST.get('username','') role = request.POST.get('role',0) user = User.objects.filter(username=colla_user) if user.exists(): if user[0] == project[0].create_user: # 用户为文集的创建者 return JsonResponse({'status':False,'data':'文集创建者无需添加'}) elif ProjectCollaborator.objects.filter(user=user[0],project=project[0]).exists(): return JsonResponse({'status':False,'data':'用户已存在'}) else: ProjectCollaborator.objects.create( project = project[0], user = user[0], role = role if role in ['1',1] else 0 ) return JsonResponse({'status':True,'data':'添加成功'}) else: return JsonResponse({'status':False,'data':'用户不存在'}) # 删除文集协作者 elif int(types) == 1: username = request.POST.get('username','') try: user = User.objects.get(username=username) pro_colla = ProjectCollaborator.objects.get(project=project[0],user=user) pro_colla.delete() return JsonResponse({'status':True,'data':'删除成功'}) except: logger.exception("删除协作者出错") return JsonResponse({'status':False,'data':'删除出错'}) # 修改协作权限 elif int(types) == 2: username = request.POST.get('username', '') role = request.POST.get('role','') try: user = User.objects.get(username=username) pro_colla = ProjectCollaborator.objects.filter(project=project[0], user=user) pro_colla.update(role=role) return JsonResponse({'status':True,'data':'修改成功'}) except: logger.exception("修改协作权限出错") return JsonResponse({'status':False,'data':'修改失败'}) else: return JsonResponse({'status':False,'data':'无效的类型'}) # 我协作的文集 @login_required() @logger.catch() def manage_pro_colla_self(request): colla_pros = ProjectCollaborator.objects.filter(user=request.user) return render(request,'app_doc/manage/manage_project_self_colla.html',locals()) # 转让文集 @login_required() @require_http_methods(['GET',"POST"]) def manage_project_transfer(request,pro_id): try: pro = Project.objects.get(id=pro_id) except ObjectDoesNotExist: return Http404 if (pro.create_user != request.user) and (request.user.is_superuser is False): return render(request,'403.html') else: if request.method == 'GET': user_list = User.objects.filter(~Q(username=request.user.username)) return render(request,'app_doc/manage/manage_project_transfer.html',locals()) elif request.method == 'POST': user_name = request.POST.get('username',None) try: transfer_user = User.objects.get(username=user_name) init_user = pro.create_user # 修改文集的创建者 pro.create_user = transfer_user pro.save() # 修改文集文档的创建者 Doc.objects.filter(create_user=init_user,top_doc=pro_id).update( create_user=transfer_user ) return JsonResponse({'status':True,'data':'ok'}) except: return JsonResponse({'status':False,'data':'用户不存在'}) # 文档浏览页 @require_http_methods(['GET']) def doc(request,pro_id,doc_id): try: if pro_id != '' and doc_id != '': # 获取文集信息 project = Project.objects.get(id=int(pro_id)) # 获取文集的文档目录 toc_list,toc_cnt = get_pro_toc(pro_id) # 获取文集的协作用户信息 if request.user.is_authenticated: colla_user = ProjectCollaborator.objects.filter(project=project,user=request.user) if colla_user.exists(): colla_user_role = colla_user[0].role colla_user = colla_user.count() else: colla_user = colla_user.count() else: colla_user = 0 # 获取文集收藏状态 if request.user.is_authenticated: is_collect_pro = MyCollect.objects.filter(collect_type=2, collect_id=pro_id, create_user=request.user).exists() # 获取文档收藏状态 is_collect_doc = MyCollect.objects.filter(collect_type=1, collect_id=doc_id, create_user=request.user).exists() else: is_collect_pro,is_collect_doc = False,False # 私密文集且访问者非创建者、协作者 - 不能访问 if (project.role == 1) and (request.user != project.create_user) and (colla_user == 0): return render(request, '404.html') # 指定用户可见文集 elif project.role == 2: user_list = project.role_value if request.user.is_authenticated: # 认证用户判断是否在许可用户列表中 if (request.user.username not in user_list) and \ (request.user != project.create_user) and \ (colla_user == 0): # 访问者不在指定用户之中，也不是协作者 return render(request, '404.html') else: # 游客直接返回404 return render(request, '404.html') # 访问码可见 elif project.role == 3: # 浏览用户不为创建者和协作者 - 需要访问码 if (request.user != project.create_user) and (colla_user == 0): viewcode = project.role_value viewcode_name = 'viewcode-{}'.format(project.id) r_viewcode = request.COOKIES[ viewcode_name] if viewcode_name in request.COOKIES.keys() else 0 # 从cookie中获取访问码 if viewcode != r_viewcode: # cookie中的访问码不等于文集访问码，跳转到访问码认证界面 return redirect('/check_viewcode/?to={}'.format(request.path)) # 获取文档内容 try: doc = Doc.objects.get(id=int(doc_id),status=1) # 文档信息 doc_tags = DocTag.objects.filter(doc=doc) # 文档标签信息 except ObjectDoesNotExist: return render(request, '404.html') # 获取文档分享信息 try: doc_share = DocShare.objects.get(doc=doc) is_share = True except ObjectDoesNotExist: is_share = False # 获取文集下一级文档 # project_docs = Doc.objects.filter(top_doc=doc.top_doc, parent_doc=0, status=1).order_by('sort') return render(request,'app_doc/doc.html',locals()) else: return HttpResponse('参数错误') except Exception as e: logger.exception("文集浏览出错") return render(request,'404.html') # 创建文档 @login_required() @require_http_methods(['GET',"POST"]) @logger.catch() def create_doc(request): # 获取用户的编辑器模式 try: user_opt = UserOptions.objects.get(user=request.user) editor_mode = user_opt.editor_mode except ObjectDoesNotExist: editor_mode = 1 if request.method == 'GET': # 获取url切换的编辑器模式 eid = request.GET.get('eid',editor_mode) if eid in [1,2,3,'1','2','3']: editor_mode = int(eid) try: editor_type = "新建文档" pid = request.GET.get('pid',-999) project_list = Project.objects.filter(create_user=request.user) # 自己创建的文集列表 colla_project_list = ProjectCollaborator.objects.filter(user=request.user) # 协作的文集列表 doctemp_list = DocTemp.objects.filter(create_user=request.user).values('id','name','create_time') return render(request, 'app_doc/editor/create_doc.html', locals()) except Exception as e: logger.exception("访问创建文档页面出错") return render(request,'404.html') elif request.method == 'POST': try: project = request.POST.get('project','') # 文集ID parent_doc = request.POST.get('parent_doc','') # 上级文档ID doc_name = request.POST.get('doc_name','') # 文档标题 doc_tags = request.POST.get('doc_tag','') # 文档标签 doc_content = request.POST.get('content','') # 文档内容 pre_content = request.POST.get('pre_content','') # 文档Markdown内容 sort = request.POST.get('sort','') # 文档排序 editor_mode = request.POST.get('editor_mode',editor_mode) #获取文档编辑器 status = request.POST.get('status',1) # 文档状态 open_children = request.POST.get('open_children', False) # 展示下级目录 show_children = request.POST.get('show_children', False) # 展示下级目录 if open_children == 'on': open_children = True else: open_children = False if show_children == 'on': show_children = True else: show_children = False if project != '' and doc_name != '' and project != '-1': # 验证请求者是否有文集的权限 check_project = Project.objects.filter(id=project,create_user=request.user) colla_project = ProjectCollaborator.objects.filter(project=project,user=request.user) if check_project.count() > 0 or colla_project.count() > 0: # 开启事务 with transaction.atomic(): save_id = transaction.savepoint() try: # 创建文档 doc = Doc.objects.create( name=doc_name, content = doc_content, pre_content= pre_content, parent_doc= int(parent_doc) if parent_doc != '' else 0, top_doc= int(project), sort = sort if sort != '' else 99, create_user=request.user, status = status, editor_mode = editor_mode, open_children = open_children, show_children = show_children ) # 设置文档标签 for t in doc_tags.split(","): if t != '': tag = Tag.objects.get_or_create(name=t,create_user=request.user) DocTag.objects.get_or_create(tag=tag[0],doc=doc) return JsonResponse({'status': True, 'data': {'pro': project, 'doc': doc.id}}) except Exception as e: logger.exception("创建文档异常") # 回滚事务 transaction.savepoint_rollback(save_id) transaction.savepoint_commit(save_id) return JsonResponse({'status': False, 'data': '创建失败'}) else: return JsonResponse({'status':False,'data':'无权操作此文集'}) else: return JsonResponse({'status':False,'data':'请确认文档标题、文集正确'}) except Exception as e: logger.exception("创建文档出错") return JsonResponse({'status':False,'data':'请求出错'}) else: return JsonResponse({'status':False,'data':'方法不允许'}) # 修改文档 @login_required() @require_http_methods(['GET',"POST"]) def modify_doc(request,doc_id): editor_type = "修改文档" if request.method == 'GET': try: doc = Doc.objects.get(id=doc_id) # 查询文档信息 editor_mode = doc.editor_mode # 设置文档编辑器为文档上一次使用的编辑模式 eid = request.GET.get('eid',editor_mode) if eid in [1,2,3,'1','2','3']: editor_mode = int(eid) doc_tags = ','.join([i.tag.name for i in DocTag.objects.filter(doc=doc)]) # 查询文档标签信息 project = Project.objects.get(id=doc.top_doc) # 查询文档所属的文集信息 pro_colla = ProjectCollaborator.objects.filter(project=project,user=request.user) # 查询用户的协作文集信息 if pro_colla.count() == 0: is_pro_colla = False elif pro_colla[0].role == 1: is_pro_colla = True else: is_pro_colla = False project_list = Project.objects.filter(create_user=request.user) # 自己创建的文集列表 colla_project_list = ProjectCollaborator.objects.filter(user=request.user) # 协作的文集列表 # 判断用户是否有权限进行修改 if (request.user == doc.create_user) or \ (is_pro_colla is True) or \ (request.user == project.create_user): doc_list = Doc.objects.filter(top_doc=project.id) doctemp_list = DocTemp.objects.filter(create_user=request.user) history_list = DocHistory.objects.filter(doc=doc).order_by('-create_time') return render(request, 'app_doc/editor/modify_doc.html', locals()) else: return render(request,'403.html') except Exception as e: logger.exception("修改文档页面访问出错") return render(request,'404.html') elif request.method == 'POST': try: doc_id = request.POST.get('doc_id','') # 文档ID project_id = request.POST.get('project', '') # 文集ID parent_doc = request.POST.get('parent_doc', '') # 上级文档ID doc_name = request.POST.get('doc_name', '') # 文档名称 doc_tags = request.POST.get('doc_tag','') # 文档标签 doc_content = request.POST.get('content', '') # 文档内容 pre_content = request.POST.get('pre_content', '') # 文档Markdown格式内容 sort = request.POST.get('sort', '') # 文档排序 editor_mode = request.POST.get('editor_mode',1) #获取文档编辑器 status = request.POST.get('status',1) # 文档状态 open_children = request.POST.get('open_children',False) # 展示下级目录 show_children = request.POST.get('show_children', False) # 展示下级目录 if open_children == 'on': open_children = True else: open_children = False if show_children == 'on': show_children = True else: show_children = False if doc_id != '' and project_id != '' and doc_name != '' and project_id != '-1': doc = Doc.objects.get(id=doc_id) project = Project.objects.get(id=project_id) pro_colla = ProjectCollaborator.objects.filter(project=project, user=request.user) if pro_colla.count() == 0: is_pro_colla = False elif pro_colla[0].role == 1: is_pro_colla = True else: is_pro_colla = False # 验证用户有权限修改文档 - 文档的创建者或文集的高级协作者 if (request.user == doc.create_user) or (is_pro_colla is True) or (request.user == project.create_user): # 开启事务 with transaction.atomic(): save_id = transaction.savepoint() try: # 将现有文档内容写入到文档历史中 DocHistory.objects.create( doc = doc, pre_content = doc.pre_content, create_user = request.user ) # 更新文档内容 Doc.objects.filter(id=int(doc_id)).update( name=doc_name, content=doc_content, pre_content=pre_content, parent_doc=int(parent_doc) if parent_doc != '' else 0, sort=sort if sort != '' else 99, modify_time = datetime.datetime.now(), status = status, editor_mode = editor_mode, open_children = open_children, show_children = show_children ) # 更新文档标签 doc_tag_list = doc_tags.split(",") if doc_tags != "" else [] # print(doc_tags,doc_tag_list) # 如果没有设置标签，则删除此文档的所有标签 if len(doc_tag_list) == 0: DocTag.objects.filter(doc=doc).delete() else: current_doc_tags = [i.tag.name for i in DocTag.objects.filter(doc=doc)] # 获取当前文档的标签 # 遍历当前文档标签，如果不在新的标签列表，则删除 for tag in current_doc_tags: if tag not in doc_tag_list: tag = Tag.objects.get(name=tag,create_user=request.user) DocTag.objects.filter(doc=doc,tag=tag).delete() # 遍历新的标签列表，如果不在当前文档标签中，则创建 for t in doc_tag_list: if t not in current_doc_tags and current_doc_tags != '': tag = Tag.objects.get_or_create(name=t, create_user=request.user) DocTag.objects.get_or_create(tag=tag[0], doc=doc) return JsonResponse({'status': True, 'data': '修改成功'}) except: logger.exception("修改文档异常") # 回滚事务 transaction.savepoint_rollback(save_id) transaction.savepoint_commit(save_id) return JsonResponse({'status': False, 'data': '修改失败'}) else: return JsonResponse({'status':False,'data':'未授权请求'}) else: return JsonResponse({'status': False,'data':'参数错误'}) except Exception as e: logger.exception("修改文档出错") return JsonResponse({'status':False,'data':'请求出错'}) # 删除文档 - 软删除 - 进入回收站 @login_required() @require_http_methods(["POST"]) def del_doc(request): try: # 获取文档ID doc_id = request.POST.get('doc_id',None) range = request.POST.get('range', 'single') if doc_id: if range == 'single': # 查询文档 try: doc = Doc.objects.get(id=doc_id) try: project = Project.objects.get(id=doc.top_doc) # 查询文档所属的文集 except ObjectDoesNotExist: logger.error("文档{}的所属文集不存在。".format(doc_id)) project = 0 # 获取文档所属文集的协作信息 pro_colla = ProjectCollaborator.objects.filter(project=project,user=request.user) if pro_colla.exists(): colla_user_role = pro_colla[0].role else: colla_user_role = 0 except ObjectDoesNotExist: return JsonResponse({'status': False, 'data': '文档不存在'}) # 如果请求用户为站点管理员、文档创建者、高级权限的协作者、文集的创建者，可以删除 if (request.user == doc.create_user) \ or (colla_user_role == 1) \ or (request.user == project.create_user)\ or (request.user.is_superuser): # 修改状态为删除 doc.status = 3 doc.modify_time = datetime.datetime.now() doc.save() # 修改其下级所有文档状态为删除 chr_doc = Doc.objects.filter(parent_doc=doc_id) # 获取下级文档 chr_doc_ids = chr_doc.values_list('id',flat=True) # 提取下级文档的ID chr_doc.update(status=3,modify_time=datetime.datetime.now()) # 修改下级文档的状态为删除 Doc.objects.filter(parent_doc__in=chr_doc_ids).update(status=3,modify_time=datetime.datetime.now()) # 修改下级文档的下级文档状态 return JsonResponse({'status': True, 'data': '删除完成'}) else: return JsonResponse({'status': False, 'data': '非法请求'}) elif range == 'multi': docs = doc_id.split(",") try: Doc.objects.filter(id__in=docs,create_user=request.user).update(status=3,modify_time=datetime.datetime.now()) Doc.objects.filter(parent_doc__in=docs).update(status=3,modify_time=datetime.datetime.now()) return JsonResponse({'status': True, 'data': '删除完成'}) except: return JsonResponse({'status': False, 'data': '非法请求'}) else: return JsonResponse({'status': False, 'data': '类型错误'}) else: return JsonResponse({'status':False,'data':'参数错误'}) except Exception as e: logger.exception("删除文档出错") return JsonResponse({'status':False,'data':'请求出错'}) # 管理文档 @login_required() @require_http_methods(['GET','POST']) def manage_doc(request): if request.method == 'GET': # 文集列表 project_list = Project.objects.filter(create_user=request.user) # 自己创建的文集列表 colla_project_list = ProjectCollaborator.objects.filter(user=request.user) # 协作的文集列表 # 文档数量 # 已发布文档数量 published_doc_cnt = Doc.objects.filter(create_user=request.user, status=1).count() # 草稿文档数量 draft_doc_cnt = Doc.objects.filter(create_user=request.user, status=0).count() # 所有文档数量 all_cnt = published_doc_cnt + draft_doc_cnt return render(request,'app_doc/manage/manage_doc.html',locals()) else: kw = request.POST.get('kw', '') project = request.POST.get('project','') status = request.POST.get('status','') if status == '-1': # 全部文档 q_status = [0,1] elif status in ['0','1']: q_status = [int(status)] else: q_status = [0, 1] if project == '': project_list = Project.objects.filter(create_user=request.user).values_list('id',flat=True) # 自己创建的文集列表 colla_project_list = ProjectCollaborator.objects.filter(user=request.user).values_list('project__id',flat=True) # 协作的文集列表 q_project = list(project_list) + list(colla_project_list) else: q_project = [project] page = request.POST.get('page', 1) limit = request.POST.get('limit', 10) # 没有搜索 if kw == '': doc_list = Doc.objects.filter( create_user=request.user, status__in=q_status, top_doc__in=q_project ).order_by('-modify_time') # 有搜索 else: doc_list = Doc.objects.filter( Q(content__icontains=kw) | Q(name__icontains=kw), create_user=request.user,status__in=q_status,top_doc__in=q_project ).order_by('-modify_time') # 文集列表 project_list = Project.objects.filter(create_user=request.user) # 自己创建的文集列表 colla_project_list = ProjectCollaborator.objects.filter(user=request.user) # 协作的文集列表 # 文档数量 # 已发布文档数量 published_doc_cnt = Doc.objects.filter(create_user=request.user, status=1).count() # 草稿文档数量 draft_doc_cnt = Doc.objects.filter(create_user=request.user, status=0).count() # 所有文档数量 all_cnt = published_doc_cnt + draft_doc_cnt # 分页处理 paginator = Paginator(doc_list, limit) page = request.GET.get('page', page) try: docs = paginator.page(page) except PageNotAnInteger: docs = paginator.page(1) except EmptyPage: docs = paginator.page(paginator.num_pages) table_data = [] for doc in docs: item = { 'id': doc.id, 'name': doc.name, 'parent':Doc.objects.get(id=doc.parent_doc).name if doc.parent_doc != 0 else '无', 'project_id': Project.objects.get(id=doc.top_doc).id, 'project_name':Project.objects.get(id=doc.top_doc).name, 'status':doc.status, 'editor_mode':doc.editor_mode, 'open_children':doc.open_children, 'create_time': doc.create_time, 'modify_time': doc.modify_time } table_data.append(item) resp_data = { "code": 0, "msg": "ok", "count": doc_list.count(), "data": table_data } return JsonResponse(resp_data) # 移动文档 @login_required() @require_http_methods(['POST']) def move_doc(request): doc_id = request.POST.get('doc_id','') # 文档ID pro_id = request.POST.get('pro_id','') # 移动的文集ID move_type = request.POST.get('move_type','') # 移动的类型 0复制 1移动 2连同下级文档移动 parent_id = request.POST.get('parent_id',0) # 判断文集是否存在且有权限 try: project = Project.objects.get(id=int(pro_id)) # 自己的文集 colla = ProjectCollaborator.objects.filter(project=project, user=request.user) # 协作文集 if (project.create_user is not request.user) and (colla.count()): # 请求者不是文集创建者和协作者返回错误 return JsonResponse({'status':False,'data':'文集无权限'}) except ObjectDoesNotExist: return JsonResponse({'status':False,'data':'文集不存在'}) # 判断文档是否存在 try: doc = Doc.objects.get(id=int(doc_id),create_user=request.user) except ObjectDoesNotExist: return JsonResponse({'status':False,'data':'文档不存在'}) # 判断上级文档是否存在 try: if parent_id != '0': parent = Doc.objects.get(id=int(parent_id),create_user=request.user) except ObjectDoesNotExist: return JsonResponse({'status':False,'data':'上级文档不存在'}) # 复制文档 if move_type == '0': copy_doc = Doc.objects.create( name = doc.name, pre_content = doc.pre_content, content = doc.content, parent_doc = parent_id, top_doc = int(pro_id), create_user = request.user, create_time = datetime.datetime.now(), modify_time = datetime.datetime.now(), # 文档状态说明：0表示草稿状态，1表示发布状态 status = doc.status ) return JsonResponse({'status':True,'data':{'pro_id':pro_id,'doc_id':copy_doc.id}}) # 移动文档，下级文档更改到根目录 elif move_type == '1': try: # 修改文档的所属文集和上级文档实现移动文档 Doc.objects.filter(id=int(doc_id)).update(parent_doc=int(parent_id),top_doc=int(pro_id)) # 修改其子文档为顶级文档 Doc.objects.filter(parent_doc=doc_id).update(parent_doc=0) return JsonResponse({'status':True,'data':{'pro_id':pro_id,'doc_id':doc_id}}) except: logger.exception("移动文档异常") return JsonResponse({'status':False,'data':'移动文档失败'}) # 包含下级文档一起移动 elif move_type == '2': try: # 修改文档的所属文集和上级文档实现移动文档 Doc.objects.filter(id=int(doc_id)).update(parent_doc=int(parent_id), top_doc=int(pro_id)) # 修改其子文档的文集归属 child_doc = Doc.objects.filter(parent_doc=doc_id) child_doc.update(top_doc=int(pro_id)) # 遍历子文档，如果其存在下级文档，那么继续修改所属文集 for child in child_doc: Doc.objects.filter(parent_doc=child.id).update(top_doc=int(pro_id)) return JsonResponse({'status': True, 'data':{'pro_id':pro_id,'doc_id':doc_id}}) except: logger.exception("移动包含下级的文档异常") return JsonResponse({'status': False, 'data': '移动文档失败'}) else: return JsonResponse({'status':False,'data':'移动类型错误'}) # 查看对比文档历史版本 @login_required() @require_http_methods(['GET',"POST"]) def diff_doc(request,doc_id,his_id): if request.method == 'GET': try: doc = Doc.objects.get(id=doc_id) # 查询文档信息 project = Project.objects.get(id=doc.top_doc) # 查询文档所属的文集信息 pro_colla = ProjectCollaborator.objects.filter(project=project, user=request.user) # 查询用户的协作文集信息 if (request.user == doc.create_user) or (pro_colla[0].role == 1): history = DocHistory.objects.get(id=his_id) history_list = DocHistory.objects.filter(doc=doc).order_by('-create_time') if history.doc == doc: return render(request, 'app_doc/diff_doc.html', locals()) else: return render(request, '403.html') else: return render(request, '403.html') except Exception as e: logger.exception("文档历史版本页面访问出错") return render(request, '404.html') elif request.method == 'POST': try: doc = Doc.objects.get(id=doc_id) # 查询文档信息 project = Project.objects.get(id=doc.top_doc) # 查询文档所属的文集信息 pro_colla = ProjectCollaborator.objects.filter(project=project, user=request.user) # 查询用户的协作文集信息 if (request.user == doc.create_user) or (pro_colla[0].role == 1): history = DocHistory.objects.get(id=his_id) if history.doc == doc: return JsonResponse({'status':True,'data':history.pre_content}) else: return JsonResponse({'status': False, 'data': '非法请求'}) else: return JsonResponse({'status':False,'data':'非法请求'}) except Exception as e: logger.exception("文档历史版本获取出错") return JsonResponse({'status':False,'data':'获取异常'}) # 管理文档历史版本 @login_required() @require_http_methods(['GET',"POST"]) def manage_doc_history(request,doc_id): if request.method == 'GET': try: doc = Doc.objects.get(id=doc_id,create_user=request.user) history_list = DocHistory.objects.filter(create_user=request.user,doc=doc_id).order_by('-create_time') paginator = Paginator(history_list, 15) page = request.GET.get('page', 1) try: historys = paginator.page(page) except PageNotAnInteger: historys = paginator.page(1) except EmptyPage: historys = paginator.page(paginator.num_pages) return render(request, 'app_doc/manage/manage_doc_history.html', locals()) except Exception as e: logger.exception("管理文档历史版本页面访问出错") return render(request, '404.html') elif request.method == 'POST': try: history_id = request.POST.get('history_id','') DocHistory.objects.filter(id=history_id,doc=doc_id,create_user=request.user).delete() return JsonResponse({'status':True,'data':'删除成功'}) except: logger.exception("操作文档历史版本出错") return JsonResponse({'status':False,'data':'出现异常'}) # 文档回收站 @login_required() @require_http_methods(['GET','POST']) def doc_recycle(request): if request.method == 'GET': # 获取状态为删除的文档 doc_list = Doc.objects.filter(status=3,create_user=request.user).order_by('-modify_time') # 分页处理 paginator = Paginator(doc_list, 15) page = request.GET.get('page', 1) try: docs = paginator.page(page) except PageNotAnInteger: docs = paginator.page(1) except EmptyPage: docs = paginator.page(paginator.num_pages) return render(request,'app_doc/manage/manage_doc_recycle.html',locals()) elif request.method == 'POST': try: # 获取参数 doc_id = request.POST.get('doc_id', None) # 文档ID types = request.POST.get('type',None) # 操作类型 if doc_id: # 查询文档 try: doc = Doc.objects.get(id=doc_id) project = Project.objects.get(id=doc.top_doc) # 查询文档所属的文集 # 获取文档所属文集的协作信息 pro_colla = ProjectCollaborator.objects.filter(project=project, user=request.user) # if pro_colla.exists(): colla_user_role = pro_colla[0].role else: colla_user_role = 0 except ObjectDoesNotExist: return JsonResponse({'status': False, 'data': '文档不存在'}) # 如果请求用户为文档创建者、高级权限的协作者、文集的创建者，可以操作 if (request.user == doc.create_user) or (colla_user_role == 1) or (request.user == project.create_user): # 还原文档 if types == 'restore': # 修改状态为草稿 doc.status = 0 doc.modify_time = datetime.datetime.now() doc.save() # 删除文档 elif types == 'del': # 删除文档历史、分享、标签 DocHistory.objects.filter(doc=doc).delete() DocShare.objects.filter(doc=doc).delete() DocTag.objects.filter(doc=doc).delete() # 删除文档 doc.delete() else: return JsonResponse({'status':False,'data':'无效请求'}) return JsonResponse({'status': True, 'data': '删除完成'}) else: return JsonResponse({'status': False, 'data': '非法请求'}) # 清空回收站 elif types == 'empty': docs = Doc.objects.filter(status=3,create_user=request.user) for doc in docs: # 删除文档历史、分享、标签 DocHistory.objects.filter(doc=doc).delete() DocShare.objects.filter(doc=doc).delete() DocTag.objects.filter(doc=doc).delete() docs.delete() return JsonResponse({'status': True, 'data': '清空成功'}) # 还原回收站 elif types == 'restoreAll': Doc.objects.filter(status=3,create_user=request.user).update(status=0) return JsonResponse({'status': True, 'data': '还原成功'}) else: return JsonResponse({'status': False, 'data': '参数错误'}) except Exception as e: logger.exception("处理文档出错") return JsonResponse({'status': False, 'data': '请求出错'}) # 一键发布文档 @login_required() @require_http_methods(['POST']) def fast_publish_doc(request): doc_id = request.POST.get('doc_id',None) # 查询文档 try: doc = Doc.objects.get(id=doc_id) project = Project.objects.get(id=doc.top_doc) # 查询文档所属的文集 # 获取文档所属文集的协作信息 pro_colla = ProjectCollaborator.objects.filter(project=project, user=request.user) # if pro_colla.exists(): colla_user_role = pro_colla[0].role else: colla_user_role = 0 except ObjectDoesNotExist: return JsonResponse({'status': False, 'data': '文档不存在'}) # 判断请求者是否有权限（文档创建者、文集创建者、文集高级协作者） # 如果请求用户为文档创建者、高级权限的协作者、文集的创建者，可以删除 if (request.user == doc.create_user) or (colla_user_role == 1) or (request.user == project.create_user): try: doc.status = 1 doc.modify_time = datetime.datetime.now() doc.save() return JsonResponse({'status':True,'data':'发布成功'}) except: logger.exception("文档一键发布失败") return JsonResponse({'status':False,'data':'发布失败'}) else: return JsonResponse({'status':False,'data':'非法请求'}) # 私密文档分享 @require_http_methods(['GET','POST']) def share_doc(request): if request.method == 'GET': share_token = request.GET.get('token') # 判断是否存在分享 try: share_doc = DocShare.objects.get(token=share_token,is_enable=True) doc = share_doc.doc # 公开分享 if share_doc.share_type == 0: return render(request, 'app_doc/share/share_doc.html', locals()) # 私密分享 else: doc_id_base64 = base64.standard_b64encode(str(share_doc.doc.id).encode()) # 不存在公开分享的文档，则判断验证分享码 share_cookie_name = 'sharedoc-{}'.format(share_token) share_value = request.COOKIES.get(share_cookie_name) if share_cookie_name in request.COOKIES.keys() else 0 if share_doc.share_value == share_value: return render(request, 'app_doc/share/share_doc.html', locals()) else: return redirect('/share_doc_check/?surl={}'.format(share_token)) except ObjectDoesNotExist: return render(request,'404.html') elif request.method == 'POST': doc_id = request.POST.get('id') try: # 获取请求参数 doc = Doc.objects.get(id=doc_id,create_user=request.user) share_type = request.POST.get('share_type',0) share_value = request.POST.get('share_value',0) is_enable = request.POST.get('is_enable',True) if is_enable == 'false': is_enable = False else: is_enable = True # 生成分享文档Token share_token = hashlib.md5() share_token.update("{}_{}".format(doc_id,request.user.username).encode()) share_token = share_token.hexdigest() # 创建或更新分享信息 doc_share = DocShare.objects.update_or_create( token=share_token, defaults={'doc': doc, 'share_type': share_type, 'share_value':share_value, 'is_enable':is_enable } ) if int(share_type) == 0: data = { 'doc':share_token } else: data = { 'doc': share_token, 'share_value':share_value } return JsonResponse({'status':True,'data':data}) except ObjectDoesNotExist: return JsonResponse({'status':False,'data':'文档不存在'}) # 验证文档分享码 def share_doc_check(request): doc_token = request.GET.get('surl', '') if request.method == 'GET': if doc_token != '': doc_share = DocShare.objects.get(token=doc_token) share_cookie_name = 'sharedoc-{}'.format(doc_token) share_value = request.COOKIES.get(share_cookie_name) if share_cookie_name in request.COOKIES.keys() else 0 if doc_share.share_value == share_value: return redirect("/share_doc/?token={}".format(doc_token)) else: return render(request,'app_doc/share/share_check.html',locals()) else: return render(request,'404.html') else: # 接收参数值 share_value = request.POST.get('share_value','') # 查询数据 if DocShare.objects.filter(token=doc_token,share_type=1,share_value=share_value).exists(): obj = redirect("/share_doc/?token={}".format(doc_token)) obj.set_cookie('sharedoc-{}'.format(doc_token),share_value) return obj else: errormsg = "分享码错误" return render(request, 'app_doc/share/share_check.html', locals()) # 管理文档分享 @login_required() @require_http_methods(['GET','POST']) def manage_doc_share(request): if request.method == 'GET': return render(request, 'app_doc/manage/manage_doc_share.html', locals()) else: types = request.POST.get('type') # 请求类型 1：获取列表 2：删除 3：修改 if types == '1': page = request.POST.get('page', 1) limit = request.POST.get('limit', 10) # share_doc = DocShare.objects.filter(doc__create_user=request.user).order_by('-create_time') docshare_list = DocShare.objects.filter(doc__create_user=request.user).order_by('-create_time') paginator = Paginator(docshare_list, limit) page = request.GET.get('page', page) try: docshares = paginator.page(page) except PageNotAnInteger: docshares = paginator.page(1) except EmptyPage: docshares = paginator.page(paginator.num_pages) share_list = [] for doc in docshares: item = { 'token':doc.token, 'doc_name':doc.doc.name, 'share_type':doc.share_type, 'share_value':doc.share_value, 'share_status':doc.is_enable, 'create_time':doc.create_time } share_list.append(item) resp_data = { "code":0, "msg":"ok", "count":docshare_list.count(), "data":share_list } return JsonResponse(resp_data) # 删除 elif types == '2': range = request.POST.get("range") token = request.POST.get("token") if range == 'single': try: share = DocShare.objects.get(token=token,doc__create_user=request.user) share.delete() return JsonResponse({'status':True,'data':'ok'}) except: return JsonResponse({'status':False,'data':'无指定内容'}) elif range == "multi": tokens = token.split(",") try: share = DocShare.objects.filter(token__in=tokens,doc__create_user=request.user) share.delete() return JsonResponse({'status':True,'data':'ok'}) except: return JsonResponse({'status':False,'data':'无指定内容'}) else: return JsonResponse({'status':False,'data':'类型错误'}) # 修改 elif types == '3': token = request.POST.get("token",'') name = request.POST.get('key','') value = request.POST.get('value','') # 修改分享状态 if name == 'share_status': is_enable = True if value == 'true' else False DocShare.objects.filter(token=token).update(is_enable=is_enable) # 修改分享类型 elif name == 'share_type': share_type = 0 if value == '0' else 1 DocShare.objects.filter(token=token).update(share_type=share_type) else: return JsonResponse({'status':False,'data':'参数错误'}) return JsonResponse({'status':True,'data':'ok'}) # 创建文档模板 @login_required() @require_http_methods(['GET',"POST"]) def create_doctemp(request): if request.method == 'GET': editor_type = "新建文档模板" # 获取用户的编辑器模式 try: user_opt = UserOptions.objects.get(user=request.user) editor_mode = user_opt.editor_mode except ObjectDoesNotExist: editor_mode = 1 doctemps = DocTemp.objects.filter(create_user=request.user) return render(request,'app_doc/editor/create_doctemp.html',locals()) elif request.method == 'POST': try: name = request.POST.get('name','') content = request.POST.get('content','') if name != '': doctemp = DocTemp.objects.create( name = name, content = content, create_user=request.user ) doctemp.save() return JsonResponse({'status':True,'data':doctemp.id}) else: return JsonResponse({'status':False,'data':'模板标题不能为空'}) except Exception as e: logger.exception("创建文档模板出错") return JsonResponse({'status':False,'data':'请求出错'}) # 修改文档模板 @login_required() @require_http_methods(['GET',"POST"]) def modify_doctemp(request,doctemp_id): if request.method == 'GET': editor_type = '修改文档模板' try: doctemp = DocTemp.objects.get(id=doctemp_id) if request.user.id == doctemp.create_user.id: # 获取用户的编辑器模式 try: user_opt = UserOptions.objects.get(user=request.user) editor_mode = user_opt.editor_mode except ObjectDoesNotExist: editor_mode = 1 doctemps = DocTemp.objects.filter(create_user=request.user) return render(request,'app_doc/editor/modify_doctemp.html',locals()) else: return HttpResponse('非法请求') except Exception as e: logger.exception("访问文档模板修改页面出错") return render(request, '404.html') elif request.method == 'POST': try: doctemp_id = request.POST.get('doctemp_id','') name = request.POST.get('name','') content = request.POST.get('content','') if doctemp_id != '' and name !='': doctemp = DocTemp.objects.get(id=doctemp_id) if request.user.id == doctemp.create_user.id: doctemp.name = name doctemp.content = content doctemp.save() return JsonResponse({'status':True,'data':'修改成功'}) else: return JsonResponse({'status':False,'data':'非法操作'}) else: return JsonResponse({'status':False,'data':'参数错误'}) except Exception as e: logger.exception("修改文档模板出错") return JsonResponse({'status':False,'data':'请求出错'}) # 删除文档模板 @login_required() def del_doctemp(request): try: doctemp_id = request.POST.get('doctemp_id','') if doctemp_id != '': doctemp = DocTemp.objects.get(id=doctemp_id) if request.user.id == doctemp.create_user.id: doctemp.delete() return JsonResponse({'status':True,'data':'删除完成'}) else: return JsonResponse({'status':False,'data':'非法请求'}) else: return JsonResponse({'status': False, 'data': '参数错误'}) except Exception as e: logger.exception("删除文档模板出错") return JsonResponse({'status':False,'data':'请求出错'}) # 管理文档模板 @login_required() @require_http_methods(['GET']) def manage_doctemp(request): try: search_kw = request.GET.get('kw', None) if search_kw: doctemp_list = DocTemp.objects.filter( create_user=request.user, content__icontains=search_kw ).order_by('-modify_time') paginator = Paginator(doctemp_list, 10) page = request.GET.get('page', 1) try: doctemps = paginator.page(page) except PageNotAnInteger: doctemps = paginator.page(1) except EmptyPage: doctemps = paginator.page(paginator.num_pages) doctemps.kw = search_kw else: doctemp_list = DocTemp.objects.filter(create_user=request.user).order_by('-modify_time') paginator = Paginator(doctemp_list, 10) page = request.GET.get('page', 1) try: doctemps = paginator.page(page) except PageNotAnInteger: doctemps = paginator.page(1) except EmptyPage: doctemps = paginator.page(paginator.num_pages) return render(request, 'app_doc/manage/manage_doctemp.html', locals()) except Exception as e: logger.exception("管理文档模板页面访问出错") return render(request, '404.html') # 获取指定文档模板 @login_required() @require_http_methods(["POST"]) def get_doctemp(request): try: doctemp_id = request.POST.get('doctemp_id','') if doctemp_id != '': content = DocTemp.objects.get(id=int(doctemp_id)).serializable_value('content') return JsonResponse({'status':True,'data':content}) else: return JsonResponse({'status':False,'data':'参数错误'}) except Exception as e: logger.exception("获取指定文档模板出错") return JsonResponse({'status':False,'data':'请求出错'}) # 获取指定文集的所有文档 @require_http_methods(["POST"]) @logger.catch() def get_pro_doc(request): pro_id = request.POST.get('pro_id','') if pro_id != '': # 获取文集所有文档的id、name和parent_doc3个字段 doc_list = Doc.objects.filter(top_doc=int(pro_id),status=1).values_list('id','name','parent_doc').order_by('parent_doc') item_list = [] # 遍历文档 for doc in doc_list: # 如果文档为顶级文档 if doc[2] == 0: # 将其数据添加到列表中 item = [ doc[0],doc[1],doc[2],'' ] item_list.append(item) # 如果文档不是顶级文档 else: # 查询文档的上级文档 try: parent = Doc.objects.get(id=doc[2]) except ObjectDoesNotExist: return JsonResponse({'status':False,'data':'文档id不存在'}) # 如果文档上级文档的上级是顶级文档，那么将其添加到列表 if parent.parent_doc == 0: # 只要二级目录 item = [ doc[0],doc[1],doc[2],parent.name+' --> ' ] item_list.append(item) return JsonResponse({'status':True,'data':list(item_list)}) else: return JsonResponse({'status':False,'data':'参数错误'}) # 获取指定文集的文档树数据 @require_http_methods(['POST']) @logger.catch() def get_pro_doc_tree(request): pro_id = request.POST.get('pro_id', None) if pro_id: # 查询存在上级文档的文档 parent_id_list = Doc.objects.filter(top_doc=pro_id,status=1).exclude(parent_doc=0).values_list('parent_doc',flat=True) # 获取存在上级文档的上级文档ID # print(parent_id_list) doc_list = [] # 获取一级文档 top_docs = Doc.objects.filter(top_doc=pro_id,parent_doc=0,status=1).values('id','name','modify_time').order_by('sort') # 遍历一级文档 for doc in top_docs: top_item = { 'id':doc['id'], 'field':doc['name'], 'title':doc['name'], 'modify_time':doc['modify_time'], 'spread':True, 'level':1 } # 如果一级文档存在下级文档，查询其二级文档 if doc['id'] in parent_id_list: # 获取二级文档 sec_docs = Doc.objects.filter(top_doc=pro_id,parent_doc=doc['id'],status=1).values('id','name','modify_time').order_by('sort') top_item['children'] = [] for doc in sec_docs: sec_item = { 'id': doc['id'], 'field': doc['name'], 'title': doc['name'], 'modify_time': doc['modify_time'], 'level':2 } # 如果二级文档存在下级文档，查询第三级文档 if doc['id'] in parent_id_list: # 获取三级文档 thr_docs = Doc.objects.filter(top_doc=pro_id,parent_doc=doc['id'],status=1).values('id','name','modify_time').order_by('sort') sec_item['children'] = [] for doc in thr_docs: item = { 'id': doc['id'], 'field': doc['name'], 'title': doc['name'], 'modify_time': doc['modify_time'], 'level': 3 } sec_item['children'].append(item) top_item['children'].append(sec_item) else: top_item['children'].append(sec_item) doc_list.append(top_item) # 如果一级文档没有下级文档，直接保存 else: doc_list.append(top_item) return JsonResponse({'status':True,'data':doc_list}) else: return JsonResponse({'status':False,'data':'参数错误'}) # 404页面 def handle_404(request): return render(request,'404.html') # 导出文集MD文件 @login_required() @require_http_methods(["POST"]) def report_md(request): pro_id = request.POST.get('project_id','') user = request.user try: project = Project.objects.get(id=int(pro_id)) if project.create_user == user: project_md = ReportMD( project_id=int(pro_id) ) md_file_path = project_md.work() # 生成并获取MD文件压缩包绝对路径 md_file_filename = os.path.split(md_file_path)[-1] # 提取文件名 md_file = "/media/reportmd_temp/"+ md_file_filename # 拼接相对链接 return JsonResponse({'status':True,'data':md_file}) else: return JsonResponse({'status':False,'data':'无权限'}) except Exception as e: logger.exception("导出文集MD文件出错") return JsonResponse({'status':False,'data':'文集不存在'}) # 生成文集文件 - 个人中心 - 文集管理 @login_required() @require_http_methods(["POST"]) def genera_project_file(request): report_type = request.POST.get('types',None) # 获取前端传入到导出文件类型参数 # 导出EPUB文件 pro_id = request.POST.get('pro_id') try: project = Project.objects.get(id=int(pro_id)) # 获取文集的协作用户信息 if request.user.is_authenticated: colla_user = ProjectCollaborator.objects.filter(project=project, user=request.user) if colla_user.exists(): colla_user_role = colla_user[0].role colla_user = colla_user.count() else: colla_user = colla_user.count() else: colla_user = 0 # 公开的文集 - 可以直接导出 if project.role == 0: allow_export = True # 私密文集 - 非创建者和协作者不可导出 elif (project.role == 1): if (request.user != project.create_user) and (colla_user == 0): allow_export = False else: allow_export = True # 指定用户可见文集 - 指定用户、文集创建者和协作者可导出 elif project.role == 2: user_list = project.role_value if request.user.is_authenticated: # 认证用户判断是否在许可用户列表中 if (request.user.username not in user_list) and \ (request.user != project.create_user) and \ (colla_user == 0): # 访问者不在指定用户之中，也不是协作者 allow_export = False else: allow_export = True else: # 游客直接返回404 allow_export = False # 访问码可见文集 - 文集创建者、协作者和通过验证即可导出 elif project.role == 3: # 浏览用户不为创建者和协作者 - 需要访问码 if (request.user != project.create_user) and (colla_user == 0): viewcode = project.role_value viewcode_name = 'viewcode-{}'.format(project.id) r_viewcode = request.COOKIES[ viewcode_name] if viewcode_name in request.COOKIES.keys() else 0 # 从cookie中获取访问码 if viewcode != r_viewcode: # cookie中的访问码不等于文集访问码，不可导出 allow_export = False else: allow_export = True else: allow_export = True else: allow_export = False # 允许被导出 if allow_export: # 导出EPUB if report_type in ['epub']: try: report_project = ReportEPUB( project_id=project.id ).work() # print(report_project) report_file_path = report_project.split('media', maxsplit=1)[-1] # 导出文件的路径 epub_file = '/media' + report_file_path + '.epub' # 文件相对路径 # 查询文集是否存在导出文件 report_cnt = ProjectReportFile.objects.filter(project=project,file_type='epub') # 存在文件删除 if report_cnt.count() != 0: for r in report_cnt: is_exist = os.path.exists(settings.BASE_DIR + r.file_path) if is_exist: os.remove(settings.BASE_DIR + r.file_path) report_cnt.delete() # 删除数据库记录 # 创建数据库记录 ProjectReportFile.objects.create( project=project, file_type='epub', file_name=epub_file, file_path=epub_file ) return JsonResponse({'status': True, 'data': epub_file}) except Exception as e: logger.exception("生成EPUB出错") return JsonResponse({'status': False, 'data': '生成出错'}) # 导出PDF elif report_type in ['pdf']: try: report_project = ReportPDF( project_id=project.id ).work() if report_project is False: return JsonResponse({'status':False,'data':'生成出错'}) report_file_path = report_project.split('media', maxsplit=1)[-1] # 导出文件的路径 pdf_file = '/media' + report_file_path # 文件相对路径 # 查询文集是否存在导出文件 report_cnt = ProjectReportFile.objects.filter(project=project, file_type='pdf') # 存在文件删除 if report_cnt.count() != 0: for r in report_cnt: is_exist = os.path.exists(settings.BASE_DIR + r.file_path) if is_exist: os.remove(settings.BASE_DIR + r.file_path) report_cnt.delete() # 删除数据库记录 # 创建数据库记录 ProjectReportFile.objects.create( project=project, file_type='pdf', file_name=pdf_file, file_path=pdf_file ) return JsonResponse({'status': True, 'data': pdf_file}) except Exception as e: return JsonResponse({'status': False, 'data': '生成出错'}) else: return JsonResponse({'status': False, 'data': '不支持的类型'}) # 不允许被导出 else: return JsonResponse({'status':False,'data':'无权限导出'}) except ObjectDoesNotExist: return JsonResponse({'status':False,'data':'文集不存在'}) except Exception as e: logger.exception("生成文集文件出错") return JsonResponse({'status':False,'data':'系统异常'}) # 获取文集前台导出文件 @allow_report_file @require_http_methods(["POST"]) def report_file(request): report_type = request.POST.get('types',None) # 获取前端传入到导出文件类型参数 pro_id = request.POST.get('pro_id') try: project = Project.objects.get(id=int(pro_id)) # 获取文集的协作用户信息 if request.user.is_authenticated: colla_user = ProjectCollaborator.objects.filter(project=project, user=request.user) if colla_user.exists(): colla_user_role = colla_user[0].role colla_user = colla_user.count() else: colla_user = colla_user.count() else: colla_user = 0 # 公开的文集 - 可以直接导出 if project.role == 0: allow_export = True # 私密文集 - 非创建者和协作者不可导出 elif (project.role == 1): if (request.user != project.create_user) and (colla_user == 0): allow_export = False else: allow_export = True # 指定用户可见文集 - 指定用户、文集创建者和协作者可导出 elif project.role == 2: user_list = project.role_value if request.user.is_authenticated: # 认证用户判断是否在许可用户列表中 if (request.user.username not in user_list) and \ (request.user != project.create_user) and \ (colla_user == 0): # 访问者不在指定用户之中，也不是协作者 allow_export = False else: allow_export = True else: # 游客直接返回404 allow_export = False # 访问码可见文集 - 文集创建者、协作者和通过验证即可导出 elif project.role == 3: # 浏览用户不为创建者和协作者 - 需要访问码 if (request.user != project.create_user) and (colla_user == 0): viewcode = project.role_value viewcode_name = 'viewcode-{}'.format(project.id) r_viewcode = request.COOKIES[ viewcode_name] if viewcode_name in request.COOKIES.keys() else 0 # 从cookie中获取访问码 if viewcode != r_viewcode: # cookie中的访问码不等于文集访问码，不可导出 allow_export = False else: allow_export = True else: allow_export = True else: allow_export = False # return JsonResponse({'status':False,'data':'不存在的文集权限'}) if allow_export: # 导出EPUB文件 if report_type in ['epub']: try: try: report_project = ProjectReportFile.objects.get(project=project,file_type='epub') except ObjectDoesNotExist: return JsonResponse({'status':False,'data':'无可用文件,请联系文集创建者'}) # print(report_project) return JsonResponse({'status': True, 'data': report_project.file_path}) except Exception as e: return JsonResponse({'status': False, 'data': '导出出错'}) # 导出PDF elif report_type in ['pdf']: try: try: report_project = ProjectReportFile.objects.get(project=project,file_type='pdf') except ObjectDoesNotExist: return JsonResponse({'status':False,'data':'无可用文件,请联系文集创建者'}) # print(report_project) return JsonResponse({'status': True, 'data': report_project.file_path}) except Exception as e: return JsonResponse({'status': False, 'data': '导出出错'}) else: return JsonResponse({'status': False, 'data': '不支持的类型'}) else: return JsonResponse({'status':False,'data':'无权限导出'}) except ObjectDoesNotExist: return JsonResponse({'status':False,'data':'文集不存在'}) except Exception as e: logger.exception("获取文集前台导出文件出错") return JsonResponse({'status':False,'data':'系统异常'}) # 图片素材管理 @login_required() @require_http_methods(['GET',"POST"]) def manage_image(request): # 获取图片 if request.method == 'GET': try: groups = ImageGroup.objects.filter(user=request.user) # 获取所有分组 all_img_cnt = Image.objects.filter(user=request.user).count() no_group_cnt = Image.objects.filter(user=request.user,group_id=None).count() # 获取所有未分组的图片数量 g_id = int(request.GET.get('group', 0)) # 图片分组id if int(g_id) == 0: image_list = Image.objects.filter(user=request.user).order_by('-create_time') # 查询所有图片 elif int(g_id) == -1: image_list = Image.objects.filter(user=request.user,group_id=None).order_by('-create_time') # 查询未分组的图片 else: image_list = Image.objects.filter(user=request.user,group_id=g_id).order_by('-create_time') # 查询指定分组的图片 paginator = Paginator(image_list, 18) page = request.GET.get('page', 1) try: images = paginator.page(page) except PageNotAnInteger: images = paginator.page(1) except EmptyPage: images = paginator.page(paginator.num_pages) images.group = g_id return render(request,'app_doc/manage/manage_image.html',locals()) except: logger.exception("图片素材管理出错") return render(request,'404.html') elif request.method == 'POST': try: img_id = request.POST.get('img_id','') types = request.POST.get('types','') # 操作类型：0表示删除，1表示修改，2表示获取 # 删除图片 if int(types) == 0: img = Image.objects.get(id=img_id) if img.user != request.user: return JsonResponse({'status': False, 'data': '未授权请求'}) file_path = settings.BASE_DIR+img.file_path is_exist = os.path.exists(file_path) if is_exist: os.remove(file_path) img.delete() # 删除记录 return JsonResponse({'status':True,'data':'删除完成'}) # 移动图片分组 elif int(types) == 1: group_id = request.POST.get('group_id',None) if group_id is None: Image.objects.filter(id=img_id,user=request.user).update(group_id=None) else: group = ImageGroup.objects.get(id=group_id,user=request.user) Image.objects.filter(id=img_id,user=request.user).update(group_id=group) return JsonResponse({'status':True,'data':'移动完成'}) # 获取图片 elif int(types) == 2: group_id = request.POST.get('group_id', None) # 接受分组ID参数 if group_id is None: # return JsonResponse({'status':False,'data':'参数错误'}) elif int(group_id) == 0: imgs = Image.objects.filter(user=request.user).order_by('-create_time') elif int(group_id) == -1: imgs = Image.objects.filter(user=request.user,group_id=None).order_by('-create_time') else: imgs = Image.objects.filter(user=request.user,group_id=group_id).order_by('-create_time') img_list = [] for img in imgs: item = { 'path':img.file_path, 'name':img.file_name, } img_list.append(item) return JsonResponse({'status':True,'data':img_list}) else: return JsonResponse({'status':False,'data':'非法参数'}) except ObjectDoesNotExist: return JsonResponse({'status':False,'data':'图片不存在'}) except: logger.exception("操作图片素材出错") return JsonResponse({'status':False,'data':'程序异常'}) # 图片分组管理 @login_required() @require_http_methods(['GET',"POST"]) @logger.catch() def manage_img_group(request): if request.method == 'GET': groups = ImageGroup.objects.filter(user=request.user) return render(request,'app_doc/manage/manage_image_group.html',locals()) # 操作分组 elif request.method == 'POST': types = request.POST.get('types',None) # 请求类型，0表示创建分组，1表示修改分组，2表示删除分组，3表示获取分组 # 创建分组 if int(types) == 0: group_name = request.POST.get('group_name', '') if group_name not in ['','默认分组','未分组']: ImageGroup.objects.get_or_create( user = request.user, group_name = group_name ) return JsonResponse({'status':True,'data':'ok'}) else: return JsonResponse({'status':False,'data':'名称无效'}) # 修改分组 elif int(types) == 1: group_name = request.POST.get("group_name",'') if group_name not in ['','默认分组','未分组']: group_id = request.POST.get('group_id', '') ImageGroup.objects.filter(id=group_id,user=request.user).update(group_name=group_name) return JsonResponse({'status':True,'data':'修改成功'}) else: return JsonResponse({'status':False,'data':'名称无效'}) # 删除分组 elif int(types) == 2: try: group_id = request.POST.get('group_id','') group = ImageGroup.objects.get(id=group_id,user=request.user) # 查询分组 images = Image.objects.filter(group_id=group_id,user=request.user).update(group_id=None) # 移动图片到未分组 group.delete() # 删除分组 return JsonResponse({'status':True,'data':'删除完成'}) except: logger.exception("删除图片分组出错") return JsonResponse({'status':False,'data':'删除错误'}) # 获取分组 elif int(types) == 3: try: group_list = [] all_cnt = Image.objects.filter(user=request.user).count() non_group_cnt = Image.objects.filter(group_id=None,user=request.user).count() group_list.append({'group_name':'全部图片','group_cnt':all_cnt,'group_id':0}) group_list.append({'group_name':'未分组','group_cnt':non_group_cnt,'group_id':-1}) groups = ImageGroup.objects.filter(user=request.user) # 查询所有分组 for group in groups: group_cnt = Image.objects.filter(group_id=group).count() item = { 'group_id':group.id, 'group_name':group.group_name, 'group_cnt':group_cnt } group_list.append(item) return JsonResponse({'status':True,'data':group_list}) except: logger.exception("获取图片分组出错") return JsonResponse({'status':False,'data':'出现错误'}) # 附件管理 @login_required() @csrf_exempt @require_http_methods(['GET',"POST"]) def manage_attachment(request): # 文件大小 字节转换 def sizeFormat(size, is_disk=False, precision=2): ''' size format for human. byte ---- (B) kilobyte ---- (KB) megabyte ---- (MB) gigabyte ---- (GB) terabyte ---- (TB) petabyte ---- (PB) exabyte ---- (EB) zettabyte ---- (ZB) yottabyte ---- (YB) ''' formats = ['KB', 'MB', 'GB', 'TB', 'PB', 'EB', 'ZB', 'YB'] unit = 1000.0 if is_disk else 1024.0 if not (isinstance(size, float) or isinstance(size, int)): raise TypeError('a float number or an integer number is required!') if size < 0: raise ValueError('number must be non-negative') for i in formats: size /= unit if size < unit: r = '{}{}'.format(round(size, precision),i) return r if request.method == 'GET': try: search_kw = request.GET.get('kw', None) # 搜索附件 if search_kw: attachment_list = Attachment.objects.filter( user=request.user, file_name__icontains=search_kw ).order_by('-create_time') paginator = Paginator(attachment_list, 15) page = request.GET.get('page', 1) try: attachments = paginator.page(page) except PageNotAnInteger: attachments = paginator.page(1) except EmptyPage: attachments = paginator.page(paginator.num_pages) attachments.kw = search_kw # 所有附件 else: attachment_list = Attachment.objects.filter(user=request.user).order_by('-create_time') paginator = Paginator(attachment_list, 15) page = request.GET.get('page', 1) try: attachments = paginator.page(page) except PageNotAnInteger: attachments = paginator.page(1) except EmptyPage: attachments = paginator.page(paginator.num_pages) return render(request, 'app_doc/manage/manage_attachment.html', locals()) except Exception as e: logger.exception("附件管理访问出错") return render(request,'404.html') elif request.method == 'POST': # types参数，0表示上传、1表示删除、2表示获取附件列表 types = request.POST.get('types','') if types in ['0',0]: attachment = request.FILES.get('attachment_upload',None) if attachment: attachment_name = attachment.name # 获取附件文件名 attachment_size = sizeFormat(attachment.size) # 获取附件文件大小 # 限制附件大小 # 获取系统设置的附件文件大小，如果不存在，默认50MB try: allow_attachment_size = SysSetting.objects.get(types='doc',name='attachment_size') allow_attach_size = int(allow_attachment_size.value) * 1048576 except Exception as e: # print(repr(e)) allow_attach_size = 52428800 if attachment.size > allow_attach_size: return JsonResponse({'status':False,'data':'文件大小超出限制'}) # 限制附件格式 # 获取系统设置允许的附件格式，如果不存在，默认仅允许zip格式文件 try: attachment_suffix_list = SysSetting.objects.get(types='doc',name='attachment_suffix') attachment_suffix_list = attachment_suffix_list.value.split(',') if attachment_suffix_list == ['']: attachment_suffix_list = ['zip'] except ObjectDoesNotExist: attachment_suffix_list = ['zip'] allow_attachment = False for suffix in attachment_suffix_list: if attachment_name.split('.')[-1] in attachment_suffix_list: allow_attachment = True if allow_attachment: a = Attachment.objects.create( file_name = attachment_name, file_size = attachment_size, file_path = attachment, user = request.user ) return JsonResponse({'status':True,'data':{'name':attachment_name,'url':a.file_path.name}}) else: return JsonResponse({'status':False,'data':'不支持的格式'}) else: return JsonResponse({'status':False,'data':'无效文件'}) elif types in ['1',1]: attach_id = request.POST.get('attach_id','') attachment = Attachment.objects.filter(id=attach_id,user=request.user) # 查询附件 for a in attachment: # 遍历附件 a.file_path.delete() # 删除文件 attachment.delete() # 删除数据库记录 return JsonResponse({'status':True,'data':'删除成功'}) elif types in [2,'2']: attachment_list = [] attachments = Attachment.objects.filter(user=request.user).order_by('-create_time') for a in attachments: item = { 'filename':a.file_name, 'filesize':a.file_size, 'filepath':a.file_path.name, 'filetime':a.create_time } attachment_list.append(item) return JsonResponse({'status':True,'data':attachment_list}) else: return JsonResponse({'status':False,'data':'无效参数'}) # 搜索 def search(request): kw = request.GET.get('kw', None) search_type = request.GET.get('type', 'doc') # 搜索类型，默认文档doc date_type = request.GET.get('d_type', 'recent') date_range = request.GET.get('d_range', 'all') # 时间范围，默认不限，all project_range = request.GET.get('p_range', 0) # 文集范围，默认不限，all # 处理时间范围 if date_type == 'recent': if date_range == 'recent1': # 最近1天 start_date = datetime.datetime.now() - datetime.timedelta(days=1) elif date_range == 'recent7': # 最近7天 start_date = datetime.datetime.now() - datetime.timedelta(days=7) elif date_range == 'recent30': # 最近30天 start_date = datetime.datetime.now() - datetime.timedelta(days=30) elif date_range == 'recent365': # 最近一年 start_date = datetime.datetime.now() - datetime.timedelta(days=365) else: start_date = datetime.datetime.strptime('1900-01-01', '%Y-%m-%d') end_date = datetime.datetime.now() elif date_type == 'day': try: date_list = date_range.split('|') start_date = datetime.datetime.strptime(date_list[0], '%Y-%m-%d') end_date = datetime.datetime.strptime(date_list[1], '%Y-%m-%d') except: start_date = datetime.datetime.now() - datetime.timedelta(days=1) end_date = datetime.datetime.now() # 是否时间筛选 if date_range == 'all': is_date_range = False else: is_date_range = True # 是否认证 if request.user.is_authenticated: is_auth = True else: is_auth = False # 存在搜索关键词 if kw: # 搜索文档 if search_type == 'doc': if is_auth: colla_list = [i.project.id for i in ProjectCollaborator.objects.filter(user=request.user)] # 用户的协作文集 open_list = [i.id for i in Project.objects.filter( Q(role=0) | Q(create_user=request.user) )] # 公开文集 view_list = list(set(open_list).union(set(colla_list))) # 合并上述两个文集ID列表 data_list = Doc.objects.filter( Q(top_doc__in=view_list), # 包含用户可浏览到的文集 Q(create_time__gte=start_date, create_time__lte=end_date), # 筛选创建时间 Q(name__icontains=kw) | Q(content__icontains=kw) | Q(pre_content__icontains=kw) # 筛选文档标题和内容中包含搜索词 ).order_by('-create_time') else: view_list = [i.id for i in Project.objects.filter(role=0)] data_list = Doc.objects.filter( Q(top_doc__in=view_list), Q(create_time__gte=start_date, create_time__lte=end_date), # 筛选创建时间 Q(name__icontains=kw) | Q(content__icontains=kw) | Q(pre_content__icontains=kw) # 筛选文档标题和内容中包含搜索词 ).order_by('-create_time') # 搜索文集 elif search_type == 'pro': # 认证用户 if is_auth: colla_list = [i.project.id for i in ProjectCollaborator.objects.filter(user=request.user)] # 用户的协作文集 # 查询所有可显示的文集 data_list = Project.objects.filter( Q(role=0) | \ Q(role=2, role_value__contains=str(request.user.username)) | \ Q(create_user=request.user) | \ Q(id__in=colla_list), Q(create_time__gte=start_date, create_time__lte=end_date), # 筛选创建时间 Q(name__icontains=kw) | Q(intro__icontains=kw) # 筛选文集名称和简介包含搜索词 ).order_by('-create_time') # 游客 else: data_list = Project.objects.filter( Q(role=0), Q(name__icontains=kw) | Q(intro__icontains=kw), Q(create_time__gte=start_date, create_time__lte=end_date), # 筛选创建时间 ).order_by("-create_time") # 搜索标签 elif search_type == 'tag': # 认证用户 if is_auth: colla_list = [i.project.id for i in ProjectCollaborator.objects.filter(user=request.user)] # 用户的协作文集 open_list = [i.id for i in Project.objects.filter( Q(role=0) | Q(create_user=request.user) )] # 公开文集和自己创建的文集 view_list = list(set(open_list).union(set(colla_list))) # 合并上述两个文集ID列表 tag_list = Tag.objects.filter(name__icontains=kw) # 查询符合条件的标签 tag_doc_list = [i.doc.id for i in DocTag.objects.filter(tag__in=tag_list)] # 获取符合条件的标签文档 data_list = Doc.objects.filter( Q(top_doc__in=view_list), # 包含用户可浏览到的文集 Q(id__in=tag_doc_list), # 包含符合条件标签的文档ID列表 Q(create_time__gte=start_date, create_time__lte=end_date), # 筛选创建时间 ).order_by('-create_time') # 游客 else: open_list = [i.id for i in Project.objects.filter(Q(role=0))] # 公开文集 view_list = list(set(open_list)) tag_list = Tag.objects.filter(name__icontains=kw) # 查询符合条件的标签 tag_doc_list = [i.doc.id for i in DocTag.objects.filter(tag__in=tag_list)] # 获取符合条件的标签文档 data_list = Doc.objects.filter( Q(top_doc__in=view_list), # 包含用户可浏览到的文集 Q(id__in=tag_doc_list), # 包含符合条件标签的文档ID列表 Q(create_time__gte=start_date, create_time__lte=end_date), # 筛选创建时间 ).order_by('-create_time') else: return render(request, 'app_doc/search.html') # 分页处理 paginator = Paginator(data_list, 12) page = request.GET.get('page', 1) try: datas = paginator.page(page) except PageNotAnInteger: datas = paginator.page(1) except EmptyPage: datas = paginator.page(paginator.num_pages) return render(request, 'app_doc/search_result.html', locals()) # 否则跳转到搜索首页 else: return render(request,'app_doc/search.html') # 文档Markdown文件下载 @require_http_methods(['GET',"POST"]) def download_doc_md(request,doc_id): if request.user.is_authenticated: if request.user.is_superuser: try: doc = Doc.objects.get(id=doc_id) except ObjectDoesNotExist: return JsonResponse({'status':False,'data':'文档不存在'}) else: try: doc = Doc.objects.get(id=doc_id,create_user = request.user) except ObjectDoesNotExist: return JsonResponse({'status':False,'data':'文档不存在'}) else: return render(request,'404.html') response = HttpResponse(content_type='text/plain') response['Content-Disposition'] = 'attachment; filename={}.md'.format(doc.name) response.write(doc.pre_content) return response # 个人中心 - 概览 @login_required() @require_http_methods(['GET','POST']) def manage_overview(request): if request.method == 'GET': pro_list = Project.objects.filter(create_user=request.user).order_by('-create_time') colla_pro_cnt = ProjectCollaborator.objects.filter(user=request.user).count() pro_cnt = pro_list.count() + colla_pro_cnt # 文集总数 doc_cnt = Doc.objects.filter(create_user=request.user).count() # 文档总数 total_tag_cnt = Tag.objects.filter(create_user=request.user).count() img_cnt = Image.objects.filter(user=request.user).count() attachment_cnt = Attachment.objects.filter(user=request.user).count() doc_active_list = Doc.objects.filter(create_user=request.user).order_by('-modify_time')[:5] return render(request,'app_doc/manage/manage_overview.html',locals()) else: pass # 个人中心 - 文档标签 @login_required() @require_http_methods(['GET','POST']) def manage_doc_tag(request): if request.method == 'GET': tags = Tag.objects.filter(create_user=request.user) return render(request,'app_doc/manage/manage_doc_tag.html',locals()) # 操作标签 elif request.method == 'POST': types = request.POST.get('types', None) # 请求类型，0表示创建标签，1表示修改标签，2表示删除标签，3表示获取标签 # 创建标签 if int(types) == 0: tag_name = request.POST.get('tag_name', '') if tag_name != '': Tag.objects.create( user=request.user, name=tag_name ) return JsonResponse({'status': True, 'data': 'ok'}) else: return JsonResponse({'status': False, 'data': '名称无效'}) # 修改标签 elif int(types) == 1: try: tag_name = request.POST.get('tag_name', '') if tag_name != "": tag_id = request.POST.get('tag_id', '') if tag_id != "": print(tag_id,tag_name) Tag.objects.filter(id=tag_id, create_user=request.user).update(name=tag_name) return JsonResponse({'status': True, 'data': '修改成功'}) else: return JsonResponse({'status': False, 'data': '标签ID无效'}) else: return JsonResponse({'status': False, 'data': '名称无效'}) except Exception as e: logger.exception("修改异常") return JsonResponse({'status': False, 'data': '异常错误'}) # 删除标签 elif int(types) == 2: try: tag_id = request.POST.get('tag_id', '') tag = Tag.objects.get(id=tag_id, create_user=request.user) # 查询分组 tag.delete() # 删除标签 return JsonResponse({'status': True, 'data': '删除完成'}) except: logger.exception("删除标签出错") return JsonResponse({'status': False, 'data': '删除错误'}) # 获取标签 elif int(types) == 3: try: tag_list = [] return JsonResponse({'status': True, 'data': tag_list}) except: logger.exception("获取文档标签出错") return JsonResponse({'status': False, 'data': '出现错误'}) # 标签文档关系页 def tag_docs(request,tag_id): # 获取标签 try: # 颜色列表 color_list = ['#37a2da', '#32c5e9', '#67e0e3', '#9fe6b8', '#ffdb5c', '#ff9f7f', '#fb7293', '#e062ae', '#e062ae'] # 获取标签信息 tag = Tag.objects.get(id=int(tag_id)) # 获取标签的文档信息 # 如果访问者已经登录 if request.user.is_authenticated: # 判断是否为标签的创建者 if request.user == tag.create_user: # 获取标签的所有文档 view_list = [i.id for i in Project.objects.filter(create_user=request.user)] docs = DocTag.objects.filter(tag=tag,doc__status=1) else: # 获取有权限的文档 colla_list = [i.project.id for i in ProjectCollaborator.objects.filter(user=request.user)] # 用户的协作文集 open_list = [i.id for i in Project.objects.filter( Q(role=0) | Q(create_user=tag.create_user) )] # 公开文集 view_list = list(set(open_list).union(set(colla_list))) # 合并上述两个文集ID列表 # 查询可浏览文集的文档 doc_list = [i for i in Doc.objects.filter(top_doc__in=view_list,status=1)] # 筛选可浏览的文档的标签文档 docs = DocTag.objects.filter(tag=tag,doc__in=doc_list) else: # 查询标签创建者的公开文集 open_list = [i.id for i in Project.objects.filter( role=0,create_user=tag.create_user )] view_list = open_list doc_list = [i for i in Doc.objects.filter(top_doc__in=view_list,status=1)] docs = DocTag.objects.filter(tag=tag,doc__in=doc_list) # 获取文档的其他标签信息 current_link_list = [] # 文档的所有标签ID列表 for doc in docs: other_tags = [str(i.tag.id) for i in DocTag.objects.filter(~Q(tag=tag), doc=doc.doc)] current_link_list.extend(other_tags) # 标签的节点列表 tag_nodes_list = [ # {'id':str(tag.id),'name':tag.name,'symbolSize':50,'value':docs.count(),'itemStyle':{'color':random.choice(color_list)}} ] # 标签的关系列表 tag_links_list = [] # 标签分类列表 tag_cate = [] # 添加用户创建的所有标签到节点列表 for t in Tag.objects.filter(create_user=tag.create_user): tag_cate.append({'name':t.name}) if t.name == tag.name: item = { 'id': str(t.id), 'name': t.name, 'symbolSize': 50, 'value': DocTag.objects.filter(tag=t,doc__status=1,doc__top_doc__in=view_list).count(), 'itemStyle': {'color': random.choice(color_list)} } else: item = { 'id':str(t.id), 'name':t.name, 'symbolSize':25, 'value':DocTag.objects.filter(tag=t,doc__status=1,doc__top_doc__in=view_list).count(), 'itemStyle':{'color':random.choice(color_list)} } tag_nodes_list.append(item) # 查询非主标签的关联标签 sub_tags = DocTag.objects.filter(tag=t,doc__status=1,doc__top_doc__in=view_list) # 获取包含t标签的文档 for sub_tag in sub_tags: sub_docs = DocTag.objects.filter(doc=sub_tag.doc,doc__top_doc__in=view_list) # 获取包含文档的标签 for sub_doc in sub_docs: if str(sub_tag.tag.id) != str(sub_doc.tag.id): item = { 'source': str(sub_tag.tag.id), 'target': str(sub_doc.tag.id), 'value' : sub_doc.doc.name, 'id': sub_doc.doc.id, 'pid': sub_doc.doc.top_doc, 'label':{ 'normal':{ 'show':'true', 'formatter':"{c}", 'fontsize':'10px', } } } item_1 = { 'source': str(sub_doc.tag.id), 'target': str(sub_tag.tag.id), 'value': sub_doc.doc.name, 'id':sub_doc.doc.id, 'pid': sub_doc.doc.top_doc, 'label': { 'normal': { 'show': 'true', 'formatter': "{c}", 'fontsize': '10px', } } } if item_1 not in tag_links_list: tag_links_list.append(item) return render(request, 'app_doc/tag_docs.html', locals()) except Exception as e: logger.exception("标签文档页访问异常") return render(request, '404.html') # 标签文档页 @require_http_methods(['GET']) def tag_doc(request,tag_id,doc_id): try: if tag_id != '' and doc_id != '': doc = Doc.objects.get(id=int(doc_id), status=1) # 获取文档的文集信息，以判断是否有权限访问 project = Project.objects.get(id=int(doc.top_doc)) # 获取文集的协作用户信息 if request.user.is_authenticated: colla_user = ProjectCollaborator.objects.filter(project=project,user=request.user) if colla_user.exists(): colla_user_role = colla_user[0].role colla_user = colla_user.count() else: colla_user = colla_user.count() else: colla_user = 0 # 私密文集且访问者非创建者、协作者 - 不能访问 if (project.role == 1) and (request.user != project.create_user) and (colla_user == 0): return render(request, '404.html') # 指定用户可见文集 elif project.role == 2: user_list = project.role_value if request.user.is_authenticated: # 认证用户判断是否在许可用户列表中 if (request.user.username not in user_list) and \ (request.user != project.create_user) and \ (colla_user == 0): # 访问者不在指定用户之中，也不是协作者 return render(request, '404.html') else: # 游客直接返回404 return render(request, '404.html') # 访问码可见 elif project.role == 3: # 浏览用户不为创建者和协作者 - 需要访问码 if (request.user != project.create_user) and (colla_user == 0): viewcode = project.role_value viewcode_name = 'viewcode-{}'.format(project.id) r_viewcode = request.COOKIES[ viewcode_name] if viewcode_name in request.COOKIES.keys() else 0 # 从cookie中获取访问码 if viewcode != r_viewcode: # cookie中的访问码不等于文集访问码，跳转到访问码认证界面 return redirect('/check_viewcode/?to={}'.format(request.path)) # 获取文档内容 try: # 获取标签信息 tag = Tag.objects.get(id=int(tag_id)) # 获取标签文档信息 docs = DocTag.objects.filter(tag=tag) # 获取文档的标签 doc_tags = DocTag.objects.filter(doc=doc) except ObjectDoesNotExist: return render(request, '404.html') return render(request,'app_doc/tag_doc_single.html',locals()) else: return HttpResponse('参数错误') except Exception as e: logger.exception("文集浏览出错") return render(request,'404.html') # 个人中心 - 个人设置 @login_required() def manage_self(request): if request.method == 'GET': user = User.objects.get_by_natural_key(request.user) try: user_opt = UserOptions.objects.get(user=request.user) except ObjectDoesNotExist: user_opt = [] return render(request,'app_doc/manage/manage_self.html',locals()) elif request.method == 'POST': first_name = request.POST.get('first_name','') # 昵称 email = request.POST.get('email',None) # 电子邮箱 editor_mode = request.POST.get('editor_mode',1) # 编辑器 user = User.objects.get_by_natural_key(request.user) if len(first_name) < 2 or len(first_name) > 10: return JsonResponse({'status': False, 'data': '昵称长度不得小于2位大于10位'}) if User.objects.filter(first_name=first_name).count() > 0 and user.first_name != first_name: return JsonResponse({'status':False,'data':'昵称已被使用'}) if User.objects.filter(email=email).count() > 0 and user.email != email: return JsonResponse({'status':False,'data':'电子邮箱已被使用'}) if email != '' and '@' in email: user.email = email user.first_name = first_name user.save() user_opt = UserOptions.objects.update_or_create( user = user, defaults={'editor_mode':editor_mode} ) return JsonResponse({'status':True,'data':'ok'}) else: return JsonResponse({'status':False,'data':'参数不正确'}) # 文集文档收藏 @login_required() def my_collect(request): if request.method == 'GET': pass elif request.method == 'POST': collect_type = request.POST.get('type',None) # 收藏类型 collect_id = request.POST.get('id',None) # 收藏对象ID if (collect_type is None) or (collect_id is None): return JsonResponse({'status':False,'data':'参数错误'}) else: is_collect = MyCollect.objects.filter(collect_type=collect_type,collect_id=collect_id,create_user=request.user) # 存在收藏 if is_collect.exists(): is_collect.delete() return JsonResponse({'status': True, 'data': '取消收藏成功'}) else: MyCollect.objects.create( collect_type = collect_type, collect_id = collect_id, create_user = request.user, create_time = datetime.datetime.now() ) return JsonResponse({'status':True,'data':'收藏成功'}) elif request.method == 'DELETE': pass # 收藏管理 @login_required() @require_http_methods(['GET','POST','DELETE']) @csrf_exempt def manage_collect(request): if request.method == 'GET': # 收藏文集数量 collect_project_cnt = MyCollect.objects.filter(create_user=request.user, collect_type=2).count() # 收藏文档数量 collect_doc_cnt = MyCollect.objects.filter(create_user=request.user, collect_type=1).count() # 所有收藏数量 all_cnt = collect_project_cnt + collect_doc_cnt return render(request,'app_doc/manage/manage_collect.html',locals()) elif request.method == 'POST': kw = request.POST.get('kw', '') # 搜索词 collect_type = request.POST.get('type', '') # 收藏类型 if collect_type in ['1', '2']: q_type = [int(collect_type)] else: q_type = [1, 2] page = request.POST.get('page', 1) limit = request.POST.get('limit', 10) # 没有搜索 if kw == '': collect_list = MyCollect.objects.filter( create_user=request.user, collect_type__in=q_type, ).order_by('-create_time') # 有搜索 else: collect_list = MyCollect.objects.filter( Q(content__icontains=kw) | Q(name__icontains=kw), create_user=request.user, collect_type__in=q_type ).order_by('-create_time') # 分页处理 paginator = Paginator(collect_list, limit) page = request.GET.get('page', page) try: collects = paginator.page(page) except PageNotAnInteger: collects = paginator.page(1) except EmptyPage: collects = paginator.page(paginator.num_pages) table_data = [] for collect in collects: if collect.collect_type == 1: item_doc = Doc.objects.get(id=collect.collect_id) item_id = item_doc.id item_name = item_doc.name item_project = Project.objects.get(id=item_doc.top_doc) item_project_name = item_project.name item_project_id = item_project.id else: item_project = Project.objects.get(id=collect.collect_id) item_id = item_project.id item_name = item_project.name item_project_name = '' item_project_id = '' item = { 'id': collect.id, 'item_id':item_id, 'item_name': item_name, 'type': collect.collect_type, 'item_project_id':item_project_id, 'item_project_name':item_project_name, 'create_time': collect.create_time, } table_data.append(item) resp_data = { "code": 0, "msg": "ok", "count": collect_list.count(), "data": table_data } return JsonResponse(resp_data) elif request.method == 'DELETE': try: # 获取收藏ID DELETE = QueryDict(request.body) collect_id = DELETE.get('collect_id', None) range = DELETE.get('range', 'single') if collect_id: if range == 'single': # 查询收藏 try: collect = MyCollect.objects.get(id=collect_id) except ObjectDoesNotExist: return JsonResponse({'status': False, 'data': '收藏不存在'}) # 如果请求用户为站点管理员、收藏的创建者，可以删除 if (request.user == collect.create_user) or (request.user.is_superuser): MyCollect.objects.filter(id=collect_id).delete() return JsonResponse({'status': True, 'data': '删除完成'}) else: return JsonResponse({'status': False, 'data': '非法请求'}) elif range == 'multi': collects = collect_id.split(",") try: MyCollect.objects.filter(id__in=collects, create_user=request.user).delete() return JsonResponse({'status': True, 'data': '删除完成'}) except: return JsonResponse({'status': False, 'data': '非法请求'}) else: return JsonResponse({'status': False, 'data': '类型错误'}) else: return JsonResponse({'status': False, 'data': '参数错误'}) except Exception as e: logger.exception("取消收藏出错") return JsonResponse({'status': False, 'data': '请求出错'}) # 获取当前版本 def get_version(request): try: version = settings.VERSIONS data = { 'status':True, 'data':version } except: data = { 'status':False, 'data':'异常' } return JsonResponse(data)