Python也能聊微信
最近研究微信API，發現個非常好用的python庫：wxpy。wxpy基於itchat，使用了 Web 微信的通訊協議，實現了微信登入、收發訊息、搜尋好友、資料統計等功能。這裡我們就來介紹一下這個庫，並在最後實現一個聊天機器人。有沒有很興奮？有沒有很期待？好了，接下來，開始我們的正題。
準備工作
安裝非常簡單，從官方源下載安裝
pip install -U wxpy
或者從豆瓣源安裝
pip install -U wxpy -i "https://pypi.doubanio.com/simple/"
模組初探
安裝完成後，我們來試下幾個基本的功能
1. 掃碼登入微信
from wxpy import *
bot = Bot()
執行上面的程式，會彈出二維碼，用手機微信掃一掃即可實現登入。
但上面的程式有一個缺點，每次執行都要掃二維碼。不過wxpy非常貼心地提供了快取的選項，如下
bot = Bot(cache_path=True)
這樣把登入資訊儲存下來，就不用每次都掃二維碼了。
2. 傳送訊息
bot.file_helper.send("hello")
這裡的file_helper就是微信的檔案傳輸助手，我們給檔案傳輸助手傳送一條訊息，可以在手機端收到如下的訊息
3. 接受訊息
我們實現一個收到訊息自動回覆的功能。
@bot.register()
def print_message(msg):
print(msg.text)
return msg.text
# 進入Python命令列，讓程式保持執行
embed()
強哥開啟自己的公眾號管理平臺，在後端向自己傳送訊息，就可以收到如下的訊息回覆
4. 搜尋好友及微信群
我們來實現一個搜尋公司群、定位老闆並轉發老闆訊息的功能
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個地區。
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
把性別分佈的資料生成圖表如下
可以看到好友中男性佔了大多數。男性朋友多多，老婆放心多多，嗯~~
聊天機器人
有了上面功能簡介部分的基礎，我們來實現一個聊天機器人。
聊天機器人基於圖靈機器人。圖靈機器人可以在圖靈機器人－中文語境下智慧度最高的機器人大腦註冊賬號，並建立機器人。
# -*- 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()
執行上面的程式，傳送訊息給自己，可以看到如下的對話
機器人太逗了，上來就要紅包，還把我當男朋友了~
附錄
wxpy文件：wxpy: 用 Python 玩微信
github地址：youfou/wxpy