最近研究微信API，發(fā)現(xiàn)個非常好用的python庫：wxpy。wxpy基于itchat，使用了 Web 微信的通訊協(xié)議，實現(xiàn)了微信登錄、收發(fā)消息、搜索好友、數(shù)據統(tǒng)計等功能。這里為大家介紹一下這個庫，并在最后實現(xiàn)一個聊天機器人。

微信機器人源碼下載

準備工作

安裝非常簡單，從官方源下載安裝

pip install -U wxpy

模塊初探

安裝完成后，我們來試下幾個基本的功能

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)一個收到消息自動回復的功能。

@bot.register()
def print_message(msg):
print(msg.text)
return msg.text

# 進入Python命令行，讓程序保持運行
embed()

打開自己的公眾號管理平臺，在后端向自己發(fā)送消息，就可以收到消息回復。

4. 搜索好友及微信群

我們來實現(xiàn)一個搜索公司群、定位老板并轉發(fā)老板消息的功能：

from wxpy import *

bot = Bot(cache_path=True)

# 定位公司群
company_group = bot.groups().search('公司微信群')[0]

# 定位老板
boss = company_group.search('老板大名')[0]

# 將老板的消息轉發(fā)到文件傳輸助手
@bot.register(company_group)
def forward_boss_message(msg):
if msg.member == boss:
msg.forward(bot.file_helper, prefix='老板發(fā)言')

# 堵塞線程
embed()

對于老板喜歡在群里面喊話的同學來說，這是一個福音，以后再也不用擔心錯失老板的重要信息了~~

數(shù)據統(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)計性別分布的代碼如下：

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

聊天機器人

有了上面功能簡介部分的基礎，我們來實現(xiàn)一個聊天機器人。聊天機器人基于圖靈機器人。圖靈機器人可以在注冊賬號，并創(chuàng)建機器人。需記下api key，調用圖靈API的時候會用到。

# -*- coding: utf-8 -*-
import json
import requests
from wxpy import *

# 調用圖靈機器人API，發(fā)送消息并獲得機器人的回復
def auto_reply(text):
url = "//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ā)送消息給自己。

附錄

• wxpy文檔：
• github項目源碼下載

【專業(yè)Python IDE推薦】——PyCharm （正版低至￥1068）：

Python IDE，其帶有一整套可以幫助用戶在使用Python語言開發(fā)時提高其效率的工具。此外，該IDE提供了一些高級功能，以用于Django框架下的專業(yè)Web開發(fā)。

PyCharm基礎教程推薦：

• PyCharm入門第一步（一）——創(chuàng)建并運行第一個Python項目
• PyCharm入門第一步（二）——調試第一個Python應用程序
• PyCharm入門第一步（三）——測試第一個Python應用程序
• PyCharm入門第一步（四）——創(chuàng)建和運行第一個Django項目

MyEclipse正版最低只要￥212！手慢無

續(xù)費優(yōu)惠：即日起授權續(xù)費的用戶，享受比原價更優(yōu)惠的價格！