From ad13e4c1f67ed328bf8609209b193bf6a739276f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E6=80=9D=E5=B7=9D?= <sichanwong0527@gmail.com>
Date: Thu, 7 Apr 2022 05:13:31 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=81=E4=B8=9A=E7=94=A8=E6=88=B7=20?=
 =?UTF-8?q?=E4=BF=AE=E6=94=B9=E9=82=AE=E7=AE=B1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 CompanyUser/CompanyUser_impl.py  | 13 +++++++++++--
 CompanyUser/CompanyUser_obj.py   |  2 +-
 CompanyUser/CompanyUser_route.py | 18 ++++++++++++++++++
 3 files changed, 30 insertions(+), 3 deletions(-)

diff --git a/CompanyUser/CompanyUser_impl.py b/CompanyUser/CompanyUser_impl.py
index d4346cb..0b7cf21 100644
--- a/CompanyUser/CompanyUser_impl.py
+++ b/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=["邮箱"])
+        )
diff --git a/CompanyUser/CompanyUser_obj.py b/CompanyUser/CompanyUser_obj.py
index 6698b62..448ec7f 100644
--- a/CompanyUser/CompanyUser_obj.py
+++ b/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):
         """修改登录邮箱"""
 
 
diff --git a/CompanyUser/CompanyUser_route.py b/CompanyUser/CompanyUser_route.py
index 63483a9..f2c363a 100644
--- a/CompanyUser/CompanyUser_route.py
+++ b/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