make it reply to self

weaming · weaming · commit 6a7fe6bd0163 · 2016-03-05T18:21:37.000+08:00
diff --git a/bot.py b/bot.py
@@ -5,12 +5,12 @@
 import ConfigParser
 import json
 
-
 class TulingWXBot(WXBot):
     def __init__(self):
         WXBot.__init__(self)
 
         self.tuling_key = ""
+        self.robot_switch = True
 
         try:
             cf = ConfigParser.ConfigParser()
@@ -35,13 +35,35 @@ def tuling_auto_reply(self, uid, msg):
             else:
                 result = respond['text'].replace('<br>', '  ')
 
+            print '    ROBOT:', result
             return result
         else:
             return u"&#30693;&#36947;&#21862;"
 
+    def button(self, msg):
+        msg_data = msg['content']['data']
+        STOP = [u'&#36864;&#19979;',u'&#36208;&#24320;',u'&#20851;&#38381;',u'&#20851;&#25481;',u'&#20241;&#24687;',u'&#28378;&#24320;']
+        START = [u'&#20986;&#26469;',u'&#21551;&#21160;',u'&#24037;&#20316;']
+        if self.robot_switch:
+            for i in STOP:
+                if i in msg_data :
+                    self.robot_switch = False
+                    self.send_msg_by_uid(u'[Robot]' + u'&#26426;&#22120;&#20154;&#24050;&#20851;&#38381;&#65281;', msg['to_user_id'])
+        else:
+            for i in START:
+                if i in msg_data:
+                    self.robot_switch = True
+
     def handle_msg_all(self, msg):
-        if msg['msg_type_id'] == 4 and msg['content']['type'] == 0:  # text message from contact
-            self.send_msg_by_uid(self.tuling_auto_reply(msg['user']['id'], msg['content']['data']), msg['user']['id'])
+        #print 'MSG ID:', msg['msg_type_id']
+        if msg['msg_type_id'] == 1 and msg['content']['type'] == 0: # replay to self
+            self.button(msg)
+            if self.robot_switch:
+                self.send_msg_by_uid(u'[Robot]' + self.tuling_auto_reply(msg['user']['id'], msg['content']['data']), msg['to_user_id'])
+        elif msg['msg_type_id'] == 4 and msg['content']['type'] == 0:  # text message from contact
+            self.button(msg)
+            if self.robot_switch:
+                self.send_msg_by_uid(u'[Robot]' + self.tuling_auto_reply(msg['user']['id'], msg['content']['data']), msg['user']['id'])
         elif msg['msg_type_id'] == 3:  # group message
             if msg['content']['data'].find('@') >= 0:  # someone @ another
                 my_names = self.get_group_member_name(msg['user']['id'], self.user['UserName'])
@@ -86,8 +108,10 @@ def main():
     bot = TulingWXBot()
     bot.DEBUG = True
     bot.conf['qr'] = 'png'
+
     bot.run()
 
 
 if __name__ == '__main__':
     main()
+
diff --git a/wxbot.py b/wxbot.py
@@ -422,6 +422,7 @@ def handle_msg(self, r):
             message = {'msg_type_id': msg_type_id,
                        'msg_id': msg['MsgId'],
                        'content': content,
+                       'to_user_id': msg['ToUserName'],
                        'user': user}
             self.handle_msg_all(message)
 
