最近研究微信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]