223 lines
10 KiB
Python
223 lines
10 KiB
Python
# 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 django.utils.translation import gettext_lazy as _
|
||
from loguru import logger
|
||
from app_doc.report_utils import *
|
||
from app_admin.decorators import check_headers,allow_report_file
|
||
from app_doc.import_utils import *
|
||
import datetime
|
||
import traceback
|
||
import re
|
||
import os.path
|
||
import json
|
||
|
||
|
||
# 导入文集
|
||
@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': _('参数错误')}) |