from django.contrib.auth import authenticate from rest_framework.request import Request from rest_framework.views import APIView from ..op_drf.filters import DataLevelPermissionsFilter from ..op_drf.viewsets import CustomModelViewSet from ..permission.filters import MenuFilter, DeptFilter, PostFilter, RoleFilter, UserProfileFilter from ..permission.models import Role, Menu, Dept, Post, UserProfile from ..permission.serializers import UserProfileSerializer, MenuSerializer, RoleSerializer, \ MenuCreateUpdateSerializer, DeptSerializer, DeptCreateUpdateSerializer, PostSerializer, PostCreateUpdateSerializer, \ RoleCreateUpdateSerializer, DeptTreeSerializer, MenuTreeSerializer, UserProfileCreateUpdateSerializer, \ PostSimpleSerializer, RoleSimpleSerializer, ExportUserProfileSerializer, ExportRoleSerializer, ExportPostSerializer, \ UserProfileImportSerializer from ..system.models import DictDetails from ..utils.export_excel import export_excel_save_model from ..utils.response import SuccessResponse, ErrorResponse class GetUserProfileView(APIView): """ 获取用户详细信息 """ def get(self, request, format=None): user_dict = UserProfileSerializer(request.user).data permissions_list = ['*:*:*'] if user_dict.get('admin') else Menu.objects.filter( role__userprofile=request.user).values_list('perms', flat=True) return SuccessResponse({ 'permissions': [ele for ele in permissions_list if ele], 'roles': Role.objects.filter(userprofile=request.user).values_list('roleKey', flat=True), 'user': user_dict }) class GetRouters(APIView): """ 获取路由详细信息 """ def depth_menu(self, menus): """ 获取菜单(通过前端递归值) :param menus: :return: """ return dict def get(self, request, format=None): menus = Menu.objects.filter(role__userprofile=request.user) \ .exclude(menuType='2').values('id', 'name', 'web_path', 'visible', 'status', 'isFrame', 'component_path', 'icon', 'parentId', 'orderNum', 'isCache').distinct() data = [] sys_show_hide = DictDetails.get_default_dictValue('sys_show_hide') for ele in menus: data.append({ 'id': ele.get('id'), 'name': ele.get('web_path', '').split('/')[-1] and ele.get('web_path', '').split('/')[-1].capitalize(), 'path': ele.get('web_path'), 'orderNum': ele.get('orderNum'), 'hidden': True if ele.get('visible') != '1' else False, 'redirect': ele.get('web_path') if ele.get('isFrame') == '1' else 'noRedirect', 'component': ele.get('component_path') or 'Layout', 'meta': {'title': ele.get('name'), 'icon': ele.get('icon'), 'noCache': True if ele.get('isCache') == sys_show_hide else False}, 'parentId': ele.get('parentId') }) return SuccessResponse(data) class MenuModelViewSet(CustomModelViewSet): """ 菜单模型 的CRUD视图 """ queryset = Menu.objects.all() serializer_class = MenuSerializer create_serializer_class = MenuCreateUpdateSerializer update_serializer_class = MenuCreateUpdateSerializer filter_class = MenuFilter extra_filter_backends = [DataLevelPermissionsFilter] # update_extra_permission_classes = (IsManagerPermission,) # destroy_extra_permission_classes = (IsManagerPermission,) # create_extra_permission_classes = (IsManagerPermission,) search_fields = ('name',) ordering = 'create_datetime' # 默认排序 def tree_select_list(self, request: Request, *args, **kwargs): """ 递归获取部门树 :param request: :param args: :param kwargs: :return: """ queryset = self.filter_queryset(self.get_queryset()) if hasattr(self, 'handle_logging'): self.handle_logging(request, *args, **kwargs) serializer = MenuTreeSerializer(queryset, many=True) return SuccessResponse(serializer.data) def role_menu_tree_select(self, request: Request, *args, **kwargs): """ 根据角色ID查询菜单下拉树结构 :param request: :param args: :param kwargs: :return: """ menu_queryset = Menu.objects.filter(role__id=kwargs.get('pk')).values_list('id', flat=True) queryset = self.filter_queryset(self.get_queryset()) if hasattr(self, 'handle_logging'): self.handle_logging(request, *args, **kwargs) serializer = MenuTreeSerializer(queryset, many=True) return SuccessResponse({ 'menus': serializer.data, 'checkedKeys': menu_queryset }) class DeptModelViewSet(CustomModelViewSet): """ 部门管理 的CRUD视图 """ queryset = Dept.objects.all() serializer_class = DeptSerializer create_serializer_class = DeptCreateUpdateSerializer update_serializer_class = DeptCreateUpdateSerializer filter_class = DeptFilter extra_filter_backends = [DataLevelPermissionsFilter] # update_extra_permission_classes = (IsManagerPermission,) # destroy_extra_permission_classes = (IsManagerPermission,) # create_extra_permission_classes = (IsManagerPermission,) search_fields = ('deptName',) ordering = 'create_datetime' # 默认排序 def exclude_list(self, request: Request, *args, **kwargs): """ 过滤剔除同级部门 :param request: :param args: :param kwargs: :return: """ dept_queryset = Dept.objects.filter(id=kwargs.get('pk')).first() parentId = dept_queryset.parentId if dept_queryset else '' queryset = self.queryset.exclude(parentId=parentId).order_by('orderNum') if hasattr(self, 'handle_logging'): self.handle_logging(request, *args, **kwargs) serializer = self.get_serializer(queryset, many=True) return SuccessResponse(serializer.data) def tree_select_list(self, request: Request, *args, **kwargs): """ 递归获取部门树 :param request: :param args: :param kwargs: :return: """ queryset = self.filter_queryset(self.get_queryset()) if hasattr(self, 'handle_logging'): self.handle_logging(request, *args, **kwargs) serializer = DeptTreeSerializer(queryset, many=True) return SuccessResponse(serializer.data) def role_dept_tree_select(self, request: Request, *args, **kwargs): """ 根据角色ID查询部门树结构 :param request: :param args: :param kwargs: :return: """ dept_queryset = Dept.objects.filter(role__id=kwargs.get('pk')).values_list('id', flat=True) queryset = self.filter_queryset(self.get_queryset()) if hasattr(self, 'handle_logging'): self.handle_logging(request, *args, **kwargs) serializer = DeptTreeSerializer(queryset, many=True) return SuccessResponse({ 'depts': serializer.data, 'checkedKeys': dept_queryset }) class PostModelViewSet(CustomModelViewSet): """ 岗位管理 的CRUD视图 """ queryset = Post.objects.all() serializer_class = PostSerializer create_serializer_class = PostCreateUpdateSerializer update_serializer_class = PostCreateUpdateSerializer filter_class = PostFilter extra_filter_backends = [DataLevelPermissionsFilter] # update_extra_permission_classes = (IsManagerPermission,) # destroy_extra_permission_classes = (IsManagerPermission,) # create_extra_permission_classes = (IsManagerPermission,) search_fields = ('postName',) ordering = ['postSort', 'create_datetime'] # 默认排序 def export(self, request: Request, *args, **kwargs): """ 导出岗位 :param request: :param args: :param kwargs: :return: """ field_data = ['岗位序号', '岗位编码', '岗位名称', '岗位排序', '状态', '创建者', '修改者', '备注'] data = ExportPostSerializer(Post.objects.all(), many=True).data return SuccessResponse(export_excel_save_model(request, field_data, data, '导出岗位数据.xls')) class RoleModelViewSet(CustomModelViewSet): """ 角色管理 的CRUD视图 """ queryset = Role.objects.all() serializer_class = RoleSerializer create_serializer_class = RoleCreateUpdateSerializer update_serializer_class = RoleCreateUpdateSerializer filter_class = RoleFilter extra_filter_backends = [DataLevelPermissionsFilter] # update_extra_permission_classes = (IsManagerPermission,) # destroy_extra_permission_classes = (IsManagerPermission,) # create_extra_permission_classes = (IsManagerPermission,) search_fields = ('roleName',) ordering = 'create_datetime' # 默认排序 def export(self, request: Request, *args, **kwargs): """ 导出角色 :param request: :param args: :param kwargs: :return: """ field_data = ['角色序号', '角色名称', '角色权限', '角色排序', '数据范围', '角色状态', '创建者', '修改者', '备注'] data = ExportRoleSerializer(Role.objects.all(), many=True).data return SuccessResponse(export_excel_save_model(request, field_data, data, '导出角色数据.xls')) class UserProfileModelViewSet(CustomModelViewSet): """ 用户管理 的CRUD视图 """ queryset = UserProfile.objects.all() serializer_class = UserProfileSerializer create_serializer_class = UserProfileCreateUpdateSerializer update_serializer_class = UserProfileCreateUpdateSerializer filter_class = UserProfileFilter extra_filter_backends = [DataLevelPermissionsFilter] # 导出 export_serializer_class = ExportUserProfileSerializer export_field_data = ['用户序号', '登录名称', '用户名称', '用户邮箱', '手机号码', '用户性别', '帐号状态', '最后登录时间', '部门名称', '部门负责人'] # 导入 import_serializer_class = UserProfileImportSerializer import_field_data = {'username': '登录账号', 'name': '用户名称', 'email': '用户邮箱', 'mobile': '手机号码', 'gender': '用户性别(男/女/未知)', 'is_active': '帐号状态(启用/禁用)', 'password': '登录密码', 'dept': '部门ID', 'role': '角色ID', 'post': '岗位ID'} # update_extra_permission_classes = (IsManagerPermission,) # destroy_extra_permission_classes = (IsManagerPermission,) # create_extra_permission_classes = (IsManagerPermission,) search_fields = ('username',) ordering = 'create_datetime' # 默认排序 def change_status(self, request: Request, *args, **kwargs): """ 修改用户状态 :param request: :param args: :param kwargs: :return: """ instance = self.queryset.get(id=request.data.get('userId')) instance.is_active = request.data.get('status') instance.save() serializer = self.get_serializer(instance) if hasattr(self, 'handle_logging'): self.handle_logging(request, instance=instance, *args, **kwargs) return SuccessResponse(serializer.data) def get_user_details(self, request: Request, *args, **kwargs): """ 获取用户详情 :param request: :param args: :param kwargs: :return: """ userId = request.query_params.get('userId') data = { 'posts': PostSimpleSerializer(Post.objects.all().order_by('postSort'), many=True).data, 'roles': RoleSimpleSerializer(Role.objects.all().order_by('roleSort'), many=True).data } if userId: instance = self.queryset.get(id=userId) serializer = self.get_serializer(instance) data['data'] = serializer.data data['postIds'] = [ele.get('id') for ele in serializer.data.get('post')] data['roleIds'] = [ele.get('id') for ele in serializer.data.get('role')] if hasattr(self, 'handle_logging'): self.handle_logging(request, instance=instance, *args, **kwargs) return SuccessResponse(data) def reset_pwd(self, request: Request, *args, **kwargs): """ 重置密码 :param request: :param args: :param kwargs: :return: """ instance = self.queryset.get(id=request.data.get('userId')) serializer = self.get_serializer(instance) instance.set_password(request.data.get('password')) instance.save() if hasattr(self, 'handle_logging'): self.handle_logging(request, instance=instance, *args, **kwargs) return SuccessResponse(serializer.data) def profile(self, request: Request, *args, **kwargs): """ 获取用户个人信息 :param request: :param args: :param kwargs: :return: """ instance = self.queryset.get(id=request.user.id) serializer = self.get_serializer(instance) if hasattr(self, 'handle_logging'): self.handle_logging(request, instance=instance, *args, **kwargs) return SuccessResponse(serializer.data) def put_profile(self, request: Request, *args, **kwargs): """ 更新用户个人信息 :param request: :param args: :param kwargs: :return: """ instance = self.queryset.get(id=request.user.id) instance.name = request.data.get('name', None) instance.mobile = request.data.get('mobile', None) instance.email = request.data.get('email', None) instance.gender = request.data.get('gender', None) instance.save() serializer = self.get_serializer(instance) if hasattr(self, 'handle_logging'): self.handle_logging(request, instance=instance, *args, **kwargs) return SuccessResponse(serializer.data) def put_avatar(self, request: Request, *args, **kwargs): """ 更新用户头像 :param request: :param args: :param kwargs: :return: """ instance = self.queryset.get(id=request.user.id) instance.avatar = request.data.get('avatar_url', None) instance.save() serializer = self.get_serializer(instance) if hasattr(self, 'handle_logging'): self.handle_logging(request, instance=instance, *args, **kwargs) return SuccessResponse(serializer.data) def update_pwd(self, request: Request, *args, **kwargs): """ 个人修改密码 :param request: :param args: :param kwargs: :return: """ instance = self.queryset.get(id=request.user.id) instance.mobile = request.data.get('newPassword', None) if not authenticate(username=request.user.username, password=request.data.get('oldPassword', None)): return ErrorResponse(msg='旧密码不正确！') instance.set_password(request.data.get('newPassword')) instance.save() serializer = self.get_serializer(instance) if hasattr(self, 'handle_logging'): self.handle_logging(request, instance=instance, *args, **kwargs) return SuccessResponse(serializer.data)