企业用户 修改邮箱

CompanyUser/CompanyUser_impl.py

@@ -183,7 +183,6 @@ class CompanyUserImpl(CompanyUser):
             "avatar_06": "623152edf36000004f001253"
         }
         self.avatar_id = avatars[avatar_option]
-
         self.db.update_single_data(
             "应用端",
             "企业用户",
@@ -195,10 +194,20 @@ class CompanyUserImpl(CompanyUser):
         """修改登录密码"""
         self.check_email_not_registered()
         self.check_vcode_correct(code)
-
         self.db.update_single_data(
             "应用端",
             "企业用户",
             {"邮箱": self.email},
             self.dict_to_save(columns=["密码"])
         )
+
+    def change_email(self, code):
+        """更换邮箱"""
+        self.check_email_registered()
+        self.check_vcode_correct(code)
+        self.db.update_single_data(
+            "应用端",
+            "企业用户",
+            {"企业ID": self.cid},
+            self.dict_to_save(columns=["邮箱"])
+        )

CompanyUser/CompanyUser_obj.py

@@ -103,7 +103,7 @@ class CompanyUser(object):
     def change_password(self, **kwargs):
         """修改登录密码"""
 
-    def change_email(self):
+    def change_email(self, **kwargs):
         """修改登录邮箱"""
 
 

CompanyUser/CompanyUser_route.py

@@ -107,3 +107,21 @@ def change_pwd():
         return {"info": e.failed_info}, e.status_code
     except KeyError:
         return {"info": "参数异常"}, 400
+
+
+@company_user_route.route('/change_email', methods=['POST'])
+@check_block
+@verify_token
+def change_email(**kwargs):
+    """更换账号邮箱"""
+    try:
+        req = request.json
+        company_user = CompanyUserImpl()
+        company_user.cid = kwargs['cid']
+        company_user.email = req['email']
+        info = company_user.change_email(req['code'])
+        return {"info": info}, 200
+    except ReturnConditionCheckFailed as e:
+        return {"info": e.failed_info}, e.status_code
+    except KeyError:
+        return {"info": "参数异常"}, 400