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)