from django.contrib.auth.hashers import check_password, make_password from django.db import transaction from django_celery_beat.models import (CrontabSchedule, IntervalSchedule, PeriodicTask) from django_celery_results.models import TaskResult from rest_framework.decorators import action from rest_framework.exceptions import ParseError, ValidationError from rest_framework.mixins import (CreateModelMixin, DestroyModelMixin, ListModelMixin, RetrieveModelMixin) from rest_framework.parsers import (JSONParser, MultiPartParser) from rest_framework.permissions import IsAuthenticated from rest_framework.response import Response from rest_framework.views import APIView from apps.utils.mixins import (CustomCreateModelMixin) from django.conf import settings from apps.utils.permission import get_user_perms_map from apps.utils.queryset import get_child_queryset2 from apps.utils.viewsets import CustomGenericViewSet, CustomModelViewSet from server.celery import app as celery_app from .filters import UserFilter from .models import (Dept, Dict, DictType, File, Permission, Post, Role, User, UserPost) from .serializers import (DeptCreateUpdateSerializer, DeptSerializer, DictCreateUpdateSerializer, DictSerializer, DictTypeCreateUpdateSerializer, DictTypeSerializer, FileSerializer, PermissionCreateUpdateSerializer, PermissionSerializer, PostCreateUpdateSerializer, PostSerializer, PTaskCreateUpdateSerializer, PTaskResultSerializer, PTaskSerializer, RoleCreateUpdateSerializer, RoleSerializer, UserCreateSerializer, UserListSerializer, UserPostCreateSerializer, UserPostSerializer, UserUpdateSerializer) # logger.info('请求成功！ response_code:{}；response_headers:{}； # response_body:{}'.format(response_code, response_headers, response_body[:251])) # logger.error('请求出错-{}'.format(error)) class TaskList(APIView): permission_classes = [IsAuthenticated] def get(self, requests): """获取注册任务列表 获取注册任务列表 """ tasks = list( sorted(name for name in celery_app.tasks if not name.startswith('celery.'))) return Response(tasks) class PTaskViewSet(CustomModelViewSet): """ list:定时任务列表 定时任务列表 retrieve:定时任务详情 定时任务详情 """ queryset = PeriodicTask.objects.exclude(name__contains='celery.') serializer_class = PTaskSerializer create_serializer_class = PTaskCreateUpdateSerializer update_serializer_class = PTaskCreateUpdateSerializer search_fields = ['name', 'task'] filterset_fields = ['enabled'] ordering = ['-create_time'] @action(methods=['put'], detail=True, perms_map={'put': 'ptask_update'}) def toggle(self, request, pk=None): """修改启用禁用状态 修改启用禁用状态 """ obj = self.get_object() obj.enabled = False if obj.enabled else True obj.save() return Response() @transaction.atomic def create(self, request, *args, **kwargs): """创建定时任务 创建定时任务 """ data = request.data timetype = data.get('timetype', None) interval_ = data.get('interval_', None) crontab_ = data.get('crontab_', None) if timetype == 'interval' and interval_: data['crontab'] = None try: interval, _ = IntervalSchedule.objects.get_or_create( **interval_, defaults=interval_) data['interval'] = interval.id except: raise ParseError('时间策略有误', 'schedule_wrong') if timetype == 'crontab' and crontab_: data['interval'] = None try: crontab_['timezone'] = 'Asia/Shanghai' crontab, _ = CrontabSchedule.objects.get_or_create( **crontab_, defaults=crontab_) data['crontab'] = crontab.id except: raise ParseError('时间策略有误', 'schedule_wrong') serializer = self.get_serializer(data=data) serializer.is_valid(raise_exception=True) serializer.save() return Response() @transaction.atomic def update(self, request, *args, **kwargs): """更新定时任务 更新定时任务 """ data = request.data timetype = data.get('timetype', None) interval_ = data.get('interval_', None) crontab_ = data.get('crontab_', None) if timetype == 'interval' and interval_: data['crontab'] = None try: if 'id' in interval_: del interval_['id'] interval, _ = IntervalSchedule.objects.get_or_create( **interval_, defaults=interval_) data['interval'] = interval.id except: raise ParseError('时间策略有误', 'schedule_wrong') if timetype == 'crontab' and crontab_: data['interval'] = None try: crontab_['timezone'] = 'Asia/Shanghai' if 'id' in crontab_: del crontab_['id'] crontab, _ = CrontabSchedule.objects.get_or_create( **crontab_, defaults=crontab_) data['crontab'] = crontab.id except: raise ParseError('时间策略有误', 'schedule_wrong') instance = self.get_object() serializer = self.get_serializer(instance, data=data) serializer.is_valid(raise_exception=True) serializer.save() return Response() class PTaskResultViewSet(ListModelMixin, RetrieveModelMixin, CustomGenericViewSet): """ list:任务执行结果列表 任务执行结果列表 retrieve:任务执行结果详情 任务执行结果详情 """ perms_map = {'get': '*'} filterset_fields = ['task_name'] queryset = TaskResult.objects.all() serializer_class = PTaskResultSerializer ordering = ['-date_created'] class DictTypeViewSet(CustomModelViewSet): """数据字典类型-增删改查 数据字典类型-增删改查 """ queryset = DictType.objects.all() serializer_class = DictTypeSerializer create_serializer_class = DictTypeCreateUpdateSerializer update_serializer_class = DictTypeCreateUpdateSerializer search_fields = ['name'] class DictViewSet(CustomModelViewSet): """数据字典-增删改查 数据字典-增删改查 """ # queryset = Dict.objects.get_queryset(all=True) # 获取全部的,包括软删除的 queryset = Dict.objects.all() filterset_fields = ['type', 'is_used', 'type__code'] serializer_class = DictSerializer create_serializer_class = DictCreateUpdateSerializer update_serializer_class = DictCreateUpdateSerializer search_fields = ['name'] class PostViewSet(CustomModelViewSet): """岗位-增删改查 岗位-增删改查 """ queryset = Post.objects.all() serializer_class = PostSerializer create_serializer_class = PostCreateUpdateSerializer update_serializer_class = PostCreateUpdateSerializer search_fields = ['name', 'description'] class PermissionViewSet(CustomModelViewSet): """菜单权限-增删改查 菜单权限-增删改查 """ queryset = Permission.objects.all() filterset_fields = ['type'] serializer_class = PermissionSerializer create_serializer_class = PermissionCreateUpdateSerializer update_serializer_class = PermissionCreateUpdateSerializer search_fields = ['name', 'code'] class DeptViewSet(CustomModelViewSet): """部门-增删改查 部门-增删改查 """ queryset = Dept.objects.all() serializer_class = DeptSerializer create_serializer_class = DeptCreateUpdateSerializer update_serializer_class = DeptCreateUpdateSerializer filterset_fields = ['type'] search_fields = ['name'] class RoleViewSet(CustomModelViewSet): """角色-增删改查 角色-增删改查 """ queryset = Role.objects.all() serializer_class = RoleSerializer create_serializer_class = RoleCreateUpdateSerializer update_serializer_class = RoleCreateUpdateSerializer search_fields = ['name', 'code'] class UserPostViewSet(CreateModelMixin, DestroyModelMixin, ListModelMixin, CustomGenericViewSet): """用户/岗位关系 用户/岗位关系 """ perms_map = {'get': '*', 'post': 'user_update', 'delete': 'user_update'} queryset = UserPost.objects.select_related('user', 'post', 'dept').all() serializer_class = UserPostSerializer create_serializer_class = UserPostCreateSerializer filterset_fields = ['user', 'post', 'dept'] def perform_create(self, serializer): instance = serializer.save() user = instance.user adept = UserPost.objects.filter(user=user).order_by('sort', 'create_time').first() if adept: user.belong_dept = adept user.update_by = self.request.user user.save() def perform_destroy(self, instance): user = instance.user instance.delete() fdept = UserPost.objects.filter(user=user).order_by('sort', 'create_time').first() if fdept: user.belong_dept = fdept else: user.belong_dept = None user.update_by = self.request.user user.save() class UserViewSet(CustomModelViewSet): queryset = User.objects.all() serializer_class = UserListSerializer create_serializer_class = UserCreateSerializer update_serializer_class = UserUpdateSerializer filterset_class = UserFilter search_fields = ['username', 'name', 'phone', 'email'] def get_queryset(self): queryset = self.queryset if hasattr(self.get_serializer_class(), 'setup_eager_loading'): queryset = self.get_serializer_class().setup_eager_loading(queryset) # 性能优化 dept = self.request.query_params.get( 'belong_dept', None) # 该部门及其子部门所有员工 if dept: dept_queryset = get_child_queryset2( Dept.objects.get(pk=dept)) queryset = queryset.filter(dept__in=dept_queryset) return queryset def create(self, request, *args, **kwargs): """创建用户 创建用户 """ password = request.data.get('password', None) if password: password = make_password(password) else: password = make_password('0000') serializer = self.get_serializer(data=request.data) serializer.is_valid(raise_exception=True) serializer.save(password=password) return Response(data=serializer.data) @action(methods=['put'], detail=False, permission_classes=[IsAuthenticated]) def password(self, request, pk=None): """修改密码 修改密码 """ user = request.user old_password = request.data['old_password'] if check_password(old_password, user.password): new_password1 = request.data['new_password1'] new_password2 = request.data['new_password2'] if new_password1 == new_password2: user.set_password(new_password2) user.save() return Response() else: raise ParseError('新密码两次输入不一致!', 'password_not_same') else: raise ValidationError('旧密码错误!', 'old_password_wrong') @action(methods=['get'], detail=False, permission_classes=[IsAuthenticated]) def info(self, request, pk=None): """登录用户信息 获取登录用户信息 """ user = request.user perms = get_user_perms_map(user) data = { 'id': user.id, 'username': user.username, 'name': user.name, 'posts': user.posts.values_list('name', flat=True), 'avatar': user.avatar, 'perms': perms, } return Response(data) class FileViewSet(CustomCreateModelMixin, RetrieveModelMixin, ListModelMixin, CustomGenericViewSet): """文件上传 list: 文件列表 文件列表 create: 文件上传 文件上传 """ permission_classes = [IsAuthenticated] parser_classes = [MultiPartParser, JSONParser] queryset = File.objects.all() serializer_class = FileSerializer filterset_fields = ['type'] search_fields = ['name'] def perform_create(self, serializer): file_obj = self.request.data.get('file') name = file_obj._name size = file_obj.size mime = file_obj.content_type file_type = '其它' if 'image' in mime: file_type = '图片' elif 'video' in mime: file_type = '视频' elif 'audio' in mime: file_type = '音频' elif 'application' or 'text' in mime: file_type = '文档' instance = serializer.save( create_by=self.request.user, name=name, size=size, type=file_type, mime=mime) instance.path = settings.MEDIA_URL + instance.file.name instance.save()