快捷導(dǎo)航

教你用Python創(chuàng)建微信聊天機(jī)器人

更新時(shí)間：2021年07月07日 13:59:07 作者：強(qiáng)哥

這篇文章主要手把手教你用Python創(chuàng)建微信聊天機(jī)器人，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下

最近研究微信API，發(fā)現(xiàn)個(gè)非常好用的python庫(kù)：wxpy。wxpy基于itchat，使用了 Web 微信的通訊協(xié)議，實(shí)現(xiàn)了微信登錄、收發(fā)消息、搜索好友、數(shù)據(jù)統(tǒng)計(jì)等功能。

這里我們就來(lái)介紹一下這個(gè)庫(kù)，并在最后實(shí)現(xiàn)一個(gè)聊天機(jī)器人。

有沒(méi)有很興奮？有沒(méi)有很期待？

好了，接下來(lái)，開始我們的正題。

準(zhǔn)備工作

安裝非常簡(jiǎn)單，從官方源下載安裝

pip install -U wxpy

或者從豆瓣源安裝

pip install -U wxpy -i "https://pypi.doubanio.com/simple/"

模塊初探

安裝完成后，我們來(lái)試下幾個(gè)基本的功能

1. 掃碼登錄微信

from wxpy import *
bot = Bot()

運(yùn)行上面的程序，會(huì)彈出二維碼，用手機(jī)微信掃一掃即可實(shí)現(xiàn)登錄。
但上面的程序有一個(gè)缺點(diǎn)，每次運(yùn)行都要掃二維碼。不過(guò)wxpy非常貼心地提供了緩存的選項(xiàng)，如下

bot = Bot(cache_path=True)

這樣把登錄信息保存下來(lái)，就不用每次都掃二維碼了。

2. 發(fā)送消息

bot.file_helper.send("hello")

這里的file_helper就是微信的文件傳輸助手，我們給文件傳輸助手發(fā)送一條消息，可以在手機(jī)端收到如下的消息

3. 接受消息

我們實(shí)現(xiàn)一個(gè)收到消息自動(dòng)回復(fù)的功能。

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

# 進(jìn)入Python命令行，讓程序保持運(yùn)行
embed()

強(qiáng)哥打開自己的公眾號(hào)管理平臺(tái)，在后端向自己發(fā)送消息，就可以收到如下的消息回復(fù)

4. 搜索好友及微信群

我們來(lái)實(shí)現(xiàn)一個(gè)搜索公司群、定位老板并轉(zhuǎn)發(fā)老板消息的功能

from wxpy import *

bot = Bot(cache_path=True)

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

# 定位老板

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

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

# 堵塞線程
embed()

對(duì)于老板喜歡在群里面喊話的同學(xué)來(lái)說(shuō)，這是一個(gè)福音，以后再也不用擔(dān)心錯(cuò)失老板的重要信息了~~

數(shù)據(jù)統(tǒng)計(jì)

wxpy的好友統(tǒng)計(jì)功能非常好用，可以很方便地統(tǒng)計(jì)好友的地理位置分布和性別分布。

下面的代碼中，強(qiáng)哥統(tǒng)計(jì)了下自己的好友的分布情況，并打印出人數(shù)最多的10個(gè)地區(qū)。

from wxpy import *
bot = Bot(cache_path=True)
friends_stat = bot.friends().stats()

friend_loc = [] # 每一個(gè)元素是一個(gè)二元列表，分別存儲(chǔ)地區(qū)和人數(shù)信息
for province, count in friends_stat["province"].iteritems():
 if province != "":
 friend_loc.append([province, count])

# 對(duì)人數(shù)倒序排序
friend_loc.sort(key=lambda x: x[1], reverse=True)

# 打印人數(shù)最多的10個(gè)地區(qū)
for item in friend_loc[:10]:
 print item[0], item[1]

統(tǒng)計(jì)出來(lái)的地區(qū)分布數(shù)據(jù)畫成圖表如下

強(qiáng)哥身在上海，大多數(shù)的好友也來(lái)自上海，上面的圖表也如實(shí)地反應(yīng)了這一情況。

統(tǒng)計(jì)性別分布的代碼如下

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

把性別分布的數(shù)據(jù)生成圖表如下

可以看到好友中男性占了大多數(shù)。男性朋友多多，老婆放心多多，嗯~~

聊天機(jī)器人

有了上面功能簡(jiǎn)介部分的基礎(chǔ)，我們來(lái)實(shí)現(xiàn)一個(gè)聊天機(jī)器人。

聊天機(jī)器人基于圖靈機(jī)器人。圖靈機(jī)器人可以在圖靈機(jī)器人－中文語(yǔ)境下智能度最高的機(jī)器人大腦注冊(cè)賬號(hào)，并創(chuàng)建機(jī)器人。

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

# 調(diào)用圖靈機(jī)器人API，發(fā)送消息并獲得機(jī)器人的回復(fù)
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()

運(yùn)行上面的程序，發(fā)送消息給自己，可以看到如下的對(duì)話