wxBot/newYearBot.py

#!/usr/bin/env python
# coding: utf-8

from wxbot import *
import ConfigParser
import json
import logging
import logging.config

class TulingWXBot(WXBot):
    def __init__(self):
        WXBot.__init__( )
        
        self.qrtype = ""
        self.robot_switch = True
        
        try:
            cf = ConfigParser.ConfigParser()
            cf.read('conf/conf.ini')
            self.qrtype = cf.get('qrtype', 'key')
        except Exception:
            pass
        logging.info('qrtype:'+self.tuling_key)

# 自动回复函数self：发送方用户id, uid, msg
    def auto_reply(self, uid, msg):
            return u"新年快乐，我是天天，主人马上就来！"

    def auto_switch(self, msg):
        msg_data = msg['content']['data']
        stop_cmd = [u'退下', u'走开', u'关闭', u'关掉', u'休息', u'滚开']
        start_cmd = [u'出来', u'启动', u'工作']
        if self.robot_switch:
            for i in stop_cmd:
                if i == msg_data:
                    self.robot_switch = False
                    self.send_msg_by_uid(u'[Robot]' + u'机器人已关闭！', msg['to_user_id'])
        else:
            for i in start_cmd:
                if i == msg_data:
                    self.robot_switch = True
                    self.send_msg_by_uid(u'[Robot]' + u'机器人已开启！', msg['to_user_id'])
# msg   dict: {'content': {'data': u'\u53bb', 'type': 0}, 'msg_id': u'3808887025959048251', 'msg_type_id': 4, 'to_user_id': u'@97745d23f2d3bbe5241cede14a3d8baa45809a1842e762b5a33fc1353f6efa3a', 'user': {'id': u'@f95a572b1284a0f75dd550a4ed9a93d788ded42a856db023cd3494faab623311', 'name': u'\u5929\u95ee'}}
    def handle_msg_all(self, msg):
        if not self.robot_switch and msg['msg_type_id'] != 1:
            return
        if msg['msg_type_id'] == 1 and msg['content']['type'] == 0:  # reply to self
            self.auto_switch(msg)
            # msg_type_id=4 发送文字
        elif 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'])
        elif msg['msg_type_id'] == 3 and msg['content']['type'] == 0:  # group text message
            if 'detail' in msg['content']:
                my_names = self.get_group_member_name(self.my_account['UserName'], msg['user']['id'])
                if my_names is None:
                    my_names = {}
                if 'NickName' in self.my_account and self.my_account['NickName']:
                    my_names['nickname2'] = self.my_account['NickName']
                if 'RemarkName' in self.my_account and self.my_account['RemarkName']:
                    my_names['remark_name2'] = self.my_account['RemarkName']

                is_at_me = False
                for detail in msg['content']['detail']:
                    if detail['type'] == 'at':
                        for k in my_names:
                            if my_names[k] and my_names[k] == detail['value']:
                                is_at_me = True
                                break
                if is_at_me:
                    src_name = msg['content']['user']['name']
                    reply = 'to ' + src_name + ': '
                    if msg['content']['type'] == 0:  # text message
                        reply += self.tuling_auto_reply(msg['content']['user']['id'], msg['content']['desc'])
                    else:
                        reply += u"对不起，只认字，其他杂七杂八的我都不认识，,,Ծ‸Ծ,,"
                    self.send_msg_by_uid(reply, msg['user']['id'])
        else:
            src_name = msg['user']['name']
            reply = 'to ' + src_name + ': '
            reply += u"主人马上就来！"
            #self.send_msg_by_uid(reply, msg['user']['id'])

'''
主函数，执行TulingWXBot类里面的run方法
'''
def main():
    logging.config.fileConfig("conf/logger.conf")
    logger=logging.getLogger("example01")
    bot = TulingWXBot()
    bot.DEBUG = True
    bot.conf['qr'] = bot.qrtype
    
    bot.run()


if __name__ == '__main__':
    main()