最近研究微信API，发现个非常好用的python库：wxpy。wxpy基于itchat，使用了 Web 微信的通讯协议，实现了微信登录、收发消息、搜索好友、数据统计等功能。这里为大家介绍一下这个库，并在最后实现一个聊天机器人。
微信机器人源码下载
准备工作
安装非常简单，从官方源下载安装
模块初探
安装完成后，我们来试下几个基本的功能
1. 扫码登录微信
| 1 2 |
from wxpy import *
bot = Bot()
|
运行上面的程序，会弹出二维码，用手机微信扫一扫即可实现登录。
但上面的程序有一个缺点，每次运行都要扫二维码。不过wxpy非常贴心地提供了缓存的选项，如下：
| 1 |
bot = Bot(cache_path=True)
|
这样把登录信息保存下来，就不用每次都扫二维码了。
2. 发送消息
| 1 |
bot.file_helper.send("hello")
|
这里的file_helper就是微信的文件传输助手，我们给文件传输助手发送一条消息，可以在手机端收到消息。
3. 接受消息
我们实现一个收到消息自动回复的功能。
| 1 2 3 4 5 6 7 |
@bot.register()
def print_message(msg):
print(msg.text)
return msg.text
# 进入Python命令行，让程序保持运行
embed()
|
打开自己的公众号管理平台，在后端向自己发送消息，就可以收到消息回复。
4. 搜索好友及微信群
我们来实现一个搜索公司群、定位老板并转发老板消息的功能：
| 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from wxpy import *
bot = Bot(cache_path=True)
# 定位公司群
company_group = bot.groups().search('公司微信群')[0]
# 定位老板
boss = company_group.search('老板大名')[0]
# 将老板的消息转发到文件传输助手
@bot.register(company_group)
def forward_boss_message(msg):
if msg.member == boss:
msg.forward(bot.file_helper, prefix='老板发言')
# 堵塞线程
embed()
|
对于老板喜欢在群里面喊话的同学来说，这是一个福音，以后再也不用担心错失老板的重要信息了~~
数据统计
wxpy的好友统计功能非常好用，可以很方便地统计好友的地理位置分布和性别分布。下面的代码中，统计了下自己的好友的分布情况，并打印出人数最多的10个地区。
| 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
from wxpy import *
bot = Bot(cache_path=True)
friends_stat = bot.friends().stats()
friend_loc = [] # 每一个元素是一个二元列表，分别存储地区和人数信息
for province, count in friends_stat["province"].iteritems():
if province != "":
friend_loc.append([province, count])
# 对人数倒序排序
friend_loc.sort(key=lambda x: x[1], reverse=True)
# 打印人数最多的10个地区
for item in friend_loc[:10]:
print item[0], item[1]
统计性别分布的代码如下：
for sex, count in friends_stat["sex"].iteritems():
# 1代表MALE, 2代表FEMALE
if sex == 1:
print "MALE %d" % count
elif sex == 2:
print "FEMALE %d" % count
|
聊天机器人
有了上面功能简介部分的基础，我们来实现一个聊天机器人。聊天机器人基于图灵机器人。图灵机器人可以在图灵机器人－中文语境下智能度最高的机器人大脑注册账号，并创建机器人。需记下api key，调用图灵API的时候会用到。
| 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
# -*- coding: utf-8 -*-
import json
import requests
from wxpy import *
# 调用图灵机器人API，发送消息并获得机器人的回复
def auto_reply(text):
url = "http://www.tuling123.com/openapi/api"
api_key = "你的api key"
payload = {
"key": api_key,
"info": text,
"userid": "123456"
}
r = requests.post(url, data=json.dumps(payload))
result = json.loads(r.content)
return "[tuling] " + result["text"]
bot = Bot(console_qr=True, cache_path=True)
@bot.register(mp)
def forward_message(msg):
return auto_reply(msg.text)
embed()
|
运行上面的程序，发送消息给自己。
附录
【专业Python IDE推荐】——PyCharm （正版低至￥1068）：
Python IDE，其带有一整套可以帮助用户在使用Python语言开发时提高其效率的工具。此外，该IDE提供了一些高级功能，以用于Django框架下的专业Web开发。
PyCharm基础教程推荐：