bot.py 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  1. #!/usr/bin/env python
  2. # coding: utf-8
  3. from wxbot import *
  4. import ConfigParser
  5. import json
  6. class TulingWXBot(WXBot):
  7. def __init__(self):
  8. WXBot.__init__(self)
  9. self.tuling_key = ""
  10. self.robot_switch = True
  11. try:
  12. cf = ConfigParser.ConfigParser()
  13. cf.read('conf.ini')
  14. self.tuling_key = cf.get('main', 'key')
  15. except Exception:
  16. pass
  17. print 'tuling_key:', self.tuling_key
  18. def tuling_auto_reply(self, uid, msg):
  19. if self.tuling_key:
  20. url = "http://www.tuling123.com/openapi/api"
  21. user_id = uid.replace('@', '')[:30]
  22. body = {'key': self.tuling_key, 'info': msg.encode('utf8'), 'userid': user_id}
  23. r = requests.post(url, data=body)
  24. respond = json.loads(r.text)
  25. result = ''
  26. if respond['code'] == 100000:
  27. result = respond['text'].replace('<br>', ' ')
  28. elif respond['code'] == 200000:
  29. result = respond['url']
  30. else:
  31. result = respond['text'].replace('<br>', ' ')
  32. print ' ROBOT:', result
  33. return result
  34. else:
  35. return u"知道啦"
  36. def auto_switch(self, msg):
  37. msg_data = msg['content']['data']
  38. stop_cmd = [u'退下', u'走开', u'关闭', u'关掉', u'休息', u'滚开']
  39. start_cmd = [u'出来', u'启动', u'工作']
  40. if self.robot_switch:
  41. for i in stop_cmd:
  42. if i == msg_data:
  43. self.robot_switch = False
  44. self.send_msg_by_uid(u'[Robot]' + u'机器人已关闭!', msg['to_user_id'])
  45. else:
  46. for i in start_cmd:
  47. if i == msg_data:
  48. self.robot_switch = True
  49. self.send_msg_by_uid(u'[Robot]' + u'机器人已开启!', msg['to_user_id'])
  50. def handle_msg_all(self, msg):
  51. if not self.robot_switch and msg['msg_type_id'] != 1:
  52. return
  53. if msg['msg_type_id'] == 1 and msg['content']['type'] == 0: # reply to self
  54. self.auto_switch(msg)
  55. elif msg['msg_type_id'] == 4 and msg['content']['type'] == 0: # text message from contact
  56. self.send_msg_by_uid(self.tuling_auto_reply(msg['user']['id'], msg['content']['data']), msg['user']['id'])
  57. elif msg['msg_type_id'] == 3 and msg['content']['type'] == 0: # group text message
  58. if 'detail' in msg['content']:
  59. my_names = self.get_group_member_name(self.my_account['UserName'], msg['user']['id'])
  60. if my_names is None:
  61. my_names = {}
  62. if 'NickName' in self.my_account and self.my_account['NickName']:
  63. my_names['nickname2'] = self.my_account['NickName']
  64. if 'RemarkName' in self.my_account and self.my_account['RemarkName']:
  65. my_names['remark_name2'] = self.my_account['RemarkName']
  66. is_at_me = False
  67. for detail in msg['content']['detail']:
  68. if detail['type'] == 'at':
  69. for k in my_names:
  70. if my_names[k] and my_names[k] == detail['value']:
  71. is_at_me = True
  72. break
  73. if is_at_me:
  74. src_name = msg['content']['user']['name']
  75. reply = 'to ' + src_name + ': '
  76. if msg['content']['type'] == 0: # text message
  77. reply += self.tuling_auto_reply(msg['content']['user']['id'], msg['content']['desc'])
  78. else:
  79. reply += u"对不起,只认字,其他杂七杂八的我都不认识,,,Ծ‸Ծ,,"
  80. self.send_msg_by_uid(reply, msg['user']['id'])
  81. def main():
  82. bot = TulingWXBot()
  83. bot.DEBUG = True
  84. bot.conf['qr'] = 'png'
  85. bot.run()
  86. if __name__ == '__main__':
  87. main()