ASJServer/Controller/EquipmentInfo.py

from django.shortcuts import HttpResponse
from django.views.generic.base import View
from django.utils.decorators import method_decorator
from django.views.decorators.csrf import csrf_exempt
from Service.TokenManager import JSONTokenManager
from Service.ModelService import ModelService
from Service.CommonService import CommonService
from Model.models import Equipment_Info, Device_Info
import traceback
from Service.ResponseService import ResponseJSON

'''
http://192.168.136.45:8077/equipment/info?token=test&devUid=2N1K3LE78TYJ38CE111A&Channel=0&eventType=1&eventTime=1234567890&operation=add&alarm=21342134&receiveTime=1234567891234567
http://192.168.136.45:8077/equipment/info?token=test&operation=query&page=1&line=10
http://192.168.136.45:8077/equipment/info?token=test&operation      =delete&devUid=UKPAH63V23U4ZHEB111A&id=5&id=6&id=7
http://192.168.136.40:8077/equipment/info?token=test&operation=update&devUid=UKPAH63V23U4ZHEB111A&id=3&id=4
http://192.168.136.45:8077/equipment/info?token=test&operation=queryByAdmin&page=1&line=10&userID_id=151547867345163613800138001
http://192.168.136.45:8077/equipment/info?token=test&operation=deleteByAdmin&id=5&id=6&id=7

'''


class EquipmentInfo(View):
    @method_decorator(csrf_exempt)
    def dispatch(self, *args, **kwargs):
        return super(EquipmentInfo, self).dispatch(*args, **kwargs)

    def get(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        return self.validation(request_dict=request.GET)

    def post(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        return self.validation(request_dict=request.POST)

    def validation(self, request_dict, *args, **kwargs):
        token = request_dict.get('token', None)
        if token is not None:
            tokenManager = JSONTokenManager()
            error_code = tokenManager.verify_AToken(token)
            if error_code == 0:
                userID = tokenManager.accessDict.get('userID', None)
                operation = request_dict.get('operation', None)
                param_flag = CommonService.get_param_flag(data=[userID, operation])
                if param_flag is True:
                    if operation == 'query':
                        return self.query_info(request_dict=request_dict, userID=userID)
                    elif operation == 'add':
                        return self.add_info(request_dict=request_dict, userID=userID)
                    elif operation == 'update':
                        return self.update_info(request_dict=request_dict, userID=userID)
                    elif operation == 'delete':
                        return self.delete_info(request_dict=request_dict, userID=userID)
                    elif operation == 'findByTime':
                        return self.findByTime_info(request_dict=request_dict, userID=userID)
                    if operation == 'queryByAdmin':
                        return self.query_all_info(request_dict=request_dict, userID=userID)
                    if operation == 'deleteByAdmin':
                        return self.delete_by_admin(request_dict=request_dict, userID=userID)
                return ResponseJSON(444)
            else:
                return HttpResponse(tokenManager.errorCodeInfo(error_code))
        else:
            return ResponseJSON(311)

    def add_info(self, request_dict, userID):
        devUid = request_dict.get('devUid', None)
        Channel = request_dict.get('Channel', None)
        eventType = request_dict.get('eventType', None)
        eventTime = request_dict.get('eventTime', None)
        receiveTime = request_dict.get('receiveTime', None)
        alarm = request_dict.get('alarm', None)
        param_flag = CommonService.get_param_flag(data=[devUid, Channel, eventType, eventTime, alarm, receiveTime])
        if param_flag is True:
            own_device = ModelService.check_own_device(userID, UID=devUid)
            if own_device is True:
                try:
                    device_info = Device_Info.objects.filter(UID=devUid,userID_id=userID)
                except Exception:
                    errorInfo = traceback.format_exc()
                    print(errorInfo)
                    return ResponseJSON(424, errorInfo)
                else:
                    if device_info.exists():
                        try:
                            equipment_info = Equipment_Info(
                                devUid=devUid,
                                Channel=Channel,
                                eventType=eventType,
                                receiveTime=receiveTime,
                                eventTime=eventTime,
                                viewPwd=device_info[0].View_Password,
                                devNickName=device_info[0].NickName,
                                viewAccont=device_info[0].View_Account,
                                alarm=alarm
                            )
                            equipment_info.save()
                        except Exception:
                            errorInfo = traceback.format_exc()
                            print(errorInfo)
                            return ResponseJSON(500, errorInfo)
                        else:
                            if equipment_info.id:
                                return ResponseJSON(0, {'infoID': equipment_info.id,
                                                        'devUid': devUid,
                                                        'Channel': Channel,
                                                        'alarm': alarm,
                                                        'eventType': eventType,
                                                        'eventTime': eventTime,
                                                        'receiveTime': receiveTime,
                                                        'viewPwd': equipment_info.viewPwd,
                                                        'devNickName': equipment_info.devNickName,
                                                        'viewAccont': equipment_info.viewAccont})
                            else:
                                return ResponseJSON(500)
                    else:
                        return ResponseJSON(14)
            else:
                return ResponseJSON(14)
        return ResponseJSON(444)

    def query_info(self, request_dict, userID):
        page = int(request_dict.get('page', None))
        line = int(request_dict.get('line', None))
        param_flag = CommonService.get_param_flag(data=[page, line])
        if param_flag is True:
            try:
                uid_list = Device_Info.objects.filter(userID_id=userID).values_list('UID', flat=True)
            except Exception:
                pass
            if len(uid_list):
                equipment_info_queryset = Equipment_Info.objects.filter(devUid__in=uid_list).order_by(
                    '-id')
                if equipment_info_queryset.exists():
                    equipment_info_count = equipment_info_queryset.count()
                    equipment_info_res = equipment_info_queryset[(page - 1) * line:page * line]
                    send_json = CommonService.query_set_to_dict(equipment_info_res)
                    send_json['count'] = equipment_info_count
                    return ResponseJSON(0, send_json)
            return ResponseJSON(0, {'datas': [], 'count': 0})
        else:
            return ResponseJSON(444)

    def update_info(self, request_dict, userID):
        id_list = request_dict.getlist('id[]', None)
        if id_list is None or len(id_list) < 1:
            id_list = request_dict.getlist('id', None)
        param_flag = CommonService.get_param_flag(data=[id_list])
        if param_flag is True:
            count = 0
            for id in id_list:
                eq = Equipment_Info.objects.filter(id=id)
                if eq.exists():
                    own_dev = ModelService.check_own_device(userID, eq[0].devUid)
                    if own_dev is True:
                        count += 1
                        eq.update(status=1)
            return ResponseJSON(0,{'update_success': count})
        else:
            return ResponseJSON(444)

    def delete_info(self, request_dict, userID):
        id_list = request_dict.getlist('id[]', None)
        if id_list is None or len(id_list) < 1:
            id_list = request_dict.getlist('id', None)
        param_flag = CommonService.get_param_flag(data=[id_list])
        if param_flag is True:
            try:
                for id in id_list:
                    eq = Equipment_Info.objects.filter(id=id)
                    if eq.exists():
                        own_dev = ModelService.check_own_device(userID, eq[0].devUid)
                        if own_dev is True:
                            eq.delete()
            except Exception as e:
                errorInfo = traceback.format_exc()
                print(errorInfo)
                return ResponseJSON(424,repr(e))
            else:
                return ResponseJSON(0)
        else:
            return ResponseJSON(444)

    def findByTime_info(self, request_dict, userID):
        startTime = request_dict.get('startTime')
        endTime = request_dict.get('endTime')
        page = int(request_dict.get('page', None))
        line = int(request_dict.get('line', None))
        param_flag = CommonService.get_param_flag(data=[startTime, endTime, page, line])
        if param_flag is True:
            uid_list = Device_Info.objects.filter(userID_id=userID).values_list('UID', flat=True)
            if len(uid_list):
                equipment_info_queryset = Equipment_Info.objects.filter(devUid__in=uid_list,
                                                                        eventTime__range=(startTime, endTime)).order_by(
                    '-id')
                if equipment_info_queryset.exists():
                    equipment_info_count = equipment_info_queryset.count()
                    equipment_info_res = equipment_info_queryset[(page - 1) * line:page * line]
                    send_json = CommonService.query_set_to_dict(equipment_info_res)
                    send_json['count'] = equipment_info_count
                    return ResponseJSON(0,send_json)
            return ResponseJSON(0,{'datas': [], 'count': 0})
        else:
            return ResponseJSON(444)

    def query_all_info(self, request_dict, userID):
        page = int(request_dict.get('page', None))
        line = int(request_dict.get('line', None))
        userID_id = request_dict.get('userID_id', None)
        param_flag = CommonService.get_param_flag(data=[page, line])
        if param_flag is True:
            check_perm = ModelService.check_permission(userID=userID, permID=30)
            if check_perm is True:
                uid_list = Device_Info.objects.filter(userID_id=userID_id).values_list('UID', flat=True)
                if len(uid_list):
                    equipment_info_queryset = Equipment_Info.objects.filter(devUid__in=uid_list).order_by(
                        '-id')
                    if equipment_info_queryset.exists():
                        equipment_info_count = equipment_info_queryset.count()
                        equipment_info_res = equipment_info_queryset[(page - 1) * line:page * line]
                        send_json = CommonService.query_set_to_dict(equipment_info_res)
                        send_json['count'] = equipment_info_count
                        return ResponseJSON(0,send_json)
                return ResponseJSON(0,{'datas': [], 'count': 0})
            else:
                return ResponseJSON(404)
        else:
            return ResponseJSON(444)

    def delete_by_admin(self, request_dict, userID):
        id_list = request_dict.getlist('id', None)
        param_flag = CommonService.get_param_flag(data=[id_list])
        if param_flag is True:
            check_perm = ModelService.check_permission(userID=userID, permID=10)
            if check_perm is True:
                try:
                    is_delete = Equipment_Info.objects.filter(id__in=id_list).delete()
                except Exception as e:
                    errorInfo = traceback.format_exc()
                    print(errorInfo)
                    return ResponseJSON(424,{'details': repr(e)})
                else:
                    return ResponseJSON(0,{'delete_count': is_delete[0]})
            else:
                return ResponseJSON(404)
        else:
            return ResponseJSON(444)