最近研究微信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進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。