bigdataAPI/apps/QxzApp/views.py

import json
import os

from django.shortcuts import render

from rest_framework.response import Response
from rest_framework import status
from rest_framework.generics import GenericAPIView
from rest_framework.views import APIView
from django.db.models import ObjectDoesNotExist, Q
from utils.JWTAuthentication_diy import MyJWTAuthentication
from utils.permissions import QxzLoginPermission
from utils.paginations import CustomPagination
from .serializers import QxzTypeConfigAddSerializer, QxzTypeConfigModifySerializer, QxzTypeConfigDeleteSerializer, \
    QxzTypeConfigListSerializer, QxzElementAddSerializer, QxzElementModifySerializer, QxzElementDeleteSerializer, \
    QxzElementListSerializer, QxzDeviceAddSerializer, QxzDeviceListSerializer
from .models import QxzTypeConfigModel, QxzElementConfigModel, QxzDeviceConfigModel
import jwt

class UserLoginView(APIView):
    def post(self, request, *args, **kwargs):
        """用户登录接口"""
        username = request.data.get('username')
        password = request.data.get('password')
        print(username, password)
        if username == "admin" and password == "hnyfqxz":
            token = jwt.encode({"username": username, "password": password}, "secret", algorithm="HS256").decode('utf-8')

            request.session["hnyfqxz"] = token
            request.session.save()
            return Response({'msg': "登录成功", "token": token}, status=200)
        return Response({'msg': "登录失败，用户名或密码不正确"}, status=400)


class UserLogoutView(APIView):
    permission_classes = [QxzLoginPermission]

    def post(self, request, *args, **kwargs):
        """用户退出接口"""
        try:
            del request.session['hnyfqxz']
            request.session.save()
        except Exception as e:
            pass
        return Response({'msg': "退出成功"}, status=200)


class QxzTypeConfigAddView(GenericAPIView):
    # authentication_classes = [MyJWTAuthentication]
    permission_classes = [QxzLoginPermission]
    serializer_class = QxzTypeConfigAddSerializer
    queryset = QxzTypeConfigModel.objects.all()

    def post(self, request, *args, **kwargs):
        """新增要素类型配置信息接口"""
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        data = serializer.validated_data
        eNum = data['eNum']
        cName = data['cName']
        units = data['units']

        queryset = self.get_queryset()
        if queryset.filter(eNum=eNum):
            return Response(data={"msg": "eNum已存在", "data": ""}, status=status.HTTP_400_BAD_REQUEST)

        if queryset.filter(cName=cName):
            return Response(data={"msg": "要素类型名称已存在", "data": ""}, status=status.HTTP_400_BAD_REQUEST)

        qsc = QxzTypeConfigModel()
        qsc.eNum = eNum
        qsc.cName = cName
        qsc.units = units
        qsc.save()

        return Response(data={"msg": "要素类型配置创建成功", "data": ""}, status=status.HTTP_200_OK)


class QxzTypeConfigModifyView(GenericAPIView):
    # authentication_classes = [MyJWTAuthentication]
    permission_classes = [QxzLoginPermission]
    serializer_class = QxzTypeConfigModifySerializer
    queryset = QxzTypeConfigModel.objects.all()

    def post(self, request, *args, **kwargs):
        """修改要素类型配置信息接口"""
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        data = serializer.validated_data
        qsc_id = data['qsc_id']
        cName = data['cName']
        units = data['units']

        queryset = self.get_queryset().exclude(id=qsc_id)
        # if queryset.filter(eNum=eNum):
        #     return Response(data={"msg": "eNum已存在", "data": ""}, status=status.HTTP_400_BAD_REQUEST)

        if queryset.filter(cName=cName):
            return Response(data={"msg": "要素类型名称已存在", "data": ""}, status=status.HTTP_400_BAD_REQUEST)

        try:
            qsc = QxzTypeConfigModel.objects.get(id=qsc_id)
        except ObjectDoesNotExist as e:
            return Response(data={"msg": "要素类型配置不存在", "data": ""}, status=status.HTTP_400_BAD_REQUEST)

        if qsc.is_default:
            return Response(data={"msg": "默认要素类型不允许修改", "data": ""}, status=status.HTTP_400_BAD_REQUEST)

        qsc.cName = cName
        qsc.units = units
        qsc.save()

        return Response(data={"msg": "要素类型配置修改成功", "data": ""}, status=status.HTTP_200_OK)


class QxzTypeConfigDeleteView(GenericAPIView):
    # authentication_classes = [MyJWTAuthentication]
    permission_classes = [QxzLoginPermission]
    serializer_class = QxzTypeConfigDeleteSerializer
    queryset = QxzTypeConfigModel.objects.all()

    def post(self, request, *args, **kwargs):
        """删除要素类型配置信息接口"""
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        data = serializer.validated_data
        qsc_id = data['qsc_id']

        try:
            qsc = QxzTypeConfigModel.objects.get(id=qsc_id)
        except ObjectDoesNotExist as e:
            return Response(data={"msg": "要素类型配置不存在", "data": ""}, status=status.HTTP_400_BAD_REQUEST)
        if qsc.is_default:
            return Response(data={"msg": "默认要素类型不允许删除", "data": ""}, status=status.HTTP_400_BAD_REQUEST)

        QxzElementConfigModel.objects.filter(eNum=qsc.eNum).delete()
        qsc.delete()

        return Response(data={"msg": "要素类型配置删除成功", "data": ""}, status=status.HTTP_200_OK)


class QxzTypeConfigListView(GenericAPIView):
    # authentication_classes = [MyJWTAuthentication]
    permission_classes = [QxzLoginPermission]
    pagination_class = CustomPagination
    serializer_class = QxzTypeConfigListSerializer
    queryset = QxzTypeConfigModel.objects.all().order_by('-id')

    def post(self, request, *args, **kwargs):
        """获取要素类型配置列表接口"""
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        data = serializer.validated_data
        cName = data.get('cName', '').strip()
        is_default = data.get('is_default')
        queryset = self.get_queryset()
        if cName:
            queryset = queryset.filter(cName__contains=cName)
        if is_default in [True, False]:
            queryset = queryset.filter(is_default=is_default)

        page = self.paginate_queryset(queryset)
        serializer = self.get_serializer(page, many=True)
        return self.get_paginated_response(serializer.data)


class QxzElementAddView(GenericAPIView):
    # authentication_classes = [MyJWTAuthentication]
    permission_classes = [QxzLoginPermission]
    serializer_class = QxzElementAddSerializer
    queryset = QxzElementConfigModel.objects.all().order_by('-id')

    def post(self, request, *args, **kwargs):
        """新增要素信息接口"""
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        data = serializer.validated_data

        eName = data['eName']
        cName = data['cName']
        eNum = data['eNum']
        units = data['units']
        is_standard = data['is_standard']
        sensor_addr = data['sensor_addr']
        start_reg = data['start_reg']
        count = data['count']
        d_pos = data['d_pos']
        d_length = data['d_length']
        factor = data['factor']
        dotled_s = data['dotled_s']

        queryset = self.get_queryset()
        if queryset.filter(eName=eName):
            return Response(data={"msg": "英文名称不能重复", "data": ""}, status=status.HTTP_400_BAD_REQUEST)

        try:
            t = QxzTypeConfigModel.objects.get(eNum=eNum)
            if t.units != units:
                raise Exception()
        except Exception as e:
            return Response(data={"msg": "要素类型异常", "data": ""}, status=status.HTTP_400_BAD_REQUEST)

        qec = QxzElementConfigModel()
        qec.eName = eName
        qec.cName = cName
        qec.units = units
        qec.eNum = eNum
        qec.is_standard = is_standard
        qec.sensor_addr = sensor_addr
        qec.start_reg = start_reg
        qec.count = count
        qec.d_pos = d_pos
        qec.d_length = d_length
        qec.factor = factor
        qec.dotled_s = dotled_s
        qec.save()

        return Response(data={"msg": "创建要素成功", "data": ""}, status=status.HTTP_200_OK)


class QxzElementModifyView(GenericAPIView):
    # authentication_classes = [MyJWTAuthentication]
    permission_classes = [QxzLoginPermission]
    serializer_class = QxzElementModifySerializer
    queryset = QxzElementConfigModel.objects.all().order_by('-id')

    def post(self, request, *args, **kwargs):
        """修改要素信息接口"""
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        data = serializer.validated_data
        qec_id = data['qec_id']
        eName = data['eName']
        cName = data['cName']
        is_standard = data['is_standard']
        sensor_addr = data['sensor_addr']
        start_reg = data['start_reg']
        count = data['count']
        d_pos = data['d_pos']
        d_length = data['d_length']
        factor = data['factor']
        dotled_s = data['dotled_s']

        try:
            qec = self.get_queryset().get(id=qec_id)
        except ObjectDoesNotExist as e:
            return Response(data={"msg": "要素不存在", "data": ""}, status=status.HTTP_400_BAD_REQUEST)

        queryset = self.get_queryset().exclude(id=qec_id)
        if queryset.filter(eName=eName):
            return Response(data={"msg": "英文名称不能重复", "data": ""}, status=status.HTTP_400_BAD_REQUEST)

        qec.eName = eName
        qec.cName = cName
        qec.is_standard = is_standard
        qec.sensor_addr = sensor_addr
        qec.start_reg = start_reg
        qec.count = count
        qec.d_pos = d_pos
        qec.d_length = d_length
        qec.factor = factor
        qec.dotled_s = dotled_s
        qec.save()

        return Response(data={"msg": "修改要素成功", "data": ""}, status=status.HTTP_200_OK)


class QxzElementDeleteView(GenericAPIView):
    # authentication_classes = [MyJWTAuthentication]
    permission_classes = [QxzLoginPermission]
    serializer_class = QxzElementDeleteSerializer
    queryset = QxzElementConfigModel.objects.all().order_by('-id')

    def post(self, request, *args, **kwargs):
        """删除要素信息接口"""
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        data = serializer.validated_data
        qec_id = data['qec_id']

        try:
            qec = self.get_queryset().get(id=qec_id)
        except ObjectDoesNotExist as e:
            return Response(data={"msg": "要素不存在", "data": ""}, status=status.HTTP_400_BAD_REQUEST)
        qec.delete()

        return Response(data={"msg": "删除要素成功", "data": ""}, status=status.HTTP_200_OK)


class QxzElementListView(GenericAPIView):
    # authentication_classes = [MyJWTAuthentication]
    permission_classes = [QxzLoginPermission]
    pagination_class = CustomPagination
    serializer_class = QxzElementListSerializer
    queryset = QxzElementConfigModel.objects.all().order_by('-id')

    def post(self, request, *args, **kwargs):
        """获取要素列表接口"""
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        data = serializer.validated_data
        ele_name = data.get('ele_name', '')
        is_standard = data.get('is_standard')

        queryset = self.get_queryset()
        if ele_name:
            queryset = queryset.filter(Q(cName__contains=ele_name) | Q(eName__contains=ele_name))
        if is_standard is not None:
            queryset = queryset.filter(is_standard=is_standard)
        page = self.paginate_queryset(queryset)
        serializer = self.get_serializer(page, many=True)
        return self.get_paginated_response(serializer.data)


class QxzDeviceAddView(GenericAPIView):
    # authentication_classes = [MyJWTAuthentication]
    permission_classes = [QxzLoginPermission]
    serializer_class = QxzDeviceAddSerializer
    queryset = QxzDeviceConfigModel.objects.all().order_by('-id')

    def post(self, request, *args, **kwargs):
        """创建气象配置信息接口"""
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        data = serializer.validated_data

        device_id = data['device_id']
        order_num = data['order_num']
        device_type = data['device_type']
        is_standard = data['is_standard']
        ele_cnf = data['ele_cnf']
        led_cnf = data['led_cnf']
        file = request.FILES.get('file')

        queryset = self.get_queryset().filter(device_id=device_id)
        if queryset:
            return Response(data={"msg": "设备配置已存在，请勿重复创建", "data": ""}, status=status.HTTP_400_BAD_REQUEST)

        base_dir = f"/data/qxz/element/config/{device_id}"
        if not os.path.isdir(base_dir):
            os.makedirs(base_dir)

        if file:
            file_name = f"{device_id}.rbl"
            file_path = os.path.join(base_dir, file_name)
            with open(file_path, 'wb') as wb:
                for chunk in file.chunks():
                    wb.write(chunk)

        config_sh_dict = {
            "config": "ch_conf",
            "ext": {
                "App_mod": device_type,
            }
        }

        e_name_list = list(ele_cnf.values())
        e_value_dict = {}
        for k, v in ele_cnf.items():
            config_sh_dict["ext"][k] = v
            e_value_dict[v] = k

        sh_file_path = os.path.join(base_dir, 'configch.c')
        with open(sh_file_path, 'w') as fw:
            json.dump(config_sh_dict, fw, ensure_ascii=False)

        if led_cnf:
            led_dict = {
                "cmd": "dotled_para",
                "ext": led_cnf
            }
            sh_file_path = os.path.join(base_dir, 'configled.c')
            with open(sh_file_path, 'w') as fw:
                json.dump(led_dict, fw, ensure_ascii=False)

        c_name_list = []
        config_list = []
        queryset = QxzElementConfigModel.objects.filter(eName__in=e_name_list)
        num = 0
        for item in queryset:
            eName = item.eName
            cName = f"{item.cName}:"
            ekey = e_value_dict[eName]
            c_name_list.append(item.cName)
            config_list.append({
                'num': num, 'sensor_addr': item.sensor_addr, 'start_reg': item.start_reg, 'count': item.count,
                'd_pos': item.d_pos, 'factor': item.factor, 'd_length': item.d_length, 'err_cnt': 0, 'online_flag': 0,
                'value': 0, 'cName': cName, 'units': item.units, 'eName': item.eName, 'eValue': '0',
                'eKey': ekey, 'eNum': item.eNum, 'dotled_s': item.dotled_s
            })
            num += 1

        c_num = 1
        for i in range(0, len(config_list), 15):
            c_list = config_list[i: i+15]
            c_path = os.path.join(base_dir, f'config0{c_num}.c')
            with open(c_path, 'w') as fw:
                c_data = {
                    "element": c_list
                }
                json.dump(c_data, fw, ensure_ascii=False)
            c_num += 1

        qdc = QxzDeviceConfigModel()
        qdc.device_id = device_id
        qdc.order_num = order_num
        qdc.device_type = device_type
        qdc.is_standard = is_standard
        qdc.ele_cnf = json.dumps(ele_cnf, ensure_ascii=False)
        qdc.led_cnf = json.dumps(led_cnf, ensure_ascii=False)
        qdc.ele_name = json.dumps(c_name_list, ensure_ascii=False)
        qdc.save()

        return Response(data={"msg": "创建设备配置成功", "data": ""}, status=status.HTTP_200_OK)


class QxzDeviceModifyView(GenericAPIView):
    # authentication_classes = [MyJWTAuthentication]
    permission_classes = [QxzLoginPermission]
    serializer_class = QxzDeviceAddSerializer
    queryset = QxzDeviceConfigModel.objects.all().order_by('-id')

    def post(self, request, *args, **kwargs):
        """修改气象配置信息接口"""
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        data = serializer.validated_data

        device_id = data['device_id']
        order_num = data['order_num']
        device_type = data['device_type']
        is_standard = data['is_standard']
        ele_cnf = data['ele_cnf']
        led_cnf = data['led_cnf']
        file = request.FILES.get('file')

        try:
            qdc = self.get_queryset().get(device_id=device_id)
        except ObjectDoesNotExist as e:
            return Response(data={"msg": "设备配置不存在", "data": ""}, status=status.HTTP_400_BAD_REQUEST)

        base_dir = f"/data/qxz/element/config/{device_id}"
        if not os.path.isdir(base_dir):
            os.makedirs(base_dir)

        if file:
            file_name = f"{device_id}.rbl"
            file_path = os.path.join(base_dir, file_name)
            with open(file_path, 'wb') as wb:
                for chunk in file.chunks():
                    wb.write(chunk)

        config_sh_dict = {
            "config": "ch_conf",
            "ext": {
                "App_mod": device_type,
            }
        }

        e_name_list = list(ele_cnf.values())
        e_value_dict = {}
        for k, v in ele_cnf.items():
            config_sh_dict["ext"][k] = v
            e_value_dict[v] = k

        sh_file_path = os.path.join(base_dir, 'configch.c')
        with open(sh_file_path, 'w') as fw:
            json.dump(config_sh_dict, fw, ensure_ascii=False)

        if led_cnf:
            led_dict = {
                "cmd": "dotled_para",
                "ext": led_cnf
            }
            sh_file_path = os.path.join(base_dir, 'configled.c')
            with open(sh_file_path, 'w') as fw:
                json.dump(led_dict, fw, ensure_ascii=False)

        c_name_list = []
        config_list = []
        queryset = QxzElementConfigModel.objects.filter(eName__in=e_name_list)
        num = 0
        for item in queryset:
            eName = item.eName
            cName = f"{item.cName}:"
            ekey = e_value_dict[eName]
            c_name_list.append(item.cName)
            config_list.append({
                'num': num, 'sensor_addr': item.sensor_addr, 'start_reg': item.start_reg, 'count': item.count,
                'd_pos': item.d_pos, 'factor': item.factor, 'd_length': item.d_length, 'err_cnt': 0, 'online_flag': 0,
                'value': 0, 'cName': cName, 'units': item.units, 'eName': item.eName, 'eValue': '0',
                'eKey': ekey, 'eNum': item.eNum, 'dotled_s': item.dotled_s
            })
            num += 1

        c_num = 1
        for i in range(0, len(config_list), 15):
            c_list = config_list[i: i+15]
            c_path = os.path.join(base_dir, f'config0{c_num}.c')
            with open(c_path, 'w') as fw:
                c_data = {
                    "element": c_list
                }
                json.dump(c_data, fw, ensure_ascii=False)
            c_num += 1

        qdc.order_num = order_num
        qdc.device_type = device_type
        qdc.is_standard = is_standard
        qdc.ele_cnf = json.dumps(ele_cnf, ensure_ascii=False)
        qdc.led_cnf = json.dumps(led_cnf, ensure_ascii=False)
        qdc.ele_name = json.dumps(c_name_list, ensure_ascii=False)
        qdc.save()

        return Response(data={"msg": "修改设备配置成功", "data": ""}, status=status.HTTP_200_OK)


class QxzDeviceListView(GenericAPIView):
    # authentication_classes = [MyJWTAuthentication]
    permission_classes = [QxzLoginPermission]
    pagination_class = CustomPagination
    serializer_class = QxzDeviceListSerializer
    queryset = QxzDeviceConfigModel.objects.all().order_by('-id')

    def post(self, request, *args, **kwargs):
        """获取气象配置列表接口"""
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        data = serializer.validated_data

        device_id = data.get('device_id')
        queryset = self.get_queryset()
        if device_id:
            queryset = queryset.filter(Q(device_id__contains=device_id) | Q(order_num__contains=device_id))
        page = self.paginate_queryset(queryset)
        serializer = self.get_serializer(page, many=True)
        return self.get_paginated_response(serializer.data)



def init_data():
    data = {
        '101': ['101', '大气温度', '℃'],
        '102': ['102', '大气湿度', '%RH'],
        '103': ['103', '模拟气压', 'hpa'],
        '104': ['104', '雨量', 'mm'],
        '105': ['105', '简易总辐射', 'w/㎡'],
        '106': ['106', '土壤温度', '℃'],
        '107': ['107', '土壤湿度', '%RH'],
        '108': ['108', '风速', 'm/s'],
        '109': ['109', '风向', '°'],
        '110': ['110', '蒸发', 'mm'],
        '111': ['111', '雪量', 'mm'],
        '112': ['112', '照度', 'LUX'],
        '113': ['113', '日照时数', 'h'],
        '114': ['114', '光合', 'w/㎡'],
        '115': ['115', '雨量累计', 'mm'],
        '116': ['116', '辐射累计', 'MJ/㎡'],
        '117': ['117', '有无雨雪', ''],
        '118': ['118', '噪声', '分贝'],
        '119': ['119', '水位', 'cm'],
        '120': ['120', '二氧化碳', 'PPM'],
        '121': ['121', '曝辐量', 'cal/cm2'],
        '122': ['122', '液位', 'mm'],
        '123': ['123', '光合有效辐射', 'W/m2'],
        '124': ['124', '电压', 'V'],
        '125': ['125', '紫外线', 'w/㎡'],
        '126': ['126', '粉尘', 'ug/m3'],
        '127': ['127', '数字气压', 'hpa'],
        '128': ['128', 'PH值', ''],
        '129': ['129', '最大风速', 'm/s'],
        '130': ['130', '平均风速', 'm/s'],
        '131': ['131', '经度', '°'],
        '132': ['132', '纬度', '°'],
        '133': ['133', '海拔高度', 'm'],
        '134': ['134', 'TBQ总辐射', 'W/m2'],
        '135': ['135', '直接辐射', 'W/m2'],
        '136': ['136', '散射辐射', 'W/m2'],
        '138': ['138', '紫外辐射', 'W/m2'],
        '139': ['139', '贴片温度', '℃'],
        '140': ['140', '露点温度', '℃'],
        '141': ['141', '一氧化碳', 'PPM'],
        '142': ['142', '电流', 'mA'],
        '143': ['143', '超声波风速', 'm/s'],
        '144': ['144', '水温', '℃'],
        '145': ['145', 'PM2.5', 'ug/m3'],
        '146': ['146', 'PM10', 'ug/m3'],
        '147': ['147', 'PH', ''],
        '148': ['148', '溶解氧', 'mg/l'],
        '149': ['149', '氨氮', 'mg/l'],
        '150': ['150', '电导率', 'mS/cm'],
        '151': ['151', '浊度', 'NTU'],
        '152': ['152', '能见度', 'm'],
        '153': ['153', '氨气', 'ppm'],
        '154': ['154', '盐分', 'mg/L'],
        '155': ['155', '氮', 'mg/kg'],
        '156': ['156', '磷', 'mg/kg'],
        '157': ['157', '钾', 'mg/kg'],
        '158': ['158', '热通量', 'W/m2'],
        '159': ['159', '叶面温度', '℃'],
        '160': ['160', '叶面湿度', '%RH'],
        '161': ['161', '茎秆', 'mm'],
        '162': ['162', '负氧离子', '个/cm3'],
        '201': ['201', '土壤水势', 'KPA'],
        '202': ['202', '露点温度', '℃'],
        '203': ['203', 'PM100', 'ug/m3'],
        '208': ['208', '照度', 'Lux'],
        '211': ['211', '电导率', 'uS/cm'],
        '212': ['212', '净辐射', 'W/m2'],
        '213': ['213', '流速', 'm/s'],
        '214': ['214', '空高', 'm'],
        '215': ['215', '瞬时流量', 'm3/s'],
        '216': ['216', '累积流量', 'm3'],
        '217': ['217', '热通量累计', 'MJ/m2'],
        '218': ['218', '果实直径', 'mm'],
        '219': ['219', '硫化氢', 'ppm'],
        '220': ['220', '氧气', 'ppm'],
        '221': ['221', '臭氧', 'ppm'],
        '222': ['222', '茎流', 'ml/h'],
        '223': ['223', 'COD', 'mg/L'],
        '224': ['224', '信号强度', 'dBm'],
        '225': ['225', '盐度', 'PSU'],
        '226': ['226', 'ORP', 'mV'],
        '227': ['227', '光量子', 'W/m2'],
        '228': ['228', '水质pH', ' '],
        '229': ['229', '氨氮', 'ppm'],
        '230': ['230', '水位', 'm'],
        '251': ['251', '湿球温度', '℃'],
        '252': ['252', '最大阵风', 'm/s'],
        '255': ['255', '土壤张力', 'kpa']
    }

    for item in data.values():
        c = QxzTypeConfigModel()
        c.eNum = item[0]
        c.cName = item[1]
        c.units = item[2]
        c.is_default = True
        c.save()