| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356 | #!/usr/bin/env python3# -*- coding: utf-8 -*-"""@Copyright (C) ansjer cop Video Technology Co.,Ltd.All rights reserved.@AUTHOR: ASJRD018@NAME: AnsjerFormal@software: PyCharm@DATE: 2018/9/11 15:08@Version: python3.6@MODIFY DECORD:ansjer dev@file: UserController.py@Contact: chanjunkai@163.com"""import datetimeimport tracebackimport simplejson as jsonfrom django.contrib.auth.hashers import make_password, check_password  # 对密码加密模块from django.db.models import Qfrom django.http import HttpResponseRedirectfrom django.utils.decorators import method_decoratorfrom django.utils.timezone import utcfrom django.views.decorators.csrf import csrf_exemptfrom django.views.generic import TemplateViewfrom ratelimit.decorators import ratelimitfrom Ansjer.config import AuthCode_Expire, SERVER_DOMAIN, APNS_CONFIG, JPUSH_CONFIG, FCM_CONFIGfrom Controller.CheckUserData import DataValid, date_handler, RandomStrfrom Model.models import Device_User, Role, SysMsgModel, UidPushModel, UidSetModelfrom Object.AWS.SesClassObject import SesClassObjectfrom Object.AliSmsObject import AliSmsObjectfrom Object.RedisObject import RedisObjectfrom Object.ResponseObject import ResponseObjectfrom Object.TokenObject import TokenObjectfrom Service.CommonService import CommonServicefrom Service.ModelService import ModelServicefrom Service.TemplateService import TemplateServicefrom django.views.generic import View# 获取验证码class authCodeView(TemplateView):    @method_decorator(csrf_exempt)    def dispatch(self, *args, **kwargs):        return super(authCodeView, self).dispatch(*args, **kwargs)    @ratelimit(key='ip', rate='2/m')    def post(self, request, *args, **kwargs):        request.encoding = 'utf-8'        lang = request.POST.get('language', None)        response = ResponseObject(lang)        was_limited = getattr(request, 'limited', False)        if was_limited is True:            return response.json(5)        username = request.POST.get('userName', None)        useremail = request.POST.get('userEmail', None)        return self.ValidationError(username, useremail, response)    # @ratelimit(key='ip', rate='2/m')    def get(self, request, *args, **kwargs):        # Device_User.objects.filter(userEmail='chanjunkai@163.com').delete()        request.encoding = 'utf-8'        lang = request.GET.get('language', None)        response = ResponseObject(lang)        was_limited = getattr(request, 'limited', False)        if was_limited is True:            return response.json(5)        username = request.GET.get('userName', None)        email = request.GET.get('userEmail', None)        return self.ValidationError(username, email, response)    def ValidationError(self, username, email, response):        if username:            username = username.strip()            return self.phoneCode(username, response)        elif email:            email = email.strip()            return self.emailCode(email, response)        else:            return response.json(800)    def phoneCode(self, phone, response):        dataValid = DataValid()        if dataValid.mobile_validate(phone):            reds = RedisObject()            identifyingCode = reds.get_data(key=phone + '_identifyingCode')            if identifyingCode is False:                user_qs = Device_User.objects.filter(username=phone)                if user_qs.exists():                    return response.json(101)                else:                    identifyingCode = RandomStr(6, True)                    if reds.set_data(key=phone + '_identifyingCode', val=identifyingCode, expire=600):                        return response.json(0, {'identifyingCode': identifyingCode})                    else:                        return response.json(10, '生成缓存系统错误')            else:                return response.json(0, {'identifyingCode': identifyingCode})        else:            return response.json(107)    def emailCode(self, email, response):        dataValid = DataValid()        if dataValid.email_validate(email):            reds = RedisObject()            identifyingCode = reds.get_data(key=email + '_identifyingCode')            if identifyingCode is False:                user_qs = Device_User.objects.filter(username=email)                email_qs = Device_User.objects.filter(userEmail=email)                if user_qs.exists():                    return response.json(103)                elif email_qs.exists():                    return response.json(103)                else:                    identifyingCode = RandomStr(6, True)                    if reds.set_data(key=email + '_identifyingCode', val=identifyingCode, expire=AuthCode_Expire):                        send_data = TemplateService.email_message(type='register_code', language=response.lang)                        ses = SesClassObject()                        send_res = ses.send_email(                            send_address_list=[email],                            subject=send_data['title'],                            body=send_data['body'].replace("{username}", email).replace("{captcha}",                                                                                        str(identifyingCode))                        )                        if send_res:                            return response.json(0, {'identifyingCode': identifyingCode})                        else:                            reds.del_data(key=email + '_identifyingCode')                            return response.json(44)                    else:                        return response.json(10, '生成缓存系统错误')            else:                return response.json(89, {'identifyingCode': identifyingCode})        else:            return response.json(107)# 验证码注册class registerView(TemplateView):    @method_decorator(csrf_exempt)    def dispatch(self, *args, **kwargs):        return super(registerView, self).dispatch(*args, **kwargs)    def post(self, request, *args, **kwargs):        request.encoding = 'utf-8'        request_dict = request.POST        return self.validates(request_dict)    def get(self, request, *args, **kwargs):        request.encoding = 'utf-8'        request_dict = request.GET        return self.validates(request_dict)    def validates(self, request_dict):        username = request_dict.get('userName', None)        userEmail = request_dict.get('userEmail', None)        password = request_dict.get('userPwd', None)        authCode = request_dict.get('identifyingCode', None)        language = request_dict.get('language', None)        response = ResponseObject(language)        if username and password and authCode:            # 过滤空格            username = username.strip()            if userEmail:                userEmail = userEmail.strip()            return self.register(username, userEmail, password, authCode, response)        else:            return response.json(800)    def register(self, username, userEmail, password, authCode, response):        dataValid = DataValid()        reds = RedisObject()        identifyingCode = reds.get_data(key=username + '_identifyingCode')        if identifyingCode is False:            if userEmail:                identifyingCode = reds.get_data(key=userEmail + '_identifyingCode')                if identifyingCode is False:                    return response.json(120)                else:                    username = userEmail            else:                return response.json(120)        if authCode != identifyingCode:            return response.json(121)        if dataValid.password_validate(password):            if dataValid.email_validate(username):                if userEmail:                    print(userEmail)                    emailValid = Device_User.objects.filter(userEmail=userEmail)                    if emailValid.exists():                        return response.json(103)                else:                    userEmail = username                if username:                    nameValid = Device_User.objects.filter(username=username)                    if nameValid.exists():                        return response.json(101)                try:                    create_data = {                        "username": username,                        "NickName": username,                        "userEmail": userEmail,                        "password": make_password(password),                        "userID": CommonService.getUserID(μs=False, setOTAID=True),                        "is_active": True,                        "user_isValid": True,                    }                    users = Device_User.objects.create(**create_data)                except Exception as e:                    errorInfo = traceback.format_exc()                    print(errorInfo)                    return response.json(424, repr(e))                else:                    if reds.del_data(key=username + '_identifyingCode'):                        return response.json(0, {                            "user": {                                "userID": users.userID,                                "username": users.username,                                "userEmail": users.userEmail,                                "NickName": users.NickName,                                "userIconUrl": str(users.userIconUrl),                                "is_superuser": users.is_superuser,                                "is_active": users.is_active,                                "data_joined": date_handler(users.data_joined),                                "last_login": date_handler(users.last_login),                            }                        })                    else:                        return response.json(10, '删除缓存验证码错误')            elif dataValid.mobile_validate(username):                nameValid = Device_User.objects.filter(username=username)                if nameValid:                    return response.json(101)                try:                    create_data = {                        "username": username,                        "NickName": username,                        "userEmail": userEmail,                        "password": make_password(password),                        "userID": CommonService.getUserID(μs=False, setOTAID=True),                        "is_active": True,                        "user_isValid": True,                    }                    users = Device_User.objects.create(**create_data)                except Exception as e:                    errorInfo = traceback.format_exc()                    print(errorInfo)                    return response.json(424, repr(e))                else:                    if reds.del_data(key=username + '_identifyingCode'):                        return response.json(0, {                            "user": {                                "userID": users.userID,                                "username": users.username,                                "userEmail": users.userEmail,                                "NickName": users.NickName,                                "userIconUrl": str(users.userIconUrl),                                "is_superuser": users.is_superuser,                                "is_active": users.is_active,                                "data_joined": date_handler(users.data_joined),                                "last_login": date_handler(users.last_login),                            }                        })                    else:                        return response.json(10, '删除缓存验证码错误')            else:                return response.json(107)        else:            return response.json(109)# 登出class LogoutView(TemplateView):    @method_decorator(csrf_exempt)    def dispatch(self, *args, **kwargs):        return super(LogoutView, self).dispatch(*args, **kwargs)    def post(self, request, *args, **kwargs):        request.encoding = 'utf-8'        request_dict = request.POST        return self.Logout(request_dict)    def get(self, request, *args, **kwargs):        request.encoding = 'utf-8'        request_dict = request.GET        return self.Logout(request_dict)    def Logout(self, request_dict):        response = ResponseObject()        token = request_dict.get('token')        tko = TokenObject(token)        if tko.code == 0:            Device_User.objects.filter(userID=tko.userID).update(online=False)            redisObj = RedisObject(db=3)            redisObj.del_data(key=tko.userID)            m_code = request_dict.get('m_code', None)            if m_code:                userID = tko.userID                try:                    UidPushModel.objects.filter(userID_id=userID, m_code=m_code).delete()                except Exception as e:                    pass                else:                    pass            return response.json(0)        else:            return response.json(tko.code)# 修改密码class ChangePwdView(TemplateView):    @method_decorator(csrf_exempt)    def dispatch(self, *args, **kwargs):        return super(ChangePwdView, self).dispatch(*args, **kwargs)    def post(self, request, *args, **kwargs):        request.encoding = 'utf-8'        request_dict = request.POST        return self.validates(request_dict)    def get(self, request, *args, **kwargs):        request.encoding = 'utf-8'        request_dict = request.GET        return self.validates(request_dict)    def validates(self, request_dict):        token = request_dict.get('token', None)        oldPwd = request_dict.get('oldPwd', None)        newPwd = request_dict.get('newPwd', None)        response = ResponseObject()        if oldPwd is None and newPwd is None:            return response.json(800)        tko = TokenObject(token)        response.lang = tko.lang        if tko.code != 0:            return response.json(tko.code)        return self.updatePwd(tko.userID, oldPwd, newPwd, response)    def updatePwd(self, userID, oldPwd, newPwd, response):        user_qs = Device_User.objects.filter(userID=userID)        if not user_qs.exists():            return response.json(104)        c_p = check_password(oldPwd, user_qs[0].password)        # 密码是否正确        if not c_p:            return response.json(111)        update = user_qs.update(password=make_password(newPwd))        if update:            return response.json(0)        else:            return response.json(177)class ForgetPwdView(TemplateView):    '''    忘记密码    '''    @method_decorator(csrf_exempt)    def dispatch(self, *args, **kwargs):        return super(ForgetPwdView, self).dispatch(*args, **kwargs)    @ratelimit(key='ip', rate='1/m')    def get(self, request, *args, **kwargs):        request.encoding = 'utf-8'        response = ResponseObject()        was_limited = getattr(request, 'limited', False)        if was_limited is True:            return response.json(5)        userName = request.GET.get('userName', None)        return self.ValidationError(userName, response)    @ratelimit(key='ip', rate='1/m')    def post(self, request):        request.encoding = 'utf-8'        userName = request.POST.get('userName', None)        response = ResponseObject()        was_limited = getattr(request, 'limited', False)        if was_limited is True:            return response.json(5)        return self.ValidationError(userName, response)    def ValidationError(self, userName, response):        # return response.json(475)        if userName != None:            userName = userName.strip()            return self.ForgetPwd(userName, response)        else:            return response.json(800)    def ForgetPwd(self, userName, response):        dataValid = DataValid()        if dataValid.mobile_validate(userName):            User = Device_User.objects.filter(username=userName)        elif dataValid.email_validate(userName):            User = Device_User.objects.filter(username=userName)            if not User.exists():                User = Device_User.objects.filter(userEmail=userName)        else:            return response.json(104)        if User:            email = User[0].userEmail            userID = User[0].userID            if email:                redisObj = RedisObject()                reset_pwd = redisObj.get_data(key=userID + '_email_reset_pwd')                if reset_pwd is False:                    tko = TokenObject()                    rest = tko.generate(data={'userID': userID, 'user': userName})                    token = rest['access_token']                    reset_pwd = CommonService.RandomStr(6)                    send_data = TemplateService.email_message(type='forget', language='en')                    reset_link = '{server_host}account/email-re-pwd?token={token}'.format(                        server_host=SERVER_DOMAIN, token=token)                    send_body = send_data['body'].format(username=email, reset_pwd=reset_pwd, reset_link=reset_link)                    ses = SesClassObject()                    send_res = ses.send_email(send_address_list=[email], subject=send_data['title'], body=send_body)                    if send_res is True:                        if redisObj.set_data(key=userID + '_email_reset_pwd', val=reset_pwd, expire=3600):                            return response.json(0)                        else:                            return response.json(10, '存储验证失败')                else:                    return response.json(89)            else:                return response.json(103)        else:            return response.json(104)class EmailResetPwdView(TemplateView):    @method_decorator(csrf_exempt)    def dispatch(self, *args, **kwargs):        return super(EmailResetPwdView, self).dispatch(*args, **kwargs)    # 查询    def get(self, request, *args, **kwargs):        response = ResponseObject()        request_dict = request.GET        return self.validate(request_dict, response, *args, **kwargs)    # 认证登录    def post(self, request, *args, **kwargs):        response = ResponseObject()        try:            print(request.body.decode("utf-8"))            json_data = json.loads(request.body.decode("utf-8"))        except Exception as e:            return response.json(10, repr(e))        else:            request_dict = json_data            return self.validate(request_dict, response, *args, **kwargs)    def validate(self, request_dict, response, *args, **kwargs):        token = request_dict.get('token', None)        tko = TokenObject(token)        if tko.code == 0:            redisObj = RedisObject()            userID = tko.userID            reset_pwd = redisObj.get_data(key=userID + '_email_reset_pwd')            if reset_pwd is not False:                user_qs = Device_User.objects.filter(userID=userID)                if user_qs.exists():                    redisObj.del_data(key=userID + '_email_reset_pwd')                    is_update = user_qs.update(password=make_password(reset_pwd))                    if is_update:                        return HttpResponseRedirect(                            "http://www.dvema.com/web/html/paw_update_success.html?code=" + reset_pwd)                    else:                        return response.json(10)                else:                    return response.json(104)            else:                return HttpResponseRedirect('http://www.dvema.com/web/html/paw_update_unsuccessful.html?lang=en')        else:            return HttpResponseRedirect('http://www.dvema.com/web/html/paw_update_unsuccessful.html?lang=en')class refreshTokenView(TemplateView):    @method_decorator(csrf_exempt)    def dispatch(self, *args, **kwargs):        return super(refreshTokenView, self).dispatch(*args, **kwargs)    def post(self, request, *args, **kwargs):        request.encoding = 'utf-8'        content_type = request.META.get('CONTENT_TYPE', None)        if content_type == 'application/json':            request_dict = json.loads(request.body.decode('utf-8'))        else:            request_dict = request.POST        return self.validation(request_dict)    def get(self, request, *args, **kwargs):        request.encoding = 'utf-8'        request_dict = request.GET        return self.validation(request_dict)    def validation(self, request_dict):        token = request_dict.get('token', None)        lang = request_dict.get('lang', None)        language = request_dict.get('language', None)        if lang is None:            lang = language        response = ResponseObject(lang)        if token is not None:            tko = TokenObject(token)            tko.lang = lang            res = tko.refresh()            code = tko.code            if code == 0:                return response.json(0, res)            else:                return response.json(code)        else:            return response.json(444, 'token')# 获取验证码class v2authCodeView(TemplateView):    @method_decorator(csrf_exempt)    def dispatch(self, *args, **kwargs):        return super(v2authCodeView, self).dispatch(*args, **kwargs)    @ratelimit(key='ip', rate='2/m')    def post(self, request, *args, **kwargs):        request.encoding = 'utf-8'        lang = request.POST.get('lang', None)        if not lang:            lang = request.POST.get('language', None)        response = ResponseObject(lang)        request_dict = request.POST        phone = request_dict.get('phone', None)        if phone is not None:            was_limited = getattr(request, 'limited', False)            if was_limited is True:                return response.json(5)        return self.ValidationError(request_dict, response)    @ratelimit(key='ip', rate='2/m')    def get(self, request, *args, **kwargs):        # Device_User.objects.filter(userEmail='chanjunkai@163.com').delete()        request.encoding = 'utf-8'        lang = request.GET.get('lang', None)        if not lang:            lang = request.GET.get('language', None)        response = ResponseObject(lang)        was_limited = getattr(request, 'limited', False)        if was_limited is True:            return response.json(5)        request_dict = request.GET        return self.ValidationError(request_dict, response)    def ValidationError(self, request_dict, response):        email = request_dict.get('email', None)        phone = request_dict.get('phone', None)        country_code = request_dict.get('country_code', None)        sign_name = request_dict.get('sign_name', None)        if email is not None:            email = email.strip()            # 阿里云的发送邮箱的调用方法            return self.aliyun_emailCode(email, response)            # return self.emailCode(email, response)        elif phone is not None:            phone = phone.strip()            if country_code is None:                return self.phoneCode(phone, response, sign_name)            else:                country_code = str(country_code.strip())                return self.phoneCodeV2(country_code, phone, response, sign_name)        else:            return response.json(444)    def emailCode(self, email, response):        dataValid = DataValid()        # 邮箱匹配        if dataValid.email_validate(email) is False:            return response.json(107)        reds = RedisObject()        identifyingCode = reds.get_data(key=email + '_identifyingCode')        # 是否以获取邮箱验证码        if identifyingCode:            return response.json(89)        user_qs = Device_User.objects.filter(username=email)        email_qs = Device_User.objects.filter(userEmail=email)        # 邮箱用户是否已存在        if user_qs.exists():            return response.json(103)        elif email_qs.exists():            return response.json(103)        # 生成随机6位数        identifyingCode = RandomStr(6, True)        # 设置随机数缓存生命周期        send_data = TemplateService.email_message(type='register_code', language=response.lang)        ses = SesClassObject()        # 发送邮件        send_res = ses.send_email(            send_address_list=[email],            subject=send_data['title'],            body=send_data['body'].replace("{username}", email).replace("{captcha}",                                                                        str(identifyingCode))        )        if send_res is not True:            return response.json(44)        if reds.set_data(key=email + '_identifyingCode', val=identifyingCode, expire=600) is not True:            return response.json(10, 'error')        return response.json(0)        # return response.json(0, {'identifyingCode': identifyingCode})    # 阿里云获取邮箱验证码    def aliyun_emailCode(self, email, response):        print('阿里云开始')        dataValid = DataValid()        # 邮箱匹配        if dataValid.email_validate(email) is False:            return response.json(107)        reds = RedisObject()        identifyingCode = reds.get_data(key=email + '_identifyingCode')        # 是否以获取邮箱验证码        if identifyingCode:            return response.json(89)        user_qs = Device_User.objects.filter(username=email)        email_qs = Device_User.objects.filter(userEmail=email)        # 邮箱用户是否已存在        if user_qs.exists():            return response.json(103)        elif email_qs.exists():            return response.json(103)        # 生成随机6位数        identifyingCode = RandomStr(6, True)        # 设置随机数缓存生命周期        send_data = TemplateService.email_message(type='register_code', language=response.lang)        ses = SesClassObject()        # 发送邮件        send_res = ses.alyEmailCode(            send_address_list=[email],            subject=send_data['title'],            body=send_data['body'].replace("{username}", email).replace("{captcha}", str(identifyingCode))        )        if send_res is not True:            return response.json(44)        if reds.set_data(key=email + '_identifyingCode', val=identifyingCode, expire=600) is not True:            return response.json(10, 'error')        return response.json(0)    def phoneCode(self, phone, response, sign_name):        dataValid = DataValid()        if dataValid.mobile_validate(phone) is not True:            return response.json(107)        reds = RedisObject()        reds_key = str(phone) + '_identifyingCode'        identifyingCode = reds.get_data(key=reds_key)        reds_key_ttl = reds.get_ttl(key=reds_key)        if reds_key_ttl > 240 and identifyingCode:            # if identifyingCode :            return response.json(90)        user_qs = Device_User.objects.filter(username=phone)        phone_qs = Device_User.objects.filter(phone=phone)        if user_qs.exists() or phone_qs.exists():            return response.json(101)        identifyingCode = RandomStr(6, True)        # 发送手机验证码        aliSms = AliSmsObject()        if sign_name == 'zosi':            sign_ms = '周视'        else:            sign_ms = 'Ansjer'        res = aliSms.send_code_sms(phone=phone, code=identifyingCode, sign_name=sign_ms,                                   temp_msg='SMS_151600991')        print(res)        if res["Code"] == "OK":            if reds.set_data(key=reds_key, val=identifyingCode, expire=300) is not True:                # if reds.set_data(key=phone + '_identifyingCode', val=identifyingCode, expire=60) is not True:                return response.json(10, '生成缓存系统错误')            return response.json(0)        else:            return response.json(10, res["Message"])    def phoneCodeV2(self, country_code, phone, response, sign_name):        dataValid = DataValid()        if dataValid.mobile_validate(phone) is not True:            return response.json(107)        reds = RedisObject()        reds_key = str(phone) + '_identifyingCode'        identifyingCode = reds.get_data(key=reds_key)        reds_key_ttl = reds.get_ttl(key=reds_key)        if reds_key_ttl > 240 and identifyingCode:            # if identifyingCode :            return response.json(90)        user_qs = Device_User.objects.filter(username=phone)        phone_qs = Device_User.objects.filter(phone=phone)        if user_qs.exists() or phone_qs.exists():            return response.json(101)        identifyingCode = RandomStr(6, True)        # 短信签名        # sign_name_dict = {        #     'ansjer':'Ansjer',        #     'zosi':'周视'        # }        sign_ms = ''        if country_code == '86':            # 国内短信推送模板            temp_msg = 'SMS_151600991'            rec_phone = phone            if sign_name == 'zosi':                sign_ms = '周视'            else:                sign_ms = 'Ansjer'        else:            # 国际短信推送模板            temp_msg = 'SMS_172165867'            rec_phone = country_code + phone            sign_ms = 'Ansjer'        # 发送手机验证码        aliSms = AliSmsObject()        res = aliSms.send_code_sms(phone=rec_phone, code=identifyingCode, sign_name=sign_ms,                                   temp_msg=temp_msg)        print(res)        if res["Code"] == "OK":            # if reds.set_data(key=reds_key, val=identifyingCode, expire=60) is not True:            if reds.set_data(key=reds_key, val=identifyingCode, expire=300) is not True:                return response.json(10, '生成缓存系统错误')            return response.json(0)        else:            return response.json(10, res["Message"])# 验证码注册class v2registerView(TemplateView):    @method_decorator(csrf_exempt)    def dispatch(self, *args, **kwargs):        return super(v2registerView, self).dispatch(*args, **kwargs)    def post(self, request, *args, **kwargs):        request.encoding = 'utf-8'        request_dict = request.POST        return self.validates(request_dict)    def get(self, request, *args, **kwargs):        request.encoding = 'utf-8'        request_dict = request.GET        return self.validates(request_dict)    def validates(self, request_dict):        phone = request_dict.get('phone', None)        email = request_dict.get('email', None)        password = request_dict.get('password', None)        authcode = request_dict.get('authcode', None)        lang = request_dict.get('lang', None)        if not lang:            lang = request_dict.get('language', None)        response = ResponseObject(lang)        if password is None:            return response.json(444, 'password')        if authcode is None:            return response.json(444, 'identifyingCode')        if phone is not None:            return self.do_phone_register(phone, password, authcode, response)        elif email is not None:            return self.do_email_register(email, password, authcode, response)        else:            return response.json(444, 'phone or email')    def do_phone_register(self, phone, password, authcode, response):        data_valid = DataValid()        if data_valid.mobile_validate(phone) is not True:            return response.json(100)        if data_valid.password_validate(password) is not True:            return response.json(109)        reds = RedisObject()        identifyingCode = reds.get_data(key=phone + '_identifyingCode')        # 判断验证码是否过期        if identifyingCode is False:            return response.json(120)        # 验证码是否正确        if authcode != identifyingCode:            return response.json(121)        phone_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone))        # 是否已存在        if phone_qs.exists():            return response.json(101)        try:            users = Device_User.objects.create(                username=phone,                NickName=phone,                phone=phone,                password=make_password(password),                userID=CommonService.getUserID(μs=False, setOTAID=True),                is_active=True,                user_isValid=True,            )        except Exception as e:            errorInfo = traceback.format_exc()            print(errorInfo)            return response.json(424, repr(e))        else:            if not reds.del_data(key=phone + '_identifyingCode'):                return response.json(10, '删除缓存验证码错误')            return self.do_login(phone_qs, response)    def do_login(self, user_qs, response):        now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)        user_qs.update(last_login=now_time, online=True)        userID = user_qs[0].userID        print('userID' + userID)        tko = TokenObject()        user_list = user_qs.values("NickName", "userIconUrl", "userIconPath", "username", "userEmail", "phone")        res = tko.generate(data={'userID': userID, 'lang': response.lang, 'user': user_list[0]["username"]})        # 增加角色        user_qs[0].role.add(Role.objects.get(rid=1))        role_dict = ModelService.own_role(userID=userID)        res['rid'] = role_dict['rid']        res['roleName'] = role_dict['roleName']        res['permList'] = ModelService.own_permission(userID)        res['userID'] = userID        # 昵称,邮箱,电话,刷新,头像        userIconPath = str(user_list[0]["userIconPath"])        if userIconPath and userIconPath.find('static/') != -1:            userIconPath = userIconPath.replace('static/', '').replace('\\', '/')            res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath        else:            res['userIconUrl'] = ''        res['NickName'] = user_list[0]["NickName"] if user_list[0]["NickName"] is not None else ''        res['username'] = user_list[0]["username"] if user_list[0]["username"] is not None else ''        res['userEmail'] = user_list[0]["userEmail"] if user_list[0]["userEmail"] is not None else ''        res['phone'] = user_list[0]["phone"] if user_list[0]["phone"] is not None else ''        print(res)        return response.json(0, res)    def do_email_register(self, email, password, authcode, response):        data_valid = DataValid()        if data_valid.email_validate(email) is not True:            return response.json(105)        if data_valid.password_validate(password) is not True:            return response.json(109)        reds = RedisObject()        identifyingCode = reds.get_data(key=email + '_identifyingCode')        # 判断验证码是否过期        if identifyingCode is False:            return response.json(120)        # 验证码是否正确        if authcode != identifyingCode:            return response.json(121)        email_qs = Device_User.objects.filter(Q(userEmail=email) | Q(username=email))        # 是否已存在        if email_qs.exists():            return response.json(103)        try:            users = Device_User.objects.create(                username=email,                NickName=email,                userEmail=email,                password=make_password(password),                userID=CommonService.getUserID(μs=False, setOTAID=True),                is_active=True,                user_isValid=True,            )        except Exception as e:            errorInfo = traceback.format_exc()            print(errorInfo)            return response.json(424, repr(e))        else:            if not reds.del_data(key=email + '_identifyingCode'):                return response.json(10, '删除缓存验证码错误')            return self.do_login(email_qs, response)# 重置密码# 忘记密码获取验证码v2class v2forgetPwdCodeView(TemplateView):    @method_decorator(csrf_exempt)    def dispatch(self, *args, **kwargs):        return super(v2forgetPwdCodeView, self).dispatch(*args, **kwargs)    @ratelimit(key='ip', rate='1/m')    def get(self, request, *args, **kwargs):        request.encoding = 'utf-8'        request_dict = request.GET        lang = request_dict.get('lang')        response = ResponseObject(lang)        was_limited = getattr(request, 'limited', False)        if was_limited is True:            return response.json(5)        return self.ValidationError(request_dict, response)    @ratelimit(key='ip', rate='1/m')    def post(self, request):        request.encoding = 'utf-8'        request_dict = request.POST        lang = request_dict.get('lang')        response = ResponseObject(lang)        was_limited = getattr(request, 'limited', False)        if was_limited is True:            return response.json(5)        return self.ValidationError(request_dict, response)    def ValidationError(self, request_dict, response):        phone = request_dict.get('phone', None)        email = request_dict.get('email', None)        country_code = request_dict.get('country_code', None)        sign_name = request_dict.get('sign_name', None)        if phone is not None:            phone = phone.strip()            if country_code is None:                return self.do_send_phone_code(phone, response, sign_name)            else:                return self.do_v2_send_phone_code(country_code, phone, response, sign_name)        elif email is not None:            return self.do_send_email_code(email, response)        else:            return response.json(444, 'phone')    def do_send_phone_code(self, phone, response, sign_name):        data_valid = DataValid()        if data_valid.mobile_validate(phone) is not True:            return response.json(100)        user_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone))        if not user_qs.exists():            return response.json(102)        reds = RedisObject()        reds_key = str(phone) + '_forgetPwdResetCode'        resetCode = reds.get_data(key=reds_key)        reds_key_ttl = reds.get_ttl(key=reds_key)        if reds_key_ttl > 240 and resetCode:            # if identifyingCode :            return response.json(90)        # if resetCode is True:        #     return response.json(120)        resetCode = RandomStr(6, True)        if sign_name == 'zosi':            sign_ms = '周视'        else:            sign_ms = 'Ansjer'        aliSms = AliSmsObject()        res = aliSms.send_code_sms(phone=phone, code=resetCode, sign_name=sign_ms,                                   temp_msg='SMS_151675019')        if res["Code"] == "OK":            if not reds.set_data(key=reds_key, val=resetCode, expire=300):                return response.json(10, '生成缓存错误')            return response.json(0)        else:            return response.json(10, res["Message"])    def do_v2_send_phone_code(self, country_code, phone, response, sign_name):        data_valid = DataValid()        if data_valid.mobile_validate(phone) is not True:            return response.json(100)        user_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone))        if not user_qs.exists():            return response.json(102)        reds = RedisObject()        reds_key = str(phone) + '_forgetPwdResetCode'        resetCode = reds.get_data(key=reds_key)        reds_key_ttl = reds.get_ttl(key=reds_key)        if reds_key_ttl > 240 and resetCode:            # if identifyingCode :            return response.json(90)        # if resetCode is True:        #     return response.json(120)        resetCode = RandomStr(6, True)        aliSms = AliSmsObject()        if country_code == '86':            rec_phone = phone            temp_msg = 'SMS_151675019'            if sign_name == 'zosi':                sign_ms = '周视'            else:                sign_ms = 'Ansjer'        else:            temp_msg = 'SMS_172200051'            rec_phone = str(country_code) + str(phone)            sign_ms = 'Ansjer'        res = aliSms.send_code_sms(phone=rec_phone, code=resetCode, sign_name=sign_ms,                                   temp_msg=temp_msg)        if res["Code"] == "OK":            if not reds.set_data(key=reds_key, val=resetCode, expire=300):                return response.json(10, '生成缓存错误')            return response.json(0)        else:            return response.json(10, res["Message"])    def do_send_email_code(self, email, response):        data_valid = DataValid()        if data_valid.email_validate(email) is not True:            return response.json(105)        user_qs = Device_User.objects.filter(Q(userEmail=email) | Q(username=email))        if not user_qs.exists():            return response.json(104)        reds = RedisObject()        resetCode = reds.get_data(key=email + '_forgetPwdResetCode')        if resetCode is True:            return response.json(120)        resetCode = RandomStr(6, True)        if not reds.set_data(key=email + '_forgetPwdResetCode', val=resetCode, expire=600):            return response.json(10, '生成缓存错误')        send_data = TemplateService.email_message(type='forgetCode', language=response.lang)        send_body = send_data['body'].format(userEmail=email, email_valid_code=resetCode)        ses = SesClassObject()        # send_res = ses.send_email(send_address_list=[email], subject=send_data['title'], body=send_body)        # 阿里云发送邮箱        send_res = ses.alyEmailCode(send_address_list=[email], subject=send_data['title'], body=send_body)        if send_res is not True:            reds.del_data(key=email + '_forgetPwdResetCode')            return response.json(44)        return response.json(0)# 忘记密码v2class v2resetPwdByCodeView(TemplateView):    @method_decorator(csrf_exempt)    def dispatch(self, *args, **kwargs):        return super(v2resetPwdByCodeView, self).dispatch(*args, **kwargs)    def get(self, request, *args, **kwargs):        request.encoding = 'utf-8'        request_dict = request.GET        lang = request_dict.get('lang')        if not lang:            lang = request_dict.get('language', None)        response = ResponseObject(lang)        was_limited = getattr(request, 'limited', False)        if was_limited is True:            return response.json(5)        return self.ValidationError(request_dict, response)    def post(self, request):        request.encoding = 'utf-8'        request_dict = request.POST        lang = request_dict.get('lang')        if not lang:            lang = request_dict.get('language', None)        response = ResponseObject(lang)        was_limited = getattr(request, 'limited', False)        if was_limited is True:            return response.json(5)        return self.ValidationError(request_dict, response)    def ValidationError(self, request_dict, response):        phone = request_dict.get('phone', None)        email = request_dict.get('email', None)        password = request_dict.get('password', None)        authcode = request_dict.get('authcode', None)        if password is None or authcode is None:            return response.json(444, 'password,authcode')        authcode = authcode.strip()        password = password.strip()        if phone is not None:            phone = phone.strip()            return self.do_phone_pwd_reset(phone, authcode, password, response)        elif email is not None:            email = email.strip()            return self.do_email_pwd_reset(email, authcode, password, response)        else:            return response.json(444, 'phone')    def do_email_pwd_reset(self, email, authcode, password, response):        data_valid = DataValid()        if data_valid.email_validate(email) is not True:            return response.json(105)        if data_valid.password_validate(password) is not True:            return response.json(109)        user_qs = Device_User.objects.filter(Q(userEmail=email) | Q(username=email))        if not user_qs.exists():            return response.json(104)        reds = RedisObject()        resetCode = reds.get_data(key=email + '_forgetPwdResetCode')        if resetCode is False:            return response.json(90)        if authcode != resetCode:            return response.json(121)        # if not reds.set_data(key=email + '_forgetPwdResetCode', val=resetCode, expire=300):        #     return response.json(10, '生成缓存错误')        user_qs.update(password=make_password(password))        if not reds.del_data(email + '_forgetPwdResetCode'):            return response.json(10, '删除缓存失败')        return response.json(0)    def do_phone_pwd_reset(self, phone, authcode, password, response):        data_valid = DataValid()        if data_valid.mobile_validate(phone) is not True:            return response.json(100)        if data_valid.password_validate(password) is not True:            return response.json(109)        user_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone))        if not user_qs.exists():            return response.json(102)        reds = RedisObject()        resetCode = reds.get_data(key=phone + '_forgetPwdResetCode')        if resetCode is False:            return response.json(90)        if authcode != resetCode:            return response.json(121)        # if not reds.set_data(key=phone + '_forgetPwdResetCode', val=resetCode, expire=300):        #     return response.json(10, '生成缓存错误')        user_qs.update(password=make_password(password))        if not reds.del_data(phone + '_forgetPwdResetCode'):            return response.json(10, '删除缓存失败')        return response.json(0)    def do_login(self, user_qs, response):        now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)        user_qs.update(last_login=now_time, online=True)        userID = user_qs[0].userID        print('userID' + userID)        tko = TokenObject()        user_list = user_qs.values("NickName", "userIconUrl", "userIconPath", "username", "userEmail", "phone")        res = tko.generate(data={'userID': userID, 'lang': response.lang, 'user': user_list[0]["username"]})        # 增加角色        user_qs[0].role.add(Role.objects.get(rid=1))        role_dict = ModelService.own_role(userID=userID)        res['rid'] = role_dict['rid']        res['roleName'] = role_dict['roleName']        res['permList'] = ModelService.own_permission(userID)        res['userID'] = userID        # 昵称,邮箱,电话,刷新,头像        userIconPath = str(user_list[0]["userIconPath"])        if userIconPath and userIconPath.find('static/') != -1:            userIconPath = userIconPath.replace('static/', '').replace('\\', '/')            res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath        else:            res['userIconUrl'] = ''        res['NickName'] = user_list[0]["NickName"] if user_list[0]["NickName"] is not None else ''        res['username'] = user_list[0]["username"] if user_list[0]["username"] is not None else ''        res['userEmail'] = user_list[0]["userEmail"] if user_list[0]["userEmail"] is not None else ''        res['phone'] = user_list[0]["phone"] if user_list[0]["phone"] is not None else ''        print(res)        return response.json(0, res)# 登录class v2LoginView(TemplateView):    @method_decorator(csrf_exempt)  # @csrf_exempt    def dispatch(self, *args, **kwargs):        return super(v2LoginView, self).dispatch(*args, **kwargs)    @ratelimit(key='ip', rate='5/m')    def post(self, request, *args, **kwargs):        request.encoding = 'utf-8'        request_dict = request.POST        language = request_dict.get('language', 'en')        response = ResponseObject(language)        was_limited = getattr(request, 'limited', False)        if was_limited is True:            return response.json(5)        return self.validates(request_dict, response)    # @ratelimit(key='ip', rate='5/m')    def get(self, request, *args, **kwargs):        request.encoding = 'utf-8'        request_dict = request.GET        language = request_dict.get('language', 'en')        response = ResponseObject(language)        was_limited = getattr(request, 'limited', False)        if was_limited is True:            return response.json(5)        return self.validates(request_dict, response)    def validates(self, request_dict, response):        username = request_dict.get('userName', None)        password = request_dict.get('userPwd', None)        if not username or not password:            return response.json(111)        username = username.strip()        password = password.strip()        data_valid = DataValid()        if data_valid.email_validate(username):            return self.do_email_login(username, password, response)        elif data_valid.mobile_validate(username):            return self.do_phone_login(username, password, response)        elif data_valid.name_validate(username):            return self.do_name_login(username, password, response)        else:            return response.json(107)    def do_email_login(self, email, password, response):        user_qs = Device_User.objects.filter(Q(username=email) | Q(userEmail=email))        return self.valid_login(user_qs, password, response)    def do_phone_login(self, phone, password, response):        user_qs = Device_User.objects.filter(Q(phone=phone) | Q(username=phone), is_active=True, user_isValid=True)        return self.valid_login(user_qs, password, response)    def do_name_login(self, username, password, response):        user_qs = Device_User.objects.filter(Q(username=username) | Q(phone=username) | Q(userEmail=username),                                             is_active=True, user_isValid=True)        return self.valid_login(user_qs, password, response)    def valid_login(self, user_qs, password, response):        if not user_qs.exists():            return response.json(104)        # users = user_qs.values('role__rid', 'role__roleName', 'userID', 'role', 'NickName', 'username', 'userEmail',        #                        'phone', 'password', 'userIconPath', 'user_isValid', 'is_active')[0]        users = user_qs.values('role__rid', 'role__roleName', 'userID', 'NickName', 'username', 'userEmail',                               'phone', 'password', 'userIconPath')[0]        if not check_password(password, users['password']):            return response.json(111)        userID = users['userID']        tko = TokenObject()        res = tko.generate(            data={'userID': userID, 'lang': response.lang, 'user': users['username'], 'm_code': '123413243214'})        if tko.code == 0:            now_time = datetime.datetime.utcnow().replace(tzinfo=utc).astimezone(utc)            user_qs.update(last_login=now_time, language=response.lang)            res['rid'] = users['role__rid']            res['roleName'] = users['role__roleName']            res['permList'] = ModelService.own_permission(userID)            res['userID'] = userID            # 昵称,邮箱,电话,刷新,头像            userIconPath = str(users['userIconPath'])            if userIconPath and userIconPath.find('static/') != -1:                userIconPath = userIconPath.replace('static/', '').replace('\\', '/')                res['userIconUrl'] = SERVER_DOMAIN + 'account/getAvatar/' + userIconPath            else:                res['userIconUrl'] = ''            res['NickName'] = users['NickName'] if users['NickName'] is not None else ''            res['username'] = users['username'] if users['username'] is not None else ''            res['userEmail'] = users['userEmail'] if users['userEmail'] is not None else ''            res['phone'] = users['phone'] if users['phone'] is not None else ''            return response.json(0, res)        else:            return response.json(tko.code)# 用户登录后初始化接口class InitInfoView(View):    def get(self, request, *args, **kwargs):        request.encoding = 'utf-8'        return self.validation(request.GET)    def post(self, request, *args, **kwargs):        request.encoding = 'utf-8'        return self.validation(request.POST)    def validation(self, request_dict):        response = ResponseObject()        token = request_dict.get('token', None)        tko = TokenObject(token)        if tko.code == 0:            userID = tko.userID            return self.init_uid(request_dict, userID, response)        else:            return response.json(tko.code)    # 初始化设备token    def init_uid(self, request_dict, userID, response):        # 未读的系统消息        sm_count = SysMsgModel.objects.filter(userID_id=userID, status=0).count()        token_val = request_dict.get('token_val', None)        m_code = request_dict.get('m_code', None)        push_type = request_dict.get('push_type', None)        appBundleId = request_dict.get('appBundleId', None)        tz = request_dict.get('tz', None)        if all([token_val, push_type, appBundleId]):            if push_type == 0:                if appBundleId not in APNS_CONFIG.keys():                    return response.json(904)            elif push_type == 1:                if appBundleId not in FCM_CONFIG.keys():                    return response.json(904)            elif push_type == 2:                if appBundleId not in JPUSH_CONFIG.keys():                    return response.json(904)            else:                return response.json(173)            uid_list = ModelService.get_uid_list(userID=userID)            if uid_list:                uid_set_id_list = UidSetModel.objects.filter(uid__in=uid_list, detect_status=1).values_list('id',                                                                                                            flat=True)                if uid_set_id_list:                    update_dict = {                        'token_val': token_val,                        'push_type': push_type,                        'tz': tz                    }                    UidPushModel.objects.filter(userID_id=userID, m_code=m_code, uid_set_id__in=uid_set_id_list). \                        update(**update_dict)        res = {            'usmsg': sm_count        }        return response.json(0, res)# 获取验证码class verifyAuthcode(TemplateView):    def post(self, request, *args, **kwargs):        request.encoding = 'utf-8'        lang = request.POST.get('lang', None)        if not lang:            lang = request.POST.get('language', None)        response = ResponseObject(lang)        request_dict = request.POST        return self.ValidationError(request_dict, response)    def get(self, request, *args, **kwargs):        # Device_User.objects.filter(userEmail='chanjunkai@163.com').delete()        request.encoding = 'utf-8'        lang = request.GET.get('lang', None)        if not lang:            lang = request.GET.get('language', None)        response = ResponseObject(lang)        request_dict = request.GET        return self.ValidationError(request_dict, response)    def ValidationError(self, request_dict, response):        email = request_dict.get('email', None)        phone = request_dict.get('phone', None)        authcode = request_dict.get('authcode', None)        if email is not None:            email = email.strip()            return self.email_validate(email, authcode, response)        elif phone is not None:            phone = phone.strip()            return self.phone_validate(phone, authcode, response)        else:            return response.json(444)    def email_validate(self, email, authcode, response):        data_valid = DataValid()        if data_valid.email_validate(email) is not True:            return response.json(105)        reds = RedisObject()        resetCode = reds.get_data(key=email + '_forgetPwdResetCode')        if resetCode is False:            return response.json(120)        if authcode != resetCode:            return response.json(121)        return response.json(0)    def phone_validate(self, phone, authcode, response):        data_valid = DataValid()        if data_valid.mobile_validate(phone) is not True:            return response.json(100)        reds = RedisObject()        resetCode = reds.get_data(key=phone + '_forgetPwdResetCode')        if resetCode is False:            return response.json(120)        print(resetCode)        if authcode != resetCode:            return response.json(121)        return response.json(0)
 |