Add function get_user_id.

cbdcyb · Mar 5, 2016 · f824fff · f824fff
1 parent cf42400
commit f824fff
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/wxbot.py b/wxbot.py
@@ -484,6 +484,16 @@ def send_msg_by_uid(self, word, dst='filehelper'):
         dic = r.json()
         return dic['BaseResponse']['Ret'] == 0
 
+    def get_user_id(self, name):
+        for contact in self.contact_list:
+            if 'RemarkName' in contact and contact['RemarkName'] == name:
+                return contact['UserName']
+            elif 'NickName' in contact and contact['NickName'] == name:
+                return contact['UserName']
+            elif 'DisplayName' in contact and contact['DisplayName'] == name:
+                return contact['UserName']
+        return ''
+
     def send_msg(self, name, word, isfile=False):
         uid = self.get_user_id(name)
         if uid: