微信聊天机器人2019

昨天，突然想试试把自己的微信号变成聊天机器人逗逗女友，于是搜集了一些资料，弄了这么个东东。目前可以调用小冰回复文字信息，可以调用baidu-aip识别图片信息并返回结果。图片识别分为通用识别和动物识别两种，要启用动物识别，先发送“动物识别”即可。效果如下：

文字聊天

图片识别

代码部分

新建一个名为wechatRobot.py的文件，内容如下：

import urllib.parse
import urllib.request
from os import remove

import itchat
import requests
from aip import AipImageClassify
from itchat.content import *


class Robot:
    
    def __init__(self, whiteList, robot):
        """填写你的baidu-aip信息"""
        APP_ID = '你的APP_ID'
        API_KEY = '你的API_KEY'
        SECRET_KEY = '你的SECRET_KEY'
        self._client = AipImageClassify(APP_ID, API_KEY, SECRET_KEY)
        self._whiteList = whiteList
        self._robot = robot
        self._pic_class = 'general'
    
    def run(self):
        client = self._client
        whiteList = self._whiteList
        robot = self._robot

        @itchat.msg_register(TEXT)
        def text_reply(msg):
            nickName = msg.user.nickName
            if(msg.text[0:1]=="#"):
                print('本人消息，忽略。')
                return
            if(nickName not in whiteList):
                print('非白名单用户')
                return
            elif(msg.text=="动物识别"):
                self._pic_class = 'animal'
                return '发张小动物的图看看~'
            else:
                robotChat(msg)

        def robotChat(msg):   # 机器人选择
            if robot=='ice':
                iceChat(msg)
                return
            if robot=='tuling':
                tulingChat(msg)
                return

        def iceChat(msg):   # 微软小冰
            print('ice chat')
            print(msg['Text'])
            chatter = msg.user
            sendMsg = msg['Text'].strip()
            try:
                r = requests.get('https://www4.bing.com/socialagent/chat?q=' + sendMsg+'&anid=123456') #小冰接口
                try:
                    r1 = r.json()
                    info = urllib.parse.unquote(r1['InstantMessage']['ReplyText'])
                    print(info)
                    chatter.send(info)
                except Exception as e2:
                    print(e2)
            except Exception as e:
                print(e)

        def tulingChat(msg):   # 图灵机器人
            print('tuling chat')
            print(msg['Text'])
            chatter = msg.user
            sendMsg = msg['Text'].strip()
            try:
                api_url = 'http://www.tuling123.com/openapi/api'
                data = {
                    'key': '你的图灵机器人key',
                    'info': sendMsg,
                    'userid': 'wechat-robot',
                }
                r = requests.post(api_url, data=data)
                try:
                    r1 = r.json()
                    info = r1.get('text')
                    print(info)
                    chatter.send(info)
                except Exception as e2:
                    print(e2)
            except Exception as e:
                print(e)

        @itchat.msg_register(PICTURE)
        def picture_reply(msg):
            chatter = msg.user
            msg.download(msg.fileName)
            image = get_file_content(msg.fileName)
            if(self._pic_class=='general'):
                general(chatter, image)
            elif(self._pic_class=='animal'):
                animal(chatter, image)
                self._pic_class = 'general'
            remove(msg.fileName)

        # 通用识别
        def general(chatter, image):
            options = {}
            options["baike_num"] = 5
            info = client.advancedGeneral(image, options)
            print(info)
            root = info['result'][0]['root']
            keyword = info['result'][0]['keyword']
            baike_info = info['result'][0]['baike_info']
            reply_str = '这是%s\n' % keyword
            if(baike_info):
                reply_str += '百科：%s\n' % baike_info['description']
                reply_str += '百科链接：%s' % baike_info['baike_url']
                if(baike_info['image_url']):
                    urllib.request.urlretrieve(baike_info['image_url'], 'reply.png')
                    chatter.send('@img@%s' % 'reply.png')
            chatter.send(reply_str)
        
        # 动物识别
        def animal(chatter, image):
            options = {}
            options["top_num"] = 3
            options["baike_num"] = 5
            info = client.animalDetect(image, options)
            print(info)
            name = info['result'][0]['name']
            baike_info = info['result'][0]['baike_info']
            reply_str = '这是%s\n' % name
            if(baike_info):
                reply_str += '百科：%s\n' % baike_info['description']
                reply_str += '百科链接：%s' % baike_info['baike_url']
                if(baike_info['image_url']):
                    urllib.request.urlretrieve(baike_info['image_url'], 'reply.png')
                    chatter.send('@img@%s' % 'reply.png')
            chatter.send(reply_str)

        # 读取图片
        def get_file_content(filePath):
            with open(filePath, 'rb') as fp:
                return fp.read()
        
        itchat.auto_login(hotReload=True)
        itchat.run()

再新建一个名为main.py的文件，内容如下：

from wechatRobot import Robot


whiteList = ['昵称1','昵称2','昵称3','昵称4']   # 聊天白名单，名字是微信昵称

myRobot = Robot(whiteList=whiteList, robot='ice')   # 创建一个自己的聊天机器人
myRobot.run()   # 启动聊天机器人

运行

在命令行输入：

python3 main.py

会弹出一个二维码，用手机微信扫描此二维码登录微信，之后你的微信账号就变成了聊天机器人，白名单中的用户和你聊天就会触发聊天机器人~