소스 검색

Update user_manager: add staff gender

StrayWarrior 2 주 전
부모
커밋
ae0a36b0e1
1개의 변경된 파일4개의 추가작업 그리고 1개의 파일을 삭제
  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:
         if not self.staff_table:
             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}'"
         data = self.db.select(sql, pymysql.cursors.DictCursor)
         if not data:
             logger.error(f"staff[{staff_id}] not found")
             return {}
         profile = data[0]
+        # 转换性别格式
+        gender_map = {0: '未知', 1: '男', 2: '女'}
+        profile['agent_gender'] = gender_map[profile['agent_gender']]
         return profile