# coding:utf-8 # @文件: views_app.py # @创建者：州的先生 # #日期：2020/5/11 # 博客地址：zmister.com from django.contrib.auth.models import User from django.shortcuts import render from django.contrib.auth import authenticate from django.core.exceptions import ObjectDoesNotExist from django.db.models import Q from django.conf import settings from django.utils.translation import gettext_lazy as _ from rest_framework.views import APIView from app_api.models import AppUserToken from rest_framework.response import Response from rest_framework.pagination import PageNumberPagination from rest_framework.authentication import SessionAuthentication from app_doc.models import * from app_api.serializers_app import * from app_api.auth_app import AppAuth,AppMustAuth from app_doc.views import validateTitle from app_doc.util_upload_img import img_upload,base_img_upload from loguru import logger import datetime import os ''' 响应： code：状态码 data：数据 状态码： 0：成功 1：资源未找到 2：无权访问 3：需要访问码 4：系统异常 5：参数不正确 6：需要登录 ''' # 生成Token的函数 def get_token_code(username): """ 根据用户名和时间戳来生成永不相同的token随机字符串 :param username: 字符串格式的用户名 :return: 字符串格式的Token """ import time import hashlib timestamp = str(time.time()) m = hashlib.md5(username.encode("utf-8")) # md5 要传入字节类型的数据 m.update(timestamp.encode("utf-8")) return m.hexdigest() # 将生成的随机字符串返回 # 登陆视图 class LoginView(APIView): ''' 登陆检测试图。 1，接收用户发过来的用户名和密码数据 2，校验用户密码是否正确 - 成功就返回登陆成功,然后发Token - 失败就返回错误提示 ''' def post(self,request): res = {"code":0} # 从post 里面取数据 # print(request.data) username = request.data.get("username") password = request.data.get("password") # 查询用户是否存在、密码是否匹配 user_obj = authenticate(username=username, password=password) if user_obj: if user_obj.is_active: # 生成Token token = get_token_code(username) # 保存或更新token AppUserToken.objects.update_or_create(defaults={"token": token}, user=user_obj) # 将token返回给用户 res["token"] = token res['username'] = username else: res['code'] = 2 res["error"] = _('账号被禁用') else: # 登陆失败 res["code"] = 1 res["error"] = _("用户名或密码错误") return Response(res) # 文集视图 class ProjectView(APIView): authentication_classes = (AppAuth,SessionAuthentication) # 获取文集 def get(self,request): pro_id = request.query_params.get('id',None) range = request.query_params.get('range',None) # 获取自己的文集创建的、协作的文集列表 if range == 'self': colla_list = [i.project.id for i in ProjectCollaborator.objects.filter(user=request.user)] # 用户的协作文集列表 project_list = Project.objects.filter( Q(create_user=request.user) | \ Q(id__in=colla_list) ).order_by('-create_time') # page = PageNumberPagination() # 实例化一个分页器 # page_projects = page.paginate_queryset(project_list, request, view=self) # 进行分页查询 # serializer = ProjectSerializer(page_projects, many=True) # 对分页后的结果进行序列化处理 serializer = ProjectSerializer(project_list, many=True) resp = { 'code': 0, 'data': serializer.data, 'count': project_list.count() } return Response(resp) # 存在文集ID，返回指定的文集 if pro_id: resp = dict() # 获取文集信息 project = Project.objects.get(id=int(pro_id)) # 获取文集的协作用户信息 # print(request.auth) # print(request.user) if request.auth: # 对登陆用户查询其协作文档信息 colla_user = ProjectCollaborator.objects.filter(project=project, user=request.user).count() else: colla_user = 0 # 获取文集前台下载权限 try: allow_download = ProjectReport.objects.get(project=project) except: allow_download = False # 私密文集并且访问者非创建者非协作者 if (project.role == 1) and (request.user != project.create_user) and (colla_user == 0): # return Response({'code': 2, 'data': []}) resp['code'] = 2 # 指定用户可见文集 elif project.role == 2: user_list = project.role_value if request.auth: # 认证用户判断是否在许可用户列表中 if (request.user.username not in user_list) and \ (request.user != project.create_user) and \ (colla_user == 0): # 访问者不在指定用户之中 resp['code'] = 2 else: # 游客直接返回404 resp['code'] = 2 # 访问码可见 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.data.get(viewcode_name,0) # 获取访问码 if viewcode != r_viewcode: # 访问码不等于文集访问码，跳转到访问码认证界面 # return Response({'code': 3}) resp['code'] = 3 else: serializer = ProjectSerializer(project) resp = {'code': 0, 'data': serializer.data} return Response(resp) # 否则，根据查询条件返回文集列表 else: kw = request.query_params.get('kw', '') # 搜索词 sort = request.query_params.get('sort', 0) # 排序,0表示按时间升序排序，1表示按时间降序排序，默认为0 role = request.query_params.get('role', -1) # 筛选文集权限，默认为显示所有可显示的文集 # 是否排序 if sort in ['', 0, '0']: sort_str = '' else: sort_str = '-' # 是否搜索 if kw == '': is_kw = False else: is_kw = True # 是否认证 if request.auth: 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 Response({'code':2,'data':[]}) # 没有搜索 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 Response({'code': 2, 'data': []}) # 有搜索 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 Response({'code':1,'data':[]}) # 有搜索 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 Response({'code':1,'data':[]}) page = PageNumberPagination() # 实例化一个分页器 page_projects = page.paginate_queryset(project_list,request,view=self) # 进行分页查询 serializer = ProjectSerializer(page_projects,many=True) # 对分页后的结果进行序列化处理 resp = { 'code':0, 'data':serializer.data, 'count':project_list.count() } return Response(resp) # 新增文集 def post(self,request): resp = dict() if request.auth: try: name = request.data.get('pname', '') name = validateTitle(name) desc = request.data.get('desc', '') role = request.data.get('role', 0) role_list = ['0', '1', '2', '3', 0, 1, 2, 3] if name != '': project = Project.objects.create( name=validateTitle(name), intro=desc[:100], create_user=request.user, role=int(role) if role in role_list else 0 ) project.save() resp = {'code':0,'data':{'id': project.id, 'name': project.name}} return Response(resp) else: resp['code'] = 5 resp['data'] = _('参数不正确') return Response(resp) except Exception as e: logger.exception(_("创建文集出错")) resp['code'] = 4 resp['data'] = _('系统异常请稍后再试') return Response(resp) else: resp['code'] = 6 resp['data'] = _('请登录后操作') return Response(resp) # 修改文集 def put(self,request): resp = dict() if request.auth: try: pro_id = request.query_params.get('id', None) project = Project.objects.get(id=pro_id) # 验证用户有权限修改文集 if (request.user == project.create_user) or request.user.is_superuser: name = request.data.get('name', None) content = request.data.get('desc', None) role = request.data.get('role',None) role_value = request.data.get('role_value',None) project.name = validateTitle(name) project.intro = content project.role = role project.role_value = role_value project.save() resp['code'] = 0 resp['data'] = 'ok' # return Response(resp) else: resp['code'] = 2 resp['data'] = _('非法请求') # return Response(resp) except ObjectDoesNotExist: resp['code'] = 1 resp['data'] = _('资源未找到') # return Response(resp) except Exception as e: logger.exception(_("修改文集出错")) resp['code'] = 4 # return Response(resp) else: resp['code'] = 6 return Response(resp) # 删除文集 def delete(self,request): resp = dict() if request.auth: try: pro_id = request.query_params.get('id', '') if pro_id != '': 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)) pro_doc_list.delete() # 删除文集 pro.delete() resp['code'] = 0 resp['data'] = 'ok' # return Response(resp) else: resp['code'] = 2 # return Response(resp) else: resp['code'] = 5 resp['data'] = _('参数错误') # return Response(resp) except ObjectDoesNotExist: resp['code'] = 1 resp['data'] = _('资源未找到') # return Response(resp) except Exception as e: logger.exception(_("API文集删除异常")) resp['code'] = 4 # return Response(resp) else: resp['code'] = 6 return Response(resp) # 文档视图 class DocView(APIView): authentication_classes = (AppAuth,SessionAuthentication) # 获取文档 def get(self,request): pro_id = request.query_params.get('pid','') # 文集ID doc_id = request.query_params.get('did','') # 文档ID doc_format = request.query_params.get('type','json') # 返回格式 # 存在文集ID和文档ID，进行数据库检索 if pro_id != '' and doc_id != '': # 获取文集信息 project = Project.objects.get(id=int(pro_id)) # 获取文集的协作用户信息 if request.auth: 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 Response({'code':2}) # 指定用户可见文集 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 Response({'code': 2}) else: # 游客直接返回404 return Response({'code': 2}) # 访问码可见 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.data.get(viewcode_name,0) # 获取访问码 if viewcode != r_viewcode: # cookie中的访问码不等于文集访问码，跳转到访问码认证界面 return Response({'code':3}) # 获取文档内容 try: doc = Doc.objects.get(id=int(doc_id), status=1) if doc_format == 'json': serializer = DocSerializer(doc) resp = {'code':0,'data':serializer.data} return Response(resp) elif doc_format == 'html': logger.info(_("返回HTML")) # return Response({'status':'html'}) return render(request,'app_api/single_doc_detail.html',locals()) else: logger.info(doc_format) except ObjectDoesNotExist: return Response({'code':4}) # 不存在文集ID和文档ID，返回用户自己的文档列表 else: if request.auth: doc_list = Doc.objects.filter(create_user=request.user,status=1).order_by('-modify_time') page = PageNumberPagination() # 实例化一个分页器 page_docs = page.paginate_queryset(doc_list, request, view=self) # 进行分页查询 serializer = DocSerializer(page_docs, many=True) # 对分页后的结果进行序列化处理 resp = { 'code': 0, 'data': serializer.data, 'count': doc_list.count() } return Response(resp) else: return Response({'code':4}) # 新建文档 def post(self, request): try: project = request.data.get('project','') parent_doc = request.data.get('parent_doc','') doc_name = request.data.get('doc_name','') doc_content = request.data.get('content','') pre_content = request.data.get('pre_content','') sort = request.data.get('sort','') status = request.data.get('status',1) 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: # 创建文档 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 ) return Response({'code':0,'data':{'pro':project,'doc':doc.id}}) else: return Response({'code':2,'data':_('无权操作此文集')}) else: return Response({'code':5,'data':_('请确认文档标题、文集正确')}) except Exception as e: logger.exception(_("api新建文档异常")) return Response({'status':4,'data':_('请求出错')}) # 修改文档 def put(self, request): try: doc_id = request.data.get('doc_id','') # 文档ID project = request.data.get('project', '') # 文集ID parent_doc = request.data.get('parent_doc', '') # 上级文档ID doc_name = request.data.get('doc_name', '') # 文档名称 doc_content = request.data.get('content', '') # 文档内容 pre_content = request.data.get('pre_content', '') # 文档Markdown格式内容 sort = request.data.get('sort', '') # 文档排序 status = request.data.get('status',1) # 文档状态 if doc_id != '' and project != '' and doc_name != '' and project != '-1': doc = Doc.objects.get(id=doc_id) pro_colla = ProjectCollaborator.objects.filter(project=project, user=request.user) # 验证用户有权限修改文档 - 文档的创建者或文集的高级协作者 if (request.user == doc.create_user) or (pro_colla[0].role == 1): # 将现有文档内容写入到文档历史中 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 ) return Response({'code': 0,'data':_('修改成功')}) else: return Response({'code':2,'data':_('未授权请求')}) else: return Response({'code': 5,'data':_('参数错误')}) except Exception as e: logger.exception(_("api修改文档出错")) return Response({'code':4,'data':_('请求出错')}) # 删除文档 def delete(self, request): try: # 获取文档ID doc_id = request.data.get('doc_id', 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 Response({'code': 1, 'data': '文档不存在'}) if (request.user == doc.create_user) or (colla_user_role == 1) or (request.user == project.create_user): # 修改状态为删除 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 Response({'code': 0, 'data': _('删除完成')}) else: return Response({'code': 2, 'data': _('非法请求')}) else: return Response({'code': 5, 'data': _('参数错误')}) except Exception as e: logger.exception(_("api删除文档出错")) return Response({'code': 4, 'data': _('请求出错')}) # 文档模板视图 class DocTempView(APIView): authentication_classes = (AppMustAuth,SessionAuthentication) # 获取文档模板 def get(self, request): temp_id = request.query_params.get('id','') if temp_id != '': doctemp = DocTemp.objects.get(id=int(temp_id)) if request.user == doctemp.create_user: serializer = DocTempSerializer(doctemp) resp = {'code': 0, 'data': serializer.data} else: resp = {'code':2,'data':_('无权操作')} else: doctemps = DocTemp.objects.filter(create_user=request.user) page = PageNumberPagination() page_doctemps = page.paginate_queryset(doctemps,request,view=self) serializer = DocTempSerializer(page_doctemps,many=True) resp = {'code':0,'data':serializer.data,'count':doctemps.count()} return Response(resp) def post(self, request): try: if request.auth: name = request.data.get('name','') content = request.data.get('content','') if name != '': doctemp = DocTemp.objects.create( name = name, content = content, create_user=request.user ) doctemp.save() return Response({'code':0,'data':_('创建成功')}) else: return Response({'code':5,'data':_('模板标题不能为空')}) else: return Response({'code':6,'data':_('请登录')}) except Exception as e: logger.exception(_("api创建文档模板出错")) return Response({'code':4,'data':_('请求出错')}) def put(self, request): try: doctemp_id = request.data.get('doctemp_id','') name = request.data.get('name','') content = request.data.get('content','') if doctemp_id != '' and name !='': doctemp = DocTemp.objects.get(id=doctemp_id) # 验证请求用户为文档模板的创建者 if request.user == doctemp.create_user: doctemp.name = name doctemp.content = content doctemp.save() return Response({'code':0,'data':_('修改成功')}) else: return Response({'code':2,'data':_('非法操作')}) else: return Response({'code':5,'data':_('参数错误')}) except Exception as e: logger.exception(_("api修改文档模板出错")) return Response({'code':4,'data':_('请求出错')}) def delete(self, request): try: doctemp_id = request.data.get('doctemp_id', '') if doctemp_id != '': doctemp = DocTemp.objects.get(id=doctemp_id) if request.user == doctemp.create_user: doctemp.delete() return Response({'code': 0, 'data': _('删除完成')}) else: return Response({'code': 2, 'data': _('非法请求')}) else: return Response({'code': 5, 'data': _('参数错误')}) except Exception as e: logger.exception(_("api删除文档模板出错")) return Response({'code': 4, 'data': _('请求出错')}) # 图片视图 class ImageView(APIView): authentication_classes = (AppMustAuth,SessionAuthentication) # 获取 def get(self, request): g_id = int(request.query_params.get('group', 0)) # 图片分组id if int(g_id) == 0: image_list = Image.objects.filter(user=request.user) # 查询所有图片 elif int(g_id) == -1: image_list = Image.objects.filter(user=request.user, group_id=None) # 查询指定分组的图片 else: image_list = Image.objects.filter(user=request.user, group_id=g_id) # 查询指定分组的图片 page = PageNumberPagination() page_images = page.paginate_queryset(image_list,request,view=self) serializer = ImageSerializer(page_images,many=True) resp = {'code':0,'data':serializer.data,'count':image_list.count()} return Response(resp) # 上传 def post(self, request): img = request.data.get("api_img_upload", None) # 编辑器上传 # manage_upload = request.data.get('manage_upload', None) # 图片管理上传 dir_name = request.data.get('dirname', '') base_img = request.data.get('base', None) if img: # 上传普通图片文件 result = img_upload(img, dir_name, request.user) resp = {'code':0,'data':result['url']} # elif manage_upload: # result = img_upload(manage_upload, dir_name, request.user) # resp = {'code': 0, 'data': result['url']} elif base_img: # 上传base64编码图片 result = base_img_upload(base_img, dir_name, request.user) resp = {'code': 0, 'data': result['url']} else: resp = {"code": 5, "message": _("出错信息")} return Response(resp) # 删除 def delete(self, request): img_id = request.data.get('id', '') img = Image.objects.get(id=img_id) if img.user != request.user: return Response({'code': 2, '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 Response({'code': 0, 'data': 'ok'}) # 图片分组视图 class ImageGroupView(APIView): authentication_classes = (AppMustAuth,SessionAuthentication) def get(self, request): 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 Response({'code': 0, 'data': group_list}) except: return Response({'code': 4, 'data': _('出现错误')}) def post(self, request): group_name = request.data.get('group_name', '') if group_name not in ['', _('默认分组'), _('未分组')]: ImageGroup.objects.create( user=request.user, group_name=group_name ) return Response({'code': 0, 'data': 'ok'}) else: return Response({'code': 5, 'data': _('名称无效')}) def put(self, request): group_name = request.data.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 Response({'code': 0, 'data': 'ok'}) else: return Response({'code': 5, 'data': _('名称无效')}) def delete(self, request): try: group_id = request.data.get('group_id', '') group = ImageGroup.objects.get(id=group_id, user=request.user) # 查询分组 images = Image.objects.filter(group_id=group_id).update(group_id=None) # 移动图片到未分组 group.delete() # 删除分组 return Response({'code': 0, 'data': 'ok'}) except: return Response({'code': 4, 'data': _('删除错误')}) # 附件视图 class AttachmentView(APIView): authentication_classes = (AppMustAuth,SessionAuthentication) # 文件大小 字节转换 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 def get(self, request): attachment_list = [] attachments = Attachment.objects.filter(user=request.user) 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 Response({'code': 0, 'data': attachment_list}) def post(self, request): attachment = request.data.get('attachment_upload', None) if attachment: attachment_name = attachment.name attachment_size = self.sizeFormat(attachment.size) # 限制附件大小在50mb以内 if attachment.size > 52428800: return Response({'code': False, 'data': _('文件大小超出限制')}) # 限制附件为ZIP格式文件 if attachment_name.endswith('.zip'): a = Attachment.objects.create( file_name=attachment_name, file_size=attachment_size, file_path=attachment, user=request.user ) return Response({'code': 0, 'data': {'name': attachment_name, 'url': a.file_path.name}}) else: return Response({'code': 5, 'data': _('不支持的格式')}) else: return Response({'code': 5, 'data': _('无效文件')}) def delete(self, request): attach_id = request.data.get('attach_id', '') attachment = Attachment.objects.filter(id=attach_id, user=request.user) # 查询附件 for a in attachment: # 遍历附件 a.file_path.delete() # 删除文件 attachment.delete() # 删除数据库记录 return Response({'code': 0, 'data': 'ok'})