Ver código fonte

Update user_manager: add staff gender

StrayWarrior 2 semanas atrás
pai
commit
ae0a36b0e1
1 arquivos alterados com 4 adições e 1 exclusões
  1. 4 1
      user_manager.py

+ 4 - 1
user_manager.py

@@ -141,13 +141,16 @@ class MySQLUserManager(UserManager):
     def get_staff_profile(self, staff_id) -> Dict:
     def get_staff_profile(self, staff_id) -> Dict:
         if not self.staff_table:
         if not self.staff_table:
             raise Exception("staff_table is not set")
             raise Exception("staff_table is not set")
-        sql = f"SELECT agent_name, agent_age, agent_region " \
+        sql = f"SELECT agent_name, agent_gender, agent_age, agent_region " \
               f"FROM {self.staff_table} WHERE third_party_user_id = '{staff_id}'"
               f"FROM {self.staff_table} WHERE third_party_user_id = '{staff_id}'"
         data = self.db.select(sql, pymysql.cursors.DictCursor)
         data = self.db.select(sql, pymysql.cursors.DictCursor)
         if not data:
         if not data:
             logger.error(f"staff[{staff_id}] not found")
             logger.error(f"staff[{staff_id}] not found")
             return {}
             return {}
         profile = data[0]
         profile = data[0]
+        # 转换性别格式
+        gender_map = {0: '未知', 1: '男', 2: '女'}
+        profile['agent_gender'] = gender_map[profile['agent_gender']]
         return profile
         return profile