search_contacts方法添加模糊搜索参数

sailor0913 · Sep 6, 2022 · 91d1257 · 91d1257
1 parent 254123f
commit 91d1257
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ntchat/core/wechat.py b/ntchat/core/wechat.py
@@ -200,7 +200,8 @@ def search_contacts(self,
                         wxid: Union[None, str] = None,
                         account: Union[None, str] = None,
                         nickname: Union[None, str] = None,
-                        remark: Union[None, str] = None):
+                        remark: Union[None, str] = None,
+                        fuzzy_search: bool = False):
         """
         根据wxid、微信号、昵称和备注模糊搜索联系人
         """
@@ -217,8 +218,9 @@ def search_contacts(self,
             return []
 
         cond_pairs = []
+        tag = '%' if fuzzy_search else ''
         for k, v in conds.items():
-            cond_pairs.append(f"{k} like '%{v}%'")
+            cond_pairs.append(f"{k} like '{tag}{v}{tag}'")
 
         cond_str = " or ".join(cond_pairs)
         sql = f"select username from contact where {cond_str}"