最近研究微信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基础教程推荐: