import json import logging import time import requests from django.db import transaction from django.views import View from Ansjer.config import CRCKey, CONFIG_INFO, CONFIG_TEST, CONFIG_US, \ CONFIG_CN, USED_SERIAL_REDIS_LIST, UNUSED_SERIAL_REDIS_LIST, SERVER_DOMAIN_US, REGION_ID_LIST from Model.models import SerialNumberModel, CompanySerialModel, UIDCompanySerialModel, UIDModel, Device_Info, \ iotdeviceInfoModel, LogModel, UidSetModel, UID_Bucket, \ Unused_Uid_Meal, Order_Model, StsCrdModel, VodHlsModel, ExperienceContextModel, UidUserModel, ExperienceAiModel, \ AiService, DeviceDomainRegionModel, RegionModel from Object.RedisObject import RedisObject from Object.TokenObject import TokenObject from Object.uidManageResponseObject import uidManageResponseObject from Service.AlgorithmService import AlgorithmBaseOn35 from Service.CommonService import CommonService from Service.EquipmentInfoService import EquipmentInfoService class SerialNumberView(View): def get(self, request, *args, **kwargs): request.encoding = 'utf-8' operation = kwargs.get('operation', None) request_dict = request.GET return self.validate(request_dict, operation, request) def post(self, request, *args, **kwargs): request.encoding = 'utf-8' operation = kwargs.get('operation', None) request_dict = request.POST return self.validate(request_dict, operation, request) def validate(self, request_dict, operation, request): response = uidManageResponseObject() if operation == 'attachUID': # 绑定uid return self.do_attach_uid(request_dict, response, request) elif operation == 'detachUID': # 解绑uid return self.do_detach_uid(request, request_dict, response) elif operation == 'create': # 创建序列号 return self.do_create(request_dict, response) elif operation == 'getUID': # 根据序列号获取uid return self.do_get_uid(request_dict, response) elif operation == 'getRegionInfo': # 根据序列号状态确认uid地区(PC工具使用) return self.get_region_info(request_dict, response) elif operation == 'saveRegion': # 保存设备地区信息(app使用) return self.save_region(request, request_dict, response) elif operation == 'getDomain': # 获取域名(设备使用) return self.get_domain(request_dict, response) elif operation == 'resetRegionId': # 重置地区id return self.reset_region_id(request_dict, response) else: return response.json(414) @staticmethod def do_create(request_dict, response): quantity = int(request_dict.get('quantity', 0)) if not quantity: return response.json(444) try: try: number = SerialNumberModel.objects.last().id except: number = 0 serial_number_bulk = [] now_time = int(time.time()) algorithm = AlgorithmBaseOn35() for i in range(quantity): serial_number = algorithm.getLetter(number) number += 1 # sum每次递增1 # 前面补0至六位 serial_number = (6 - len(serial_number)) * '0' + serial_number serial_number_bulk.append(SerialNumberModel(serial_number=serial_number, add_time=now_time)) # 开启事务写入 with transaction.atomic(): SerialNumberModel.objects.bulk_create(serial_number_bulk) return response.json(0) except Exception as e: print(e) return response.json(500, repr(e)) def do_attach_uid(self, request_dict, response, request): serial_number = request_dict.get('serial_number', None) token = request_dict.get('token', None) time_stamp = request_dict.get('time_stamp', None) if not all([serial_number, token, time_stamp]): return response.json(444) # 时间戳token校验 if not CommonService.check_time_stamp_token(token, time_stamp): return response.json(13) now_time = int(time.time()) serial = serial_number[0:6] company_identity = serial_number[6:9] full_serial = serial_number[0:9] # 根据企业标识确认企业秘钥 company_secret = '' if company_identity == '11A': company_secret = 'MTEyMTNB' elif company_identity == '11L': company_secret = 'VmXEWnBR' elif company_identity == '11Z': company_secret = 'ZsKWcxdD' try: # 判断序列号是否已和企业关联 company_serial_qs = CompanySerialModel.objects.filter(company__secret=company_secret, serial_number=serial) if not company_serial_qs.exists(): return response.json(173) company_serial = company_serial_qs[0] if company_serial.status == 0: # 该序列号未绑定企业 return response.json(173) elif company_serial.status == 1: # 绑定uid # redis加锁，防止同一个序列号重复绑定 key = serial + 'do_attach_uid' redisObj = RedisObject() isLock = redisObj.CONN.setnx(key, 1) redisObj.CONN.expire(key, 60) if not isLock: return response.json(5) # 获取并判断region_id region_id = CommonService.confirm_region_id() if region_id not in REGION_ID_LIST: return response.json(444, {'invalid region_id': region_id}) p2p_type = request_dict.get('p2ptype', 1) if serial_number[9:10]: p2p_type = serial_number[9:10] p2p_type = int(p2p_type) with transaction.atomic(): count = 0 while count < 3: # 查询是否存在未绑定序列号的uid uid_qs = UIDModel.objects.filter(vpg__company_id=company_serial.company.id, vpg__region_id=region_id, status=0, p2p_type=p2p_type). \ order_by('id') if not uid_qs.exists(): return response.json(375) uid = uid_qs[0] # 判断uid是否已绑定过序列号 uid_company_serial_qs = UIDCompanySerialModel.objects.filter(uid_id=uid.id) if uid_company_serial_qs.exists(): return response.json(377) result = UIDModel.objects.filter(id=uid.id, status=0).update(status=2, update_time=now_time) if int(result) <= 0: # 更新失败 count += 1 continue # UID关联【企业关联序列号】表创建数据 UIDCompanySerialModel.objects.create(uid_id=uid.id, company_serial_id=company_serial.id, add_time=now_time, update_time=now_time) company_serial.status = 2 company_serial.save() dev = Device_Info.objects.filter(UID=uid.uid) if dev.exists(): dev.update(serial_number=full_serial) full_uid_code = uid.full_uid_code if uid.platform in CRCKey.keys(): full_uid_code += ':' + CRCKey[uid.platform] res = { 'full_uid_code': CommonService.encode_data(full_uid_code), 'uid': CommonService.encode_data(uid.uid), 'extra': uid.uid_extra, 'platform': uid.platform, 'initString': uid.init_string, 'initStringApp': uid.init_string_app, } # 记录操作日志 ip = CommonService.get_ip_address(request) content = json.loads(json.dumps(request_dict)) log = { 'ip': ip, 'user_id': 1, 'status': 200, 'time': now_time, 'content': json.dumps(content), 'url': 'serialNumber/attachUID', 'operation': '序列号{}绑定uid: {}'.format(serial, uid.uid), } LogModel.objects.create(**log) if CONFIG_INFO != CONFIG_TEST: # 不为测试服,则序列号写入redis列表 redisObj.rpush(USED_SERIAL_REDIS_LIST, serial) redisObj.del_data(key=key) return response.json(0, res) return response.json(5) elif company_serial.status == 2: # 返回uid uid_qs = UIDCompanySerialModel.objects.filter(company_serial_id=company_serial.id) if not uid_qs.exists(): return response.json(173) uid = uid_qs.values('uid__uid', 'uid__mac', 'uid__uid_extra', 'uid__full_uid_code', 'uid__platform', 'uid__init_string', 'uid__init_string_app')[0] full_uid_code = uid['uid__full_uid_code'] if uid['uid__platform'] in CRCKey.keys(): full_uid_code += ':' + CRCKey[uid['uid__platform']] res = { 'full_uid_code': CommonService.encode_data(full_uid_code), 'uid': CommonService.encode_data(uid['uid__uid']), 'extra': uid['uid__uid_extra'], 'platform': uid['uid__platform'], 'initString': uid['uid__init_string'], 'initStringApp': uid['uid__init_string_app'], } return response.json(0, res) elif company_serial.status == 3: return response.json(10042) except Exception as e: djangoLogger = logging.getLogger('django') djangoLogger.exception(repr(e)) return response.json(176, str(e)) def do_get_uid(self, request_dict, response): serial_number = request_dict.get('serial_number', None) token = request_dict.get('token', None) time_stamp = request_dict.get('time_stamp', None) if token and time_stamp and serial_number: # 时间戳token校验 if not CommonService.check_time_stamp_token(token, time_stamp): return response.json(13) mark = serial_number[6:9] serial = serial_number[0:6] uid_company_serial_qs = UIDCompanySerialModel.objects.filter(company_serial__company__mark=mark, company_serial__serial_number__serial_number=serial) if uid_company_serial_qs.exists(): uid = uid_company_serial_qs.values('uid__uid', 'uid__mac', 'uid__uid_extra')[0] res = { 'uid': CommonService.encode_data(uid['uid__uid']), 'mac': CommonService.encode_data(uid['uid__mac']), 'extra': uid['uid__uid_extra'] } return response.json(0, res) else: return response.json(173) else: return response.json(444) def do_detach_uid(self, request, request_dict, response): token = request_dict.get('token', None) time_stamp = request_dict.get('time_stamp', None) serial_number = request_dict.get('serial_number', None) if not all([token, time_stamp, serial_number]): return response.json(444) # 时间戳token校验 if not CommonService.check_time_stamp_token(token, time_stamp): return response.json(13) now_time = int(time.time()) serial = serial_number[0:6] company_serial_qs = CompanySerialModel.objects.filter(serial_number=serial).values('status') if not company_serial_qs.exists(): return response.json(379) status = company_serial_qs[0]['status'] if status == 1: return response.json(0, {'success': '序列号未绑定uid'}) elif status == 3: return response.json(10043) uid_serial_qs = UIDCompanySerialModel.objects.filter(company_serial__serial_number=serial) if not uid_serial_qs.exists(): return response.json(173) uid_serial = uid_serial_qs[0] redisObj = RedisObject() try: with transaction.atomic(): uid = uid_serial.uid.uid company_serial_qs.update(status=1) # 更新序列号状态 # 删除设备相关数据,参考后台的设备重置删除的数据 Device_Info.objects.filter(UID=uid).delete() UidSetModel.objects.filter(uid=uid).delete() UidUserModel.objects.filter(UID=uid).delete() iotdeviceInfoModel.objects.filter(serial_number=serial).delete() # 删除推送消息 for val in range(1, 8): EquipmentInfoService.get_equipment_info_model('', val).filter(device_uid=uid).delete() # 重置设备云存 UID_Bucket.objects.filter(uid=uid).delete() Unused_Uid_Meal.objects.filter(uid=uid).delete() Order_Model.objects.filter(UID=uid).delete() StsCrdModel.objects.filter(uid=uid).delete() VodHlsModel.objects.filter(uid=uid).delete() ExperienceContextModel.objects.filter(uid=uid).delete() # 重置AI ExperienceAiModel.objects.filter(uid=uid).delete() AiService.objects.filter(uid=uid).delete() if CONFIG_INFO != CONFIG_TEST: # 不为测试服,则序列号写入redis列表 redisObj.rpush(UNUSED_SERIAL_REDIS_LIST, serial) UIDModel.objects.filter(uid=uid).update(status=0, mac='') # 重置uid的使用状态为未使用 uid_serial.delete() # 重置region_id if CONFIG_INFO == CONFIG_US: DeviceDomainRegionModel.objects.filter(serial_number=serial).update(region_id=0) else: try: url = SERVER_DOMAIN_US + 'serialNumber/resetRegionId' data = {'serial_number': serial} r = requests.post(url, data, timeout=3) assert r.status_code == 200 res = r.json() assert res['result_code'] == 0 except (TimeoutError, AssertionError): return response.json(5) # 记录操作日志 ip = CommonService.get_ip_address(request) content = json.loads(json.dumps(request_dict)) log = { 'ip': ip, 'user_id': 1, 'status': 200, 'time': now_time, 'content': json.dumps(content), 'url': 'serialNumber/detachUID', 'operation': '序列号{}解绑uid: {}'.format(serial, uid), } LogModel.objects.create(**log) return response.json(0) except Exception as e: djangoLogger = logging.getLogger('django') djangoLogger.exception(repr(e)) return response.json(176, str(e)) @staticmethod def get_region_info(request_dict, response): """ 根据序列号状态确认uid地区 @param request_dict: 请求参数 @param response: 响应对象 @request_dict serial_number: 序列号 @return: response """ serial_number = request_dict.get('serial_number', None) if not serial_number: return response(444) company_serial_qs = CompanySerialModel.objects.filter(serial_number=serial_number[:6]).values('status') if not company_serial_qs.exists(): return response.json(173) status = company_serial_qs[0]['status'] if status == 2: if CONFIG_INFO == CONFIG_CN: return response.json(0, {'region': 1}) elif CONFIG_INFO == CONFIG_US: return response.json(0, {'region': 3}) elif status == 3: if CONFIG_INFO == CONFIG_CN: return response.json(0, {'region': 3}) elif CONFIG_INFO == CONFIG_US: return response.json(0, {'region': 1}) return response.json(0) @staticmethod def save_region(request, request_dict, response): """ 保存设备地区信息 @param request: 请求体 @param request_dict: 请求参数 @param response: 响应对象 @request_dict token: 令牌 @request_dict time_stamp: 时间戳 @return: response """ token = request_dict.get('token', None) serial_number = request_dict.get('serial_number', None) if not all([serial_number, token]): return response(444) try: serial_number = serial_number[:6] # 不是美洲服,请求美洲域名保存数据 if CONFIG_INFO != CONFIG_US: # token认证 token_obj = TokenObject(token) if token_obj.code != 0: return response.json(token_obj.code) response.lang = token_obj.lang url = SERVER_DOMAIN_US + 'serialNumber/saveRegion' data = request_dict.dict() data['ip'] = CommonService.get_ip_address(request) data['region_id'] = CommonService.confirm_region_id() try: r = requests.post(url=url, data=data, timeout=3) assert r.status_code == 200 res = r.json() assert res['result_code'] == 0 except (TimeoutError, AssertionError): return response.json(5) # 美洲服,处理请求和保存数据 else: data = {} # 处理其他服务器发起请求的情况 region_id = request_dict.get('region_id', None) if region_id is not None: region_id = int(region_id) data['region_id'] = region_id data['ip'] = request_dict.get('ip', '') else: # token认证 token_obj = TokenObject(token) if token_obj.code != 0: return response.json(token_obj.code) response.lang = token_obj.lang region_id = CommonService.confirm_region_id() data['region_id'] = region_id data['ip'] = CommonService.get_ip_address(request) # 写入数据 device_domain_region_qs = DeviceDomainRegionModel.objects.filter(serial_number=serial_number).\ values('region_id') if not device_domain_region_qs.exists(): data['serial_number'] = serial_number DeviceDomainRegionModel.objects.create(**data) else: # 设备解绑过uid,更新region_id if device_domain_region_qs[0]['region_id'] == 0: device_domain_region_qs.update(**data) return response.json(0) except Exception as e: return response.json(500, repr(e)) @staticmethod def get_domain(request_dict, response): """ 获取域名 @param request_dict: 请求参数 @param response: 响应对象 @request_dict serial_number: 序列号 @request_dict time_stamp_token: 时间戳token @request_dict time_stamp: 时间戳 @return: response """ serial_number = request_dict.get('serial_number', None) time_stamp_token = request_dict.get('time_stamp_token', None) time_stamp = request_dict.get('time_stamp', None) if not all([serial_number, time_stamp_token, time_stamp]): return response(444) # 时间戳token校验 if not CommonService.check_time_stamp_token(time_stamp_token, time_stamp): return response.json(13) try: serial_number = serial_number[:6] device_domain_region_qs = DeviceDomainRegionModel.objects.filter(serial_number=serial_number).values( 'region_id') if not device_domain_region_qs.exists(): return response.json(173) region_id = device_domain_region_qs[0]['region_id'] region_qs = RegionModel.objects.filter(id=region_id).values('api') res = { 'api': region_qs[0]['api'] } return response.json(0, res) except Exception as e: return response.json(500, repr(e)) @staticmethod def reset_region_id(request_dict, response): """ 重置地区id @param request_dict: 请求参数 @param response: 响应对象 @request_dict serial_number: 序列号 @return: response """ serial_number = request_dict.get('serial_number', None) if not serial_number: return response(444) try: serial_number = serial_number[:6] DeviceDomainRegionModel.objects.filter(serial_number=serial_number).update(region_id=0) return response.json(0) except Exception as e: return response.json(500, repr(e))