# -*- encoding: utf-8 -*- """ @File : UserDeviceShareController.py @Time : 2023/2/3 13:25 @Author : guanhailong @Email : guanhailong@asj6.wecom.work @Software: PyCharm """ import datetime import time from django.db import transaction from Controller.CheckUserData import RandomStr from Model.models import Device_User, Device_Info, DeviceSuperPassword, SysMsgModel from Object.RedisObject import RedisObject from Object.ResponseObject import ResponseObject from Object.TokenObject import TokenObject from django.views import View from Service.CommonService import CommonService class SuperPasswordView(View): def get(self, request, *args, **kwargs): request.encoding = 'utf-8' operation = kwargs.get('operation') request_dict = request.GET return self.validation(request_dict, request, operation) def post(self, request, *args, **kwargs): request.encoding = 'utf-8' operation = kwargs.get('operation') request_dict = request.GET return self.validation(request_dict, request, operation) def validation(self, request_dict, request, operation): token = TokenObject(request.META.get('HTTP_AUTHORIZATION')) lang = request_dict.get('lang', token.lang) response = ResponseObject(lang) userID = token.userID if operation == 'getAuthorizationCode': # 用户提交请求 return self.getAuthorizationCode(request_dict, response, userID) if operation == 'customerServiceManagement': # 审核用户请求/生成超级密码 return self.customerServiceManagement(request_dict, response) if operation == 'getDeviceSuperPassword': # 查询用户请求表 return self.getDeviceSuperPassword(request_dict, response) if operation == 'verifyTheVerificationCode': # 检验验证码 return self.verifyTheVerificationCode(request_dict, response, userID) if operation == 'deleteInformation': # 删除信息 return self.deleteInformation(request_dict, response) else: return response.json(404) def getAuthorizationCode(self, request_dict, response, userID): """ 用户提交请求 @param uid:设备id @param request_dict:请求参数 @param response:响应对象 @param describe:需求描述 @param Purchase_channel:购买渠道描述 @param orderID:订单id @param buyTime:购买时间 @return: """ uid = request_dict.get('uid', None) describe = request_dict.get('describe', None) if not all([uid, describe]): return response.json(444) purchase_channel = request_dict.get('purchase_channel', None) orderID = request_dict.get('orderID', None) buyTime = request_dict.get('buyTime', None) lang = request_dict.get('lang', 'en') try: now = int(time.time()) addTime = now device_user_qs = Device_User.objects.filter(userID=userID) if not device_user_qs.exists(): return response.json(173) if buyTime: buyTime = datetime.datetime.strptime(buyTime, '%Y-%m-%d') buyTime = CommonService.str_to_timestamp(str_time=str(buyTime)) DeviceSuperPassword.objects.create(uid=uid, orderID=orderID, describe=describe, purchase_channel=purchase_channel, addTime=addTime, userID_id=userID, buyTime=buyTime, status=0, lang=lang) return response.json(0) except Exception as e: print('生成验证码异常,errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e))) return response.json(500, repr(e)) def customerServiceManagement(self, request_dict, response): """ 审核用户请求/生成超级密码 @param request_dict:请求参数 @param response:响应对象 @param hint:提示内容 @return: """ userID = request_dict.get('userID', None) uid = request_dict.get('uid', None) status = request_dict.get('status', None) hint = request_dict.get('hint', None) lang = request_dict.get('lang', 'en') if not all({uid, userID}): return response.json(444) now = int(time.time()) try: with transaction.atomic(): device_super_password_qs = DeviceSuperPassword.objects.filter(uid=uid, userID=userID) if not device_super_password_qs.exists(): return response.json(173) status = int(status) authcode = '' if status == 1: # 验证码生成 super_code = RandomStr(6, True) super_password_id = "super_password_%s" % userID redisObj = RedisObject() redis = redisObj.set_data(key=super_password_id, val=super_code, expire=86400) if redis is False: return response.json(121) device_super_password_qs.update(status=status) if lang == 'en': msg = "Your authorization code is " + super_code + ",valid within 24 hours" else: msg = "您的授权代码：" + super_code + "，24小时内有效" SysMsgModel.objects.create(userID_id=userID, msg=msg, addTime=now, updTime=now, uid=uid, eventType=2) return response.json(0) if status == 0 and len(hint) > 1: device_super_password_qs.update(status=status, hint=hint) msg = hint SysMsgModel.objects.create(userID_id=userID, msg=msg, addTime=now, updTime=now, uid=uid, eventType=2) return response.json(0) return response.json(177) except Exception as e: print('修改状态异常,errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e))) return response.json(500, repr(e)) def getDeviceSuperPassword(self, request_dict, response): """ 查询用户请求表 @param request_dict:请求参数 @param response:响应对象 @return: """ pageNo = request_dict.get('pageNo', None) pageSize = request_dict.get('pageSize', None) status = request_dict.get('status', None) userID = request_dict.get('userID', None) uid = request_dict.get('uid', None) if not all([pageNo, pageSize]): return response.json(444) page = int(pageNo) line = int(pageSize) try: device_super_password_qs = DeviceSuperPassword.objects.all() if status: device_super_password_qs = device_super_password_qs.filter(status=status) if userID: device_super_password_qs = device_super_password_qs.filter(userID=userID) if uid: device_super_password_qs = device_super_password_qs.filter(uid=uid) if not device_super_password_qs.exists(): return response.json(0, []) count = device_super_password_qs.count() device_super_password_qs = device_super_password_qs.values('id', 'uid', 'userID', 'orderID', 'describe', 'purchase_channel', 'addTime', 'status', 'buyTime', 'hint', 'lang') device_super_password_qs = device_super_password_qs.order_by('-addTime')[ (page - 1) * line:page * line] return response.json(0, {'list': list(device_super_password_qs), 'count': count}) except Exception as e: print('查询异常,errLine:{}, errMsg:{}'.format(e.__traceback__.tb_lineno, repr(e))) return response.json(500, repr(e)) def verifyTheVerificationCode(self, request_dict, response, userID): """ 检验验证码 @param request_dict:请求参数 @param response:响应对象 @param userID:用户ID @param authcode:验证码 @return: """ authcode = request_dict.get('authcode', None) if authcode: authcode = CommonService.decode_data(authcode) super_password_id = 'super_password_' + userID redisObj = RedisObject() # redis里面的验证码 redis_image_code = redisObj.get_data(key=super_password_id) # 验证用户输入的验证码和redis中的验证码 if redis_image_code is False or authcode.lower() != redis_image_code.lower(): return response.json(121) else: return response.json(0) else: return response.json(444) def deleteInformation(self, request_dict, response): """ 删除信息 @param request_dict:请求参数 @param response:响应对象 @param id:主键 """ id = request_dict.get('id', None) if not id: return response.json(444) device_super_password_qs = DeviceSuperPassword.objects.filter(id=id) if not device_super_password_qs.exists(): return response.json(173) device_super_password_qs.delete() return response.json(0)