最近研究微信API，發(fā)現(xiàn)個非常好用的python庫：wxpy。wxpy基于itchat，使用了 Web 微信的通訊協(xié)議，實現(xiàn)了微信登錄、收發(fā)消息、搜索好友、數(shù)據(jù)統(tǒng)計等功能。這里我們就來介紹一下這個庫，并在最后實現(xiàn)一個聊天機器人。有沒有很興奮？有沒有很期待？好了，接下來，開始我們的正題。
準(zhǔn)備工作
安裝非常簡單，從官方源下載安裝
pip?install?-U wxpy
或者從豆瓣源安裝
pip?install?-U wxpy -i?"https://pypi.doubanio.com/simple/"
模塊初探
安裝完成后，我們來試下幾個基本的功能
1. 掃碼登錄微信
from?wxpy?import?*
bot = Bot()
運行上面的程序，會彈出二維碼，用手機微信掃一掃即可實現(xiàn)登錄。
但上面的程序有一個缺點，每次運行都要掃二維碼。不過wxpy非常貼心地提供了緩存的選項，如下
bot?= Bot(cache_path=True)
這樣把登錄信息保存下來，就不用每次都掃二維碼了。
2. 發(fā)送消息
bot.file_helper.send("hello")
這里的file_helper就是微信的文件傳輸助手，我們給文件傳輸助手發(fā)送一條消息，可以在手機端收到如下的消息
3. 接受消息
我們實現(xiàn)一個收到消息自動回復(fù)的功能。
@bot.register()
def?print_message(msg):??
print(msg.text)?
return?msg.text
# 進入Python命令行，讓程序保持運行
embed()
強哥打開自己的公眾號管理平臺，在后端向自己發(fā)送消息，就可以收到如下的消息回復(fù)
4. 搜索好友及微信群
我們來實現(xiàn)一個搜索公司群、定位老板并轉(zhuǎn)發(fā)老板消息的功能
from?wxpy?import?*
bot = Bot(cache_path=True)
# 定位公司群
company_group = bot.groups().search('公司微信群')[0]
# 定位老板
boss = company_group.search('老板大名')[0]
# 將老板的消息轉(zhuǎn)發(fā)到文件傳輸助手
@bot.register(company_group)
def?forward_boss_message(msg):??
if?msg.member == boss:?
msg.forward(bot.file_helper, prefix='老板發(fā)言')
# 堵塞線程
embed()
對于老板喜歡在群里面喊話的同學(xué)來說，這是一個福音，以后再也不用擔(dān)心錯失老板的重要信息了~~
數(shù)據(jù)統(tǒng)計
wxpy的好友統(tǒng)計功能非常好用，可以很方便地統(tǒng)計好友的地理位置分布和性別分布。
下面的代碼中，強哥統(tǒng)計了下自己的好友的分布情況，并打印出人數(shù)最多的10個地區(qū)。
from?wxpy?import?*
bot = Bot(cache_path=True)
friends_stat = bot.friends().stats()
friend_loc = []?# 每一個元素是一個二元列表，分別存儲地區(qū)和人數(shù)信息
for?province, count?in?friends_stat["province"].iteritems():
if?province !=?"":
friend_loc.append([province, count])
# 對人數(shù)倒序排序
friend_loc.sort(key=lambda?x: x[1], reverse=True)
# 打印人數(shù)最多的10個地區(qū)
for?item?in?friend_loc[:10]:
print?item[0], item[1]
統(tǒng)計出來的地區(qū)分布數(shù)據(jù)畫成圖表如下
好友來自上海，上面的圖表也如實地反應(yīng)了這一情況。統(tǒng)計性別分布的代碼如下
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
把性別分布的數(shù)據(jù)生成圖表如下
可以看到好友中男性占了大多數(shù)。男性朋友多多，老婆放心多多，嗯~~
聊天機器人
有了上面功能簡介部分的基礎(chǔ)，我們來實現(xiàn)一個聊天機器人。
聊天機器人基于圖靈機器人。圖靈機器人可以在圖靈機器人－中文語境下智能度最高的機器人大腦注冊賬號，并創(chuàng)建機器人。
# -*- coding: utf-8 -*-
import?json
import?requests
from?wxpy?import?*
# 調(diào)用圖靈機器人API，發(fā)送消息并獲得機器人的回復(fù)
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()
運行上面的程序，發(fā)送消息給自己，可以看到如下的對話
機器人太逗了，上來就要紅包，還把我當(dāng)男朋友了~
附錄
wxpy文檔：wxpy: 用 Python 玩微信
github地址：youfou/wxpy
