最近研究微信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文檔:http://wxpy.readthedocs.io/zh/latest/
github地址:https://github.com/youfou/wxpy/blob/master/docs/index.rst
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持億速云。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。