|
@@ -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
|
|
|
|
|
|
|