新智元 关注
手机版

手把手教你用Python创建微信聊天机器人

  1. 云栖社区>
  2. 新智元>
  3. 博客>
  4. 正文

手把手教你用Python创建微信聊天机器人

技术小能手 2018-03-14 13:37:07 浏览2548 评论0

摘要: 微信联系人太多回复不过来?大神教你用Python创建微信聊天机器人,手把手、附代码。

最近研究微信 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 就是微信的文件传输助手,我们给文件传输助手发送一条消息,可以在手机端收到如下的消息

8393253f424e7f396a92a2f8d770c02db7f16ab7

3. 接受消息

我们实现一个收到消息自动回复的功能。

@bot.register()
def print_message(msg):
print(msg.text)
return msg.text
# 进入Python命令行,让程序保持运行
embed()

强哥打开自己的公众号管理平台,在后端向自己发送消息,就可以收到如下的消息回复

e44703cb3249970b61bb84674a56980456505ec6

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]

统计出来的地区分布数据画成图表如下

f10c02216ac190cd4ddeb3c6efb67f2283f60a34

强哥身在上海,大多数的好友也来自上海,上面的图表也如实地反应了这一情况。

统计性别分布的代码如下

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

把性别分布的数据生成图表如下

7aa27d9f006b3151aee28df36944e016545da0b4

可以看到好友中男性占了大多数。男性朋友多多,老婆放心多多,嗯~~

聊天机器人

有了上面功能简介部分的基础,我们来实现一个聊天机器人。

聊天机器人基于图灵机器人。图灵机器人可以在图灵机器人-中文语境下智能度最高的机器人大脑注册账号,并创建机器人。

# -*- 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()

运行上面的程序,发送消息给自己,可以看到如下的对话

1196f5299e21821a703a39ff707208d935a138cc




原文发布时间为:2018-03-14

本文作者:强哥

本文来自云栖社区合作伙伴新智元,了解相关信息可以关注“AI_era”微信公众号

原文链接:手把手教你用Python创建微信聊天机器人

用云栖社区APP,舒服~

【云栖快讯】云栖社区技术交流群汇总,阿里巴巴技术专家及云栖社区专家等你加入互动,老铁,了解一下?  详情请点击

网友评论

技术小能手
文章3780篇 | 关注957
关注
阿里云机器学习是基于阿里云分布式计算引擎的一款机器学习算法平台。用户通过拖拉拽的方式可视化的... 查看详情
是基于语音识别、语音合成、自然语言理解等技术,为企业在多种实际应用场景下,赋予产品“能听、会... 查看详情
阿里云推出的一款移动App数据统计分析产品,为开发者提供一站式数据化运营服务 查看详情
为您提供简单高效、处理能力可弹性伸缩的计算服务,帮助您快速构建更稳定、安全的应用,提升运维效... 查看详情
阿里中间件云大使

阿里中间件云大使