最近研究微信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就是微信的檔案傳輸助手，我們給檔案傳輸助手發送一條消息，可以在手機端收到如下的消息
打開百度App，看更多美圖
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 jsonimport requestsfrom 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()
運作上面的程式，發送消息給自己，可以看到如下的對話
機器人太逗了，上來就要紅包，還把我當男朋友了~
附錄
from:https://mbd.baidu.com/newspage/data/landingshare?context=%7B%22nid%22:%22news_10269076271510995878%22%7D&pageType=1&isWiseFrom=1