ChaCeRndTrans/apps/common/views/dict.py

# -*- coding: utf-8 -*-
import datetime
import distutils
import json
import logging
import traceback
import uuid
from collections import OrderedDict

from django.db.models import Q
from rest_framework.views import View
from rest_framework import status
from rest_framework.decorators import action
from rest_framework.filters import SearchFilter, OrderingFilter
from rest_framework.permissions import IsAuthenticated
from rest_framework.viewsets import ModelViewSet
from rest_framework_jwt.authentication import JSONWebTokenAuthentication

from ChaCeRndTrans.basic import CCAIResponse
from ChaCeRndTrans.code import SERVER_ERROR, PARAMS_ERR
from utils.custom import RbacPermission, CustomViewBase, req_operate_by_user, create_operation_history_log
from ..models import DataDictionary, DataDictionaryDetail, OperationHistoryLog, DataDictionaryDetailZY
from ..serializers.dict_serializer import DictSerializer, DictDetailSerializer

logger = logging.getLogger('error')


class DataDictionaryViewSet(CustomViewBase):
    '''
    字典管理：增删改查
    '''
    perms_map = ({'*': 'admin'}, {'*': 'dictionary_all'}, {'get': 'dictionary_list'}, {'post': 'dictionary_create'},
                 {'put': 'dictionary_edit'},
                 {'delete': 'dictionary_delete'})
    queryset = DataDictionary.objects.all()
    serializer_class = DictSerializer
    filter_backends = (SearchFilter, OrderingFilter)
    search_fields = ('DictionaryCode', 'DictionaryValue')
    ordering_fields = ('id',)
    authentication_classes = (JSONWebTokenAuthentication,)
    permission_classes = (RbacPermission,)

    def list(self, request, *args, **kwargs):
        '''
        重写list方法
        '''
        queryset = self.filter_queryset(self.get_queryset())
        serializer = self.get_serializer(queryset, many=True)
        tree_dict = {}
        tree_data = []
        try:
            for item in serializer.data:
                tree_dict[item['dict_id']] = item
            for i in tree_dict:
                if tree_dict[i]['parent_id'] and tree_dict[i]['parent_id'] != "0":  # 顶级的父级是 0，所以跳过
                    pid = tree_dict[i]['parent_id']
                    parent = tree_dict[pid]
                    parent.setdefault('children', []).append(tree_dict[i])
                else:
                    tree_data.append(tree_dict[i])
            results = tree_data
        except Exception as e:
            logger.error("get dict list failed: \n%s" % traceback.format_exc())
            return CCAIResponse("获取字典列表失败", SERVER_ERROR)

        return CCAIResponse(results)

    def create(self, request, *args, **kwargs):
        '''添加数据POST'''
        try:
            data = req_operate_by_user(request)
            data['dict_id'] = uuid.uuid4().__str__()

            serializer = self.get_serializer(data=data)
            serializer.is_valid(raise_exception=True)
            self.perform_create(serializer)
            # headers = self.get_success_headers(serializer.data)
            # 操作记录
            # info = {
            #     "des": "添加数据字典",
            #     "detail": "字典id: " + data['dict_id'] + ", " + "字典名称: " + data["label"]
            # }
            # create_operation_history_log(request, info, OperationHistoryLog)
            return CCAIResponse(data="success")
        except Exception as e:
            logger.error("dict  create failed: \n%s" % traceback.format_exc())
            return CCAIResponse("create failed", SERVER_ERROR)


class DataDictionaryDetailTreeViewSet(CustomViewBase):
    '''
    字典详情树
    '''
    perms_map = ({'*': 'admin'}, {'*': 'dictionarydetail_all'}, {'get': 'dictionarydetail_list'},
                 {'post': 'dictionarydetail_create'}, {'put': 'dictionarydetail_edit'},
                 {'delete': 'dictionarydetail_delete'})
    queryset = DataDictionaryDetail.objects.all()
    serializer_class = DictDetailSerializer
    filter_backends = (SearchFilter, OrderingFilter)
    ordering_fields = ('id',)
    authentication_classes = (JSONWebTokenAuthentication,)
    permission_classes = (IsAuthenticated,)

    def list(self, request, *args, **kwargs):
        '''
        重写list方法
        '''
        parent_code = request.GET.get("parent_code")
        is_enabled = request.GET.get("is_enabled")
        label = request.GET.get("label")
        order = request.GET.get("order")
        sort = request.GET.get("ordering")
        order_by = 'id'
        if order and sort:
            if order == '1':
                order_by = sort
            else:
                order_by = '-' + sort

        tree_dict = {}
        tree_data = []
        dict_map = {}
        results = {}

        if parent_code:
            try:
                condtions = OrderedDict()
                if is_enabled:
                    condtions['IsEnabled'] = is_enabled
                if label:
                    condtions['FullName__contains'] = label
                if 'TechArea' == parent_code:
                    Tech = DataDictionary.objects.filter(DictionaryCode=parent_code).first()
                    condtions['DataDictionaryId'] = Tech.DataDictionaryId
                else:
                    condtions['ParentCode__istartswith'] = parent_code
                # condtions['ParentCode'] = parent_code

                queryset = DataDictionaryDetail.objects.filter(**condtions).order_by(order_by)
                serializer = self.get_serializer(queryset, many=True)
                for item in serializer.data:
                    dict_map[item['dict_code']] = item['label']

                for item in serializer.data:
                    tree_dict[item['dict_detail_id']] = item
                for i in tree_dict:
                    data = tree_dict[i]
                    parent_id = tree_dict[i]['parent_id']
                    parent_code = tree_dict[i]['parent_code']
                    if parent_id and parent_code in dict_map and parent_code != 'CusMYType_XSSRZZL':
                        pid = parent_id
                        parent = tree_dict[pid]
                        parent.setdefault('children', []).append(data)
                    else:
                        tree_data.append(data)
                if 'RND_COST' == request.GET.get("parent_code"):
                    results['list'] = sorted(tree_data, key=lambda x: distutils.version.LooseVersion(x['dict_code']), reverse=False)
                else:
                    results['list'] = tree_data

                results['map'] = dict_map
                return CCAIResponse(results)
            except Exception as e:
                logger.error("get dict detail tree failed: \n%s" % traceback.format_exc())
                return CCAIResponse("获取字典详情树失败", SERVER_ERROR)
        else:
            return CCAIResponse(PARAMS_ERR, SERVER_ERROR)

    def create(self, request, *args, **kwargs):
        '''添加数据POST'''
        try:
            data = req_operate_by_user(request)
            data['dict_detail_id'] = uuid.uuid4().__str__()

            serializer = self.get_serializer(data=data)
            serializer.is_valid(raise_exception=True)
            self.perform_create(serializer)
            # headers = self.get_success_headers(serializer.data)
            # 操作记录
            # info = {
            #     "des": "添加字典详情",
            #     "detail": "字典详情id: " + data['dict_detail_id'] + ", " + "字典详情名称: " + data["label"]
            # }
            # create_operation_history_log(request, info, OperationHistoryLog)
            return CCAIResponse(data="success")
        except Exception as e:
            logger.error("dict detail  create failed: \n%s" % traceback.format_exc())
            return CCAIResponse("create failed", SERVER_ERROR)

    # 获取地区字典接口 - 从redis从获取
    @action(methods=["get"], detail=False, permission_classes=[IsAuthenticated],
            url_path="getDictList", url_name="getDictList")
    def getDictList(self, request):
        parent_code = request.GET.get("parent_code")
        if parent_code:
            if parent_code == '':
                return CCAIResponse("parent_code参数不能为空!", status=status.HTTP_400_BAD_REQUEST)
            from django_redis import get_redis_connection
            startTime = datetime.datetime.now()
            # print(datetime.datetime.now())
            conn = get_redis_connection('default')
            # conn.hset('dicList', 'area', 1)
            # print(conn.hget('dicList', parent_code))
            str = conn.hget('dicList', parent_code)
            endTime = datetime.datetime.now()
            # print(datetime.datetime.now())
            # print((endTime - startTime).seconds)
            return CCAIResponse(data=str)
        else:
            return CCAIResponse("parent_code参数不能为空!", status=status.HTTP_400_BAD_REQUEST)

    # 设置地区字典接口 - 存储到redis中
    @action(methods=["get"], detail=False, permission_classes=[IsAuthenticated],
            url_path="setDictList", url_name="setDictList")
    def setDictList(self, request):

        parent_code = request.GET.get("parent_code")
        is_enabled = request.GET.get("is_enabled")
        label = request.GET.get("label")
        order = request.GET.get("order")
        sort = request.GET.get("ordering")
        order_by = 'id'
        if order and sort:
            if order == '1':
                order_by = sort
            else:
                order_by = '-' + sort

        tree_dict = {}
        tree_data = []
        dict_map = {}
        results = {}

        if parent_code:
            try:
                condtions = OrderedDict()
                if is_enabled:
                    condtions['IsEnabled'] = is_enabled
                if label:
                    condtions['FullName__contains'] = label
                condtions['ParentCode__istartswith'] = parent_code
                # condtions['ParentCode'] = parent_code

                queryset = DataDictionaryDetail.objects.filter(**condtions).order_by(order_by)
                serializer = self.get_serializer(queryset, many=True)
                for item in serializer.data:
                    dict_map[item['dict_code']] = item['label']
                for item in serializer.data:
                    tree_dict[item['dict_detail_id']] = item
                for i in tree_dict:
                    if tree_dict[i]['parent_id'] and tree_dict[i]['parent_code'] in dict_map:
                        pid = tree_dict[i]['parent_id']
                        parent = tree_dict[pid]
                        parent.setdefault('children', []).append(tree_dict[i])
                    else:
                        tree_data.append(tree_dict[i])
                results['list'] = tree_data

                results['map'] = dict_map
                from django_redis import get_redis_connection
                conn = get_redis_connection('default')
                str = json.dumps(results).encode('utf-8').decode('unicode_escape')
                # print(str)
                conn.hset('dicList', parent_code, str)
                return CCAIResponse(data="success")
            except Exception as e:
                logger.error("get dict detail tree failed: \n%s" % traceback.format_exc())
                return CCAIResponse("获取字典详情树失败", SERVER_ERROR)


class DataDictionaryDetailListViewSet(ModelViewSet):
    '''
    字典详情列表
    '''
    perms_map = ({'*': 'admin'}, {'*': 'dictionarydetail_list_all'}, {'get': 'dictionarydetail_list_list'},
                 {'post': 'dictionarydetail_list_create'}, {'put': 'dictionarydetail_listlist_edit'},
                 {'delete': 'dictionarydetail_list_delete'})
    queryset = DataDictionaryDetail.objects.all()
    serializer_class = DictDetailSerializer
    filter_backends = (SearchFilter, OrderingFilter)
    ordering_fields = ('id',)
    authentication_classes = (JSONWebTokenAuthentication,)
    permission_classes = (IsAuthenticated,)

    def list(self, request, *args, **kwargs):
        '''
        重写list方法
        '''
        parent_code = request.GET.get("parent_code")
        is_enabled = request.GET.get("is_enabled")
        label = request.GET.get("label")
        order = request.GET.get("order")
        sort = request.GET.get("ordering")
        select_all = request.GET.get("select_all")

        dict_list = []
        dict_map = {}
        dict_value_map = {}
        results = {}

        order_by = '-id'
        if order and sort:
            if order == '1':
                order_by = sort
            else:
                order_by = '-' + sort

        if parent_code:
            try:
                condtions = OrderedDict()
                if label:
                    condtions['FullName__contains'] = label
                if is_enabled:
                    condtions['IsEnabled'] = is_enabled
                if select_all:
                    condtions['ParentCode__istartswith'] = parent_code
                else:
                    condtions['ParentCode'] = parent_code

                queryset = DataDictionaryDetail.objects.filter(**condtions).order_by(order_by)
                serializer = self.get_serializer(queryset, many=True)

                for item in serializer.data:
                    item['children'] = []
                    item['hasChildren'] = True
                    dict_list.append(item)
                    dict_map[item['dict_code']] = item['label']
                    dict_value_map[item['dict_val']] = item['label']

                results['list'] = dict_list

                results['map'] = dict_map
                results['value_map'] = dict_value_map
                return CCAIResponse(results)
            except Exception as e:
                logger.error("get dict detail tree failed: \n%s" % traceback.format_exc())
                return CCAIResponse("获取字典详情列表失败", SERVER_ERROR)
        else:
            return CCAIResponse(PARAMS_ERR, SERVER_ERROR)


# 获取智云那边的区域
class DataDictionaryDetailTreeViewSetZY(CustomViewBase):
    '''
    字典详情树
    '''
    perms_map = ({'*': 'admin'}, {'*': 'dictionarydetail_all'}, {'get': 'dictionarydetail_list'},
                 {'post': 'dictionarydetail_create'}, {'put': 'dictionarydetail_edit'},
                 {'delete': 'dictionarydetail_delete'})
    queryset = DataDictionaryDetailZY.objects.all()
    serializer_class = DictDetailSerializer
    filter_backends = (SearchFilter, OrderingFilter)
    ordering_fields = ('id',)
    authentication_classes = (JSONWebTokenAuthentication,)
    permission_classes = (IsAuthenticated,)

    def list(self, request, *args, **kwargs):
        '''
        重写list方法
        '''
        parent_code = request.GET.get("parent_code")
        is_enabled = request.GET.get("is_enabled")
        label = request.GET.get("label")
        order = request.GET.get("order")
        sort = request.GET.get("ordering")
        order_by = 'id'
        if order and sort:
            if order == '1':
                order_by = sort
            else:
                order_by = '-' + sort

        tree_dict = {}
        tree_data = []
        dict_map = {}
        results = {}

        if parent_code:
            try:
                condtions = OrderedDict()
                if is_enabled:
                    condtions['IsEnabled'] = is_enabled
                if label:
                    condtions['FullName__contains'] = label
                condtions['ParentCode__istartswith'] = parent_code
                # condtions['ParentCode'] = parent_code

                queryset = DataDictionaryDetailZY.objects.filter(**condtions).order_by(order_by)
                serializer = self.get_serializer(queryset, many=True)
                for item in serializer.data:
                    dict_map[item['dict_code']] = item['label']

                for item in serializer.data:
                    tree_dict[item['dict_detail_id']] = item
                for i in tree_dict:
                    data = tree_dict[i]
                    parent_id = tree_dict[i]['parent_id']
                    parent_code = tree_dict[i]['parent_code']
                    if parent_id and parent_code in dict_map and parent_code != 'CusMYType_XSSRZZL':
                        pid = parent_id
                        parent = tree_dict[pid]
                        parent.setdefault('children', []).append(data)
                    else:
                        tree_data.append(data)
                results['list'] = tree_data

                results['map'] = dict_map
                return CCAIResponse(results)
            except Exception as e:
                logger.error("get dict detail tree failed: \n%s" % traceback.format_exc())
                return CCAIResponse("获取字典详情树失败", SERVER_ERROR)
        else:
            return CCAIResponse(PARAMS_ERR, SERVER_ERROR)