# coding:utf-8 # @文件: import_views.py # @创建者：州的先生 # #日期：2020/6/17 # 博客地址：zmister.com # 文集导入相关视图函数 from django.shortcuts import render,redirect from django.http.response import JsonResponse,Http404,HttpResponseNotAllowed,HttpResponse from django.http import HttpResponseForbidden from django.contrib.auth.decorators import login_required # 登录需求装饰器 from django.views.decorators.http import require_http_methods,require_GET,require_POST # 视图请求方法装饰器 from django.views.decorators.csrf import csrf_exempt from django.core.paginator import Paginator,PageNotAnInteger,EmptyPage,InvalidPage # 后端分页 from django.core.exceptions import PermissionDenied,ObjectDoesNotExist 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 from app_doc.report_utils import * from app_admin.decorators import check_headers,allow_report_file import os.path import json from app_doc.import_utils import * # 导入文集 @login_required() @require_http_methods(['GET','POST']) def import_project(request): if request.method == 'GET': return render(request,'app_doc/manage/manage_project_import.html',locals()) elif request.method == 'POST': file_type = request.POST.get('type',None) # 上传Zip压缩文件 if file_type == 'zip': import_file = request.FILES.get('import_file',None) if import_file: file_name = import_file.name # 限制文件大小在50mb以内 if import_file.size > 52428800: return JsonResponse({'status': False, 'data': '文件大小超出限制'}) # 限制文件格式为.zip if file_name.endswith('.zip'): if os.path.exists(os.path.join(settings.MEDIA_ROOT,'import_temp')) is False: os.mkdir(os.path.join(settings.MEDIA_ROOT,'import_temp')) temp_file_name = str(time.time())+'.zip' temp_file_path = os.path.join(settings.MEDIA_ROOT,'import_temp/'+temp_file_name) with open(temp_file_path,'wb+') as zip_file: for chunk in import_file: zip_file.write(chunk) if os.path.exists(temp_file_path): import_file = ImportZipProject() project = import_file.read_zip(temp_file_path,request.user) # 返回文集id或None if project: pro = Project.objects.get(id=project) docs = Doc.objects.filter(top_doc=project).values_list('id','name') # 查询存在上级文档的文档 parent_id_list = Doc.objects.filter(top_doc=project).exclude( parent_doc=0).values_list('parent_doc', flat=True) # 获取存在上级文档的上级文档ID doc_list = [] # 获取一级文档 top_docs = Doc.objects.filter( top_doc=project, parent_doc=0).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=project, parent_doc=doc['id']).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=project, parent_doc=doc['id'],).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 JsonResponse({ 'status':True, 'data':doc_list, 'project':{ 'id':project, 'name':pro.name, 'desc':pro.intro } }) else: return JsonResponse({'status':False,'data':'上传失败'}) else: return JsonResponse({'status':False,'data':'上传失败'}) else: return JsonResponse({'status':False,'data':'仅支持.zip格式'}) else: return JsonResponse({'status':False,'data':'无有效文件'}) else: return JsonResponse({'status':False,'data':'参数错误'}) # 文集文档排序 @login_required() @require_http_methods(['POST']) def project_doc_sort(request): project_id = request.POST.get('pid',None) # 文集ID title = request.POST.get('title',None) # 文集名称 desc = request.POST.get('desc',None) # 文集简介 role = request.POST.get('role',1) # 文集权限 sort_data = request.POST.get('sort_data','[]') # 文档排序列表 doc_status = request.POST.get('status',0) # 文档状态 # print(sort_data) try: sort_data = json.loads(sort_data) except Exception: return JsonResponse({'status':False,'data':'文档参数错误'}) try: Project.objects.get(id=project_id,create_user=request.user) except ObjectDoesNotExist: return JsonResponse({'status':False,'data':'没有匹配的文集'}) # 修改文集信息 Project.objects.filter(id=project_id).update( name = title, intro = desc, role = role ) # 文档排序 n = 10 # 第一级文档 for data in sort_data: Doc.objects.filter(id=data['id']).update(sort = n,status=doc_status) 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'],status=doc_status) 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'],status=doc_status) return JsonResponse({'status':True,'data':'ok'}) # 导入docx文档 @login_required() @csrf_exempt @require_POST def import_doc_docx(request): file_type = request.POST.get('type', None) editor_mode = request.POST.get('editor_mode',1) # 上传Zip压缩文件 if file_type == 'docx': import_file = request.FILES.get('import_doc_docx', None) if import_file: file_name = import_file.name # 限制文件大小在50mb以内 if import_file.size > 52428800: return JsonResponse({'status': False, 'data': '文件大小超出限制'}) # 限制文件格式为.zip if file_name.endswith('.docx'): if os.path.exists(os.path.join(settings.MEDIA_ROOT, 'import_temp')) is False: os.mkdir(os.path.join(settings.MEDIA_ROOT, 'import_temp')) temp_file_name = str(time.time()) + '.docx' temp_file_path = os.path.join(settings.MEDIA_ROOT, 'import_temp/' + temp_file_name) with open(temp_file_path, 'wb+') as docx_file: for chunk in import_file: docx_file.write(chunk) if os.path.exists(temp_file_path): import_file = ImportDocxDoc( docx_file_path=temp_file_path, editor_mode=editor_mode, create_user=request.user ).run() return JsonResponse(import_file) else: return JsonResponse({'status': False, 'data': '上传失败'}) else: return JsonResponse({'status': False, 'data': '仅支持.docx格式'}) else: return JsonResponse({'status': False, 'data': '无有效文件'}) else: return JsonResponse({'status': False, 'data': '参数错误'})