2

30行代码实现微信自动回复机器人

一、写在前面

今天下午又学到了itchat另一种有趣的玩法---微信自动回复机器人。

程序很简单仅仅三十行代码左右,实现了机器人自动与你的微信好友聊天,下面是我的机器人小笼包跟自己微信好友聊天的截图。

img

精心准备了大量学习资料,加入千人交流群:923414804获取大量学习资料与入门教程。

二、程序介绍

本程序通过itchat获取微信的相关消息,将微信消息传输到机器人接口(本程序使用的是图灵),获取机器人的返回消息。将返回消息返回给微信消息的发送人。以此实现将微信个人号变身为聊天机器人。

三、具体实现

安装

itchat的安装可以通过下面这行代码

pip install itchat

登录
itchat.auto_login() 这种方法将会通过微信扫描二维码登录,但是这种登录的方式确实短时间的登录,并不会保留登录的状态,也就是下次登录时还是需要扫描二维码。

# 扫描二维码登陆微信,实际上就是通过网页版微信登陆
    itchat.auto_login()

如果加上hotReload==True,那么就会保留登录的状态,至少在后面的几次登录过程中不会再次扫描二维码,该参数生成一个静态文件itchat.pkl用于存储登录状态

itchat.auto_login(hotReload=True)

微信信息的发送
微信可以发送各类消息,文本、图片、文件等,下面是具体的方法的介绍。

itchat.send('Message Content', 'toUserName')

该发送消息的函数需要两个参数,消息的内容与接受者的UserName,即标识符。
用下面的代码举一个小栗子,实现向文件助手发送信息。

import itchat

itchat.auto_login(hotReload=True)
itchat.send(u'测试消息发送', 'filehelper')   

微信消息的获取

itchat 将根据接受到的消息类型寻找对应的已注册的方法。如果一个消息类型没有对应的注册方法, 该消息将会被舍弃。在运行过程中也可以动态注册方法, 注册方式与结果不变。注册方法不带具体对象注册, 将注册为普通消息的回复方法.

    import itchat
    from itchat.content import *
    @itchat.msg_register(TEXT)   #这里的TEXT表示如果有人发送文本消息,那么就会调用下面的方法
    def simple_reply(msg):
        #这个是向发送者发送消息
        itchat.send_msg('已经收到了文本消息,消息内容为%s'%msg['Text'],toUserName=msg['FromUserName'])
        return "T reveived: %s" % msg["Text"]     #返回的给对方的消息,msg["Text"]表示消息的内容

带对象参数注册, 对应消息对象将调用该方法,其中isFriendChat表示好友之间,isGroupChat表示群聊,isMapChat表示公众号。

import itchat
from itchat.content import *

@itchat.msg_register(TEXT, isFriendChat=True, isGroupChat=True,isMpChat=True)
def text_reply(msg):
    msg.user.send("%s : %s" % (mst.type, msg.text))

消息类型
向注册方法传入的 msg 包含微信返回的字典的所有内容.itchat 增加 Text, Type(也就是参数) 键值, 方便操作。

itcaht.content 中包含所有的消息类型参数例如图片、语音、名片、分享等,也对应不同的注册参数:

  • 图片对应

    itchat.content.PICTURE

  • 语音对应

    itchat.content.RECORDING

  • 名片对应

    itchat.content.CARD

其余的可以参考官方API
https://pypi.org/project/itchat/

import itchat

@itchat.msg_register(itchat.content.TEXT)
def print_content(msg):
    print(msg['Text'])

itchat.auto_login()
itchat.run()

通过运行上面的代码,实现基础的文本信息的接收。

实现与图灵机器人的交互

实现机器人有很多种方法,最简单的就是使用他人提供的接口。这里以图灵机器人为例,演示这一功能。图灵机器人简单而言就是以一定的规则给图灵的服务器发送数据包(包含你对他说的话)图灵的服务器会以一定的规则给你返回数据包(包含他回复你的话)

你需要一个Tuling Key来告诉图灵服务器你有权和他对话。

18edce3ce905a4c1dbb965e6b35c3834d
2eb720a8970964f3f855d863d24406576
31107d5601866433dba9599fac1bc0083
471f28bf79c820df10d39b4074345ef8c

接下来需配置图灵机器人

{
    'key'    : 'TULING_KEY',
    'info'   : 'YOUR_MSG',
    'userid' : 'USERID',
}

下面使用图灵机器人向好友发送了hello

#coding=utf8
import requests

apiUrl = 'http://www.tuling123.com/openapi/api'
data = {
    'key'    : '8edce3ce905a4c1dbb965e6b35c3834d', # 如果这个Tuling Key不能用,那就换一个
    'info'   : 'hello', # 这是我们发出去的消息
    'userid' : 'wechat-robot', # 这里你想改什么都可以
}
# 我们通过如下命令发送一个post请求
r = requests.post(apiUrl, data=data).json()
# 让我们打印一下返回的值,看一下我们拿到了什么
print(r)

完整代码:

import requests
import itchat
KEY = '8edce3ce905a4c1dbb965e6b35c3834d'

def get_response(msg):
    # 这里实现与图灵机器人的交互
    # 构造了要发送给服务器的数据
    apiUrl = 'http://www.tuling123.com/openapi/api'
    data = {
        'key' : KEY,
      'info' : msg,
      'userid' : 'wechat-robot',
    }
    try:
        r = requests.post(apiUrl, data=data).json()
        # 字典的get方法在字典没有'text'值的时候会返回None而不会抛出异常
        return r.get('text')
    # 为了防止服务器没有正常响应导致程序异常退出,这里用try-except捕获了异常
    # 如果服务器没能正常交互(返回非json或无法连接),那么就会进入下面的return
    except:
        # 将会返回一个None
        return

# 这里实现微信消息的获取
@itchat.msg_register(itchat.content.TEXT)
def tuling_reply(msg):
    # 为了保证在图灵Key出现问题的时候仍旧可以回复,这里设置一个默认回复
    defaultReply = 'I received: ' + msg['Text']
    # 如果图灵Key出现问题,那么reply将会是None
    reply = get_response(msg['Text'])
    # a or b的意思是,如果a有内容,那么返回a,否则返回b
    # 有内容一般就是指非空或者非None,你可以用`if a: print('True')`来测试
    return reply or defaultReply

# 为了让实验过程更加方便(修改程序不用多次扫码),我们使用热启动
itchat.auto_login(hotReload=True)
itchat.run()

itchat不仅仅能够实现自动回复,还有其他的相关功能比如群发助手,好友单删检测,获取微信好友信息等功能,后续会慢慢去体验各种玩法。

小提示:不要频繁操作,否则会禁止登录web微信。


派森学python
175 声望13 粉丝