Explorar o código

v2forgetPwdCodeView add sign_name

chenjunkai %!s(int64=6) %!d(string=hai) anos
pai
achega
0c146e1c9f
Modificáronse 1 ficheiros con 17 adicións e 6 borrados
  1. 17 6
      Controller/UserController.py

+ 17 - 6
Controller/UserController.py

@@ -909,18 +909,19 @@ class v2forgetPwdCodeView(TemplateView):
         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)
+                return self.do_send_phone_code(phone, response,sign_name)
             else:
-                return self.do_v2_send_phone_code(country_code, phone, response)
+                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):
+    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)
@@ -937,8 +938,12 @@ class v2forgetPwdCodeView(TemplateView):
         # 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='Ansjer',
+        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):
@@ -947,7 +952,7 @@ class v2forgetPwdCodeView(TemplateView):
         else:
             return response.json(10, res["Message"])
 
-    def do_v2_send_phone_code(self, country_code, phone, response):
+    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)
@@ -968,10 +973,16 @@ class v2forgetPwdCodeView(TemplateView):
         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)
-        res = aliSms.send_code_sms(phone=rec_phone, code=resetCode, sign_name='Ansjer',
+            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):