# -*- coding: utf-8 -*- from random import Random # 用于生成随机码 from wsgiref.util import FileWrapper from django.utils.timezone import utc from django.views.decorators.csrf import csrf_exempt from django.views.generic import TemplateView, View from django.utils.decorators import method_decorator from django.contrib.auth.hashers import make_password from django.shortcuts import HttpResponseRedirect from itsdangerous import URLSafeTimedSerializer as utsr # itsdangerous序列化 import re, base64, json, traceback, random, string from qcloudsms_py.httpclient import HTTPError from qcloudsms_py import SmsSingleSender from qcloudsms_py import SmsVoiceVerifyCodeSender, SmsVoicePromptSender from Ansjer.settings import * from Model.models import Email_Captcha, Device_User, Auth_Captcha from Service.ResponseService import * #生成随机字符串 def RandomStr(randomlength = 8, number = False): str = '' if number == False: characterSet = 'AaBbCcDdEeFfGgHhIiJjKkLlMmNnOoPpQqRrSsT' \ 'tUuVvWwXxYyZz0123456789' else: characterSet = '0123456789' length = len(characterSet) - 1 random = Random() for index in range(randomlength): str += characterSet[random.randint(0, length)] return str def Active_User(request, token): tokenConfirm = TokenLink(SECRET_KEY) try: username = tokenConfirm.confirm_validate_token(token) except: return ResponseJSON(1000) try: UserValid = Device_User.objects.get(userName=username) except Device_User.DoesNotExist: return ResponseJSON(113) UserValid = Device_User.objects.get(username=username) UserValid.is_active = True UserValid.user_isValid = True UserValid.save() return ResponseJSON(0) class MobiletoResetPwdView(TemplateView): @method_decorator(csrf_exempt) def dispatch(self, *args, **kwargs): return super(MobiletoResetPwdView, self).dispatch(*args, **kwargs) def get(self, request, *args, **kwargs): request.encoding = 'gb2312' userName = request.POST.get('userName', None) authCode = request.POST.get('identifyingCode', None) newPwd = request.POST.get('newPwd', None) return self.ValidationError(userName, newPwd, authCode) def post(self, request, *args, **kwargs): request.encoding = 'utf-8' userName = request.GET.get('userName', None) authCode = request.GET.get('identifyingCode', None) newPwd = request.GET.get('newPwd', None) return self.ValidationError(userName, newPwd, authCode) def ValidationError(self, userName, newPwd, authCode): if userName != None and authCode != None and newPwd != None: return self.MobiletoResetPwd(userName, authCode, newPwd) else: return ResponseJSON(800) def MobiletoResetPwd(self, userName, authCode, newPwd): UserValid = Device_User.objects.filter(username = userName).order_by('-data_joined') if UserValid: Users = Auth_Captcha.objects.filter(username = userName).order_by('-sendtime') if Users: for User in Users: if User.authcaptca == authCode and User.sendtype == 'forget': now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc) timeValid = (now_time - User.sendtime).total_seconds() if timeValid < 300: is_flag = UserValid.update(password=make_password(newPwd)) if is_flag: return ResponseJSON(0) else: return ResponseJSON(405) else: return ResponseJSON(120) else: return ResponseJSON(102) def date_handler(obj): return obj.isoformat() class ActiveUserView(View): """账户激活的View""" @method_decorator(csrf_exempt) def dispatch(self, *args, **kwargs): return super(ActiveUserView, self).dispatch(*args, **kwargs) def get(self, request, active_code): # 用code在数据库中过滤处信息 all_records = Email_Captcha.objects.filter(userCaptcha = active_code) if all_records: for record in all_records: # 通过邮箱查找到对应的用户 userEmail = record.userEmail user = Device_User.objects.get(userEmail = userEmail) Device_User.objects.filter(userEmail = userEmail).update(user_isValid=True, is_active = True) response = HttpResponseRedirect('/login/') return response class TokenLink: ''' 1. security_key就是settings.py中设置的SECRET_KEY 2. generate_validate_token函数通过URLSafeTimedSerializer 在用户注册时生成一个令牌。用户名在令牌中被编了码。生成令牌之后， 会将带有token的验证链接发送到注册邮箱。 3.在confirm_validate_token函数中，只要令牌没过期， 那它就会返回一个用户名，过期时间为3600秒 ''' def __init__(self, security_key): self.security_key = security_key self.salt = base64.encodestring(security_key.encode()) def generate_validate_token(self, token): serializer = utsr(self.security_key) return serializer.dumps(token, self.salt) def confirm_validate_token(self, token, expiration=3600): serializer = utsr(self.security_key) return serializer.loads(token, salt=self.salt, max_age=expiration) def remove_validate_token(self, token): serializer = utsr(self.security_key) return serializer.loads(token, salt=self.salt) class getRandom: def __init__(self): self.__seed = '1234567890abcdefghijklmnopqrstuvwxyzABCDEFGH' \ 'IJKLMNOPQRSTUVWXYZ!@#$%^&*()_+=-' def getSalt (self, strlen = 8): sa = [] for index in range (strlen): sa.append(random.choice(self.__seed)) salt = ''.join(sa) return salt def getSaltStr(self, strlen = 8): salt = ''.join(random.sample(string.ascii_letters + string.digits, strlen)) return salt class ModifyPwdView(TemplateView): @method_decorator(csrf_exempt) def dispatch(self, *args, **kwargs): return super(ModifyPwdView, self).dispatch(*args, **kwargs) def post(self, request, *args, **kwargs): request.encoding = 'utf-8' userEmail = request.POST.get('userEmail', None) oldPwd = request.POST.get('oldPwd', None) newPwd = request.POST.get('newPwd', None) return self.ValidationError(userEmail, oldPwd, newPwd) def get(self, request, *args, **kwargs): request.encoding = 'gb2312' userEmail = request.GET.get('userEmail', None) oldPwd = request.GET.get('oldPwd', None) newPwd = request.GET.get('newPwd', None) return self.ValidationError(userEmail, oldPwd, newPwd) def ValidationError(self, userEmail, oldPwd, newPwd): dataValid = DataValid() if dataValid.password_validate(newPwd): if oldPwd != newPwd: return ResponseJSON(118) try: User = Device_User.objects.get(userEmail = userEmail) User.userPwd = make_password(newPwd) User.save() except Exception as e: errorInfo = traceback.format_exc() print('更新密码到数据库: %s' % errorInfo) return ResponseJSON(501,{'msg':repr(e)}) else: return ResponseJSON(0) else: return ResponseJSON(109) class DataValid: def __init__(self): # 用户名正则 # self.re_name = re.compile(r'^[A-Za-z0-9\u4e00-\u9fa5\.\_]{1,16}$') self.re_name = re.compile(r'^[A-Za-z0-9\u4e00-\u9fa5\.\_\-\@]{4,40}$') # 密码强度正则 self.re_password = re.compile(r'^\w{1,16}$') # 手机号码正则 self.re_mobile = re.compile(r'^\d{1,16}$') # 邮箱地址正则 self.re_email = re.compile(r'^[A-Za-z0-9\u4e00-\u9fa5\.\_\-]+@[A-Za-z0-9_-]+(\.[A-Za-z0-9_-]+)+$') def name_validate(self, value): if self.re_name.match(value): return True else: return False def password_validate(self, value): if self.re_password.match(value): return True else: return False def email_validate(self, value): if self.re_email.match(value): return True else: return False def mobile_validate(self, value): if self.re_mobile.match(value): return True else: return False def sms(phoneNumbers, sendModel, msg, sendType = 'register', template_id = 7839): appid = SDK_AppID appkey = App_Key print(phoneNumbers, msg, sendModel, template_id) ssender = SmsSingleSender(appid, appkey) try: if sendType == 'register': result = ssender.send(0, "86", phoneNumbers, u"{authCode}为您的注册验证码，请于2分钟内填写。" u"如非本人操作，请忽略本短信。".format(authCode=msg), "", "") elif sendType == 'forget': result = ssender.send(0, "86", phoneNumbers, u"{userPwd}为您的账号新密码,请登录账号后,修改为" u"用户自定义密码.".format(userPwd=msg), "", "") except HTTPError as e: errorInfo = traceback.format_exc() print('发送手机验证码出现网络错误: %s' % errorInfo) return ResponseFormal(1022,{'msg':repr(e)}) except Exception as e: errorInfo = traceback.format_exc() print('发送手机短信验证码错误: %s' % errorInfo) return ResponseFormal(1021) else: resultCode = result.get('result', None) if resultCode == 0: JSON = json.dumps(result, ensure_ascii=False) return JSON else: errorJSON = json.dumps(result, ensure_ascii=False) return errorJSON class QCloudSms(object): def __init__(self, nationcode, msg, template_id): self.appid = SDK_AppID self.appkey = App_Key self.msg = msg # 验证码msg self.nationcode = nationcode # 国家码 self.template_id = template_id # 模板ID def sms(self, mobiles, model, type = 'register'): ssender = SmsSingleSender(self.appid, self.appkey) try: if type == 'register': result = ssender.send(0, "86", mobiles, u"{authCode}为您的注册验证码，请于2分钟内填写。" u"如非本人操作，请忽略本短信。".format(authCode= self.msg), "", "") elif type == 'forget': result = ssender.send(0, "86", mobiles, u"{userPwd}为您的账号新密码,请登录账号后,修改为" u"用户自定义密码.".format(userPwd = self.msg), "", "") except HTTPError as e: errorInfo = traceback.format_exc() print('发送手机短信验证码出现网络错误: %s' % errorInfo) return ResponseFormal(1022,repr(e)) except Exception as e: errorInfo = traceback.format_exc() print('发送手机验证码错误: %s' % errorInfo) return ResponseFormal(1021,repr(e)) else: resultCode = result.get('result', None) if resultCode == 0: JSON = json.dumps(result, ensure_ascii=False) return JSON else: errorJSON = json.dumps(result, ensure_ascii=False) return errorJSON def voice(self, mobiles, model, type = 'register'): # 语音验证码请求 vvcsender = SmsVoiceVerifyCodeSender(appid = self.appid, appkey = self.appkey) try: result = vvcsender.send(nation_code=self.nationcode, phone_number = \ mobiles, playtimes = 2, msg = self.msg, ext = '') except HTTPError as e: errorInfo = traceback.format_exc() print('发送手机验证码出现网络错误: %s' % errorInfo) return ResponseFormal(1022,repr(e)) except Exception as e: errorInfo = traceback.format_exc() print('发送手机语音验证码错误: %s' % errorInfo) return ResponseFormal(1023,repr(e)) else: resultCode = result.get('result', None) if resultCode == 0: JSON = json.dumps(result, ensure_ascii=False) return JSON else: errorJSON = json.dumps(result, ensure_ascii=False) return errorJSON # 发送语音通知 vpsender = SmsVoicePromptSender(appid=self.appid, appkey=self.appkey) # note: msg内容，首先需要申请内容模板，通过后才可以发送 try: result = vpsender.send(nation_code=self.nationcode, phone_number = \ mobiles, playtimes = 2, msg = self.msg, ext= '') except HTTPError as e: errorInfo = traceback.format_exc() print('发送手机验证码出现网络错误: %s' % errorInfo) return ResponseFormal(1022,repr(e)) except Exception as e: errorInfo = traceback.format_exc() print('发送手机语音验证码错误: %s' % errorInfo) return ResponseFormal(1023,repr(e)) else: resultCode = result.get('result', None) if resultCode == 0: JSON = json.dumps(result, ensure_ascii=False) return JSON else: errorJSON = json.dumps(result, ensure_ascii=False) return errorJSON def upload_file(request): """ 文件接收 :param request: 请求 :return: """ if request.method == "POST": # 请求方法为POST时，进行处理 upFile =request.FILES.get("upfile", None) # 获取上传的文件，如果没有文件，则默认为None if not upFile: return HttpResponse("no files for upload!") else: upgradePath = os.path.join(MEDIA_URL, '/Upgrade/IPC/') destination = open(os.path.join(upgradePath, upFile.name), 'wb+') # 打开特定的文件进行二进制的写操作 for chunk in upFile.chunks(): # 分块写入文件 destination.write(chunk) destination.close() return HttpResponse("upload over!") @csrf_exempt def download_file(request, Upgradename, *callback_args, **callback_kwargs): """ 下载单个文件 :param request: :param Upgradename: :param callback_args: :param callback_kwargs: :return: """ print(Upgradename) print(callback_args, callback_kwargs) for value in callback_args: print("other args:", value) # 打印dict类型的不定长参数 args for key in callback_kwargs: print("dictargs:" + key + ":" + bytes(callback_kwargs[key])) #received_json_data = json.loads(request.body) print(request.body) file_name = os.path.join(BASE_DIR, "static/Upgrade/DVR/").replace('\\', '/') + Upgradename if os.path.isfile(file_name): try: print(file_name) JSON = ResponseFormal(0) wrapper = FileWrapper(open(file_name, 'rb')) response = HttpResponse(wrapper, content_type="application/octet-stream") response['Content-Length'] = os.path.getsize(file_name) response['Content-Disposition'] = 'attachment; filename=%s' % os.path.basename(file_name) response['Content-Error'] = JSON return response except Exception as e: errorJSON = ResponseFormal(10,'Wrong reason:' + repr(e)) response = HttpResponse(errorJSON, content_type='text/plain', charset='utf-8') response['Content-Error'] = errorJSON return response else: errorJSON = ResponseFormal(907) response = HttpResponse(errorJSON, content_type='text/plain', charset='utf-8') response['Content-Error'] = errorJSON return response