import logging from django.db import transaction from django.db.models import Q from django.shortcuts import render from rest_framework.response import Response from rest_framework.views import APIView from rest_framework import viewsets from app.pagination import MyPageNumberPagination from user.models import UserProfile from .models import Department from .serializers import DepartmentSerializer # Create your views here. logger = logging.getLogger('mylogger') class DepartmentView(viewsets.GenericViewSet): def list(self, request, *args, **kwargs): data = dict() try: parent_departments = list(Department.objects.filter(parent_id__isnull=True).values()) for parent_department in parent_departments: parent_id = parent_department.get('id') son_departs = list(Department.objects.filter(parent_id=parent_id).values()) parent_department['children'] = son_departs data['data'] = parent_departments return Response(data) except Exception as e: return Response({'msg': f'查询失败, {e}'}) def query_department_users(self, request, *args, **kwargs): res = dict() data = request.data department_id = data.get('department_id') try: users = list(UserProfile.objects.filter(department_id=department_id).values('id', 'username', 'phone_number', 'status', 'department_id')) res['data'] = users return Response(res) except Exception as e: return Response({'msg': f'查询失败，原因{e}'}) # def list_son_departments(self, request, *args, **kwargs): # query_params = dict(request.query_params) # parent_name = query_params.get('parent_name') # if isinstance(parent_name, list): # parent_name = ''.join(parent_name) # logger.info(parent_name) # parent_obj = Department.objects.filter(name=parent_name).first() # if not parent_obj: # return Response({'msg': '非法请求，该父级部门不存在'}) # parent_id = parent_obj.id # queryset = Department.objects.filter(parent_id=parent_id).order_by('-id') # paginator = MyPageNumberPagination() # paginated_queryset = paginator.paginate_queryset(queryset, request) # serializer = DepartmentSerializer(paginated_queryset, many=True) # # return paginator.get_paginated_response(serializer.data) # def create(self, request, *args, **kwargs): # data = request.data # parent_depart_name = data.get('parent_department') # if not parent_depart_name: # return Response({"msg": "请输入父级部门名称"}) # parent_obj = Department.objects.filter(name=parent_depart_name).first() # if parent_obj: # return Response({'msg': '该部门已存在，请换一个'}) # try: # with transaction.atomic(): # new_obj = Department.objects.create(name=parent_depart_name) # return Response({'msg': '创建父级部门成功'}) # except Exception as e: # return Response({'msg': f'创建失败，原因{e}'}) # def create(self, request, *args, **kwargs): # data = request.data # parent_name = data.get('parent_department_name') # if not parent_name: # return Response({'msg': '请输入父级部门名称'}) # depart_name = data.get('department_name') # try: # with transaction.atomic(): # if not depart_name: # old_parent_obj = Department.objects.filter(name=parent_name).first() # if old_parent_obj: # return Response({'msg': '该父级部门名称已存在，请换一个'}) # parent_obj = Department.objects.create(name=parent_name) # return Response({'msg': '创建父级部门成功'}) # else: # old_parent_obj = Department.objects.filter(name=parent_name).first() # if not old_parent_obj: # old_parent_obj = Department.objects.create(name=parent_name) # old_depart_obj = Department.objects.filter(name=depart_name, parent_id=old_parent_obj.id).first() # if old_depart_obj: # return Response({'msg': f'{parent_name}下已有{depart_name}啦，请换一个'}) # new_depart_obj = Department.objects.create(name=depart_name, parent_id=old_parent_obj.id) # return Response({'msg': '创建子部门成功'}) # except Exception as e: # return Response({'msg': f'创建失败，{e}'}) def create(self, request, *args, **kwargs): data = request.data department_name = data.get('department_name') department_id = data.get('department_id') if not department_id or department_id == '': old_depart_obj = Department.objects.filter(name=department_name, parent_id__isnull=True).first() if old_depart_obj: return Response({'msg': f'该父级部门{department_name}已存在'}) depart_obj = Department.objects.create(name=department_name) return Response({'msg': f'创建父级部门{department_name}成功', 'success': True}) else: parent_obj = Department.objects.filter(id=department_id, parent_id__isnull=True).first() if not parent_obj: return Response({'msg': '非法请求'}) old_son_obj = Department.objects.filter(name=department_name, parent_id=department_id).first() if old_son_obj: return Response({'msg': f'该父级部门{parent_obj.name}下已有{old_son_obj.name}啦'}) son_obj = Department.objects.create(name=department_name, parent_id=department_id) return Response({'msg': f'创建父级部门{parent_obj.name}下{son_obj.name}成功', 'success': True}) # def create_son_department(self, request, *args, **kwargs): # data = request.data # parent_name = data.get('parent_name') # if not parent_name: # return Response({'msg': '请输入父级部门名称以便创建子部门'}) # parent_obj = Department.objects.filter(name=parent_name).first() # parent_id = parent_obj.id # department_name = data.get('department_name') # if not department_name: # return Response({'msg': '请输入子部门名称'}) # depart_obj = Department.objects.filter(name=department_name, parent_id=parent_id).first() # if depart_obj: # return Response({'msg': f'{parent_name}下已有{department_name}，请换一个'}) # try: # with transaction.atomic(): # Department.objects.create(name=department_name, parent_id=parent_id) # return Response({'msg': '创建成功'}) # except Exception as e: # return Response({'msg': f'创建失败，原因{e}'}) def update(self, request, pk, *args, **kwargs): data = request.data try: with transaction.atomic(): depart_obj = Department.objects.filter(id=pk).update(**data) return Response({'msg': '更新成功', 'success': True}) except Department.DoesNotExist: return Response({'msg': '该部门不存在'}) except Exception as e: return Response({'msg': f'更新失败,原因{e}'}) def destroy(self, request, pk, *args, **kwargs): try: with transaction.atomic(): depart_obj = Department.objects.filter(id=pk).delete() return Response({'msg': '删除成功', 'success': True}) except Department.DoesNotExist: return Response({'msg': '该部门不存在'}) except Exception as e: return Response({'msg': f'删除失败，原因{e}'})