判断主用户

Controller/UserController.py

@@ -3392,16 +3392,30 @@ class SingleLoginView(TemplateView):
         sign_name = request_dict.get('sign_name', None)
         country_code = request_dict.get('country_code', None)
         code_type = request_dict.get('code_type', None)
+        uid = request_dict.get('uid', None)
         if not any([email, phone]):
             return response.json(444)
         try:
+            # 查询主用户信息
+            device_info_qs = Device_Info.objects.filter(UID=uid).values('vodPrimaryMaster')
+            if not device_info_qs.exists():
+                return response.json(173)
+            primary_master = device_info_qs[0]['vodPrimaryMaster']
             # 邮箱验证码
             if email is not None:
                 email = email.strip()
+                if uid is None:
+                    return response.json(444)
+                # 判断是否为主用户
+                if email != primary_master:
+                    return response.json(12)
                 return self.get_aliyun_email_code(email, code_type, response)
             # 手机验证码
             else:
                 phone = phone.strip()
+                # 判断是否为主用户
+                if phone != primary_master:
+                    return response.json(12)
                 country_code = str(country_code.strip())
                 # 短信签名
                 sign_name = CommonService.confirm_msg_sign_name(sign_name)