浏览代码

Fix user_manager get_user_tags sql

StrayWarrior 2 天之前
父节点
当前提交
e9709f5f4e
共有 1 个文件被更改,包括 4 次插入1 次删除
  1. 4 1
      pqai_agent/user_manager.py

+ 4 - 1
pqai_agent/user_manager.py

@@ -273,8 +273,11 @@ class MySQLUserManager(UserManager):
             sql = f"""
                 SELECT a.third_party_user_id, b.tag_id, b.name FROM qywx_user_tag a
                     JOIN qywx_tag b ON a.tag_id = b.tag_id
-                    AND a.third_party_user_id IN {str(tuple(batch_user_ids))}
                 """
+            if len(batch_user_ids) == 1:
+                sql += f" AND a.third_party_user_id = '{batch_user_ids[0]}'"""
+            else:
+                sql += f" AND a.third_party_user_id IN {str(tuple(batch_user_ids))}"
             rows = self.db.select(sql, pymysql.cursors.DictCursor)
             # group by user_id
             for row in rows: