200行代码，一行行教你自制微信机器人

第一时间关注程序猿（媛）身边的故事

作者

上海小胖

作者是四大咨询的TechLead，mongoDB Professional 获得者。

「Python专栏」专注Python领域的各种技术：爬虫、DevOps、人工智能、Web开发等。

已获原作者授权，如需转载，请联系原作者。

之前的一篇文章「自制微信机器人：群发消息、自动接收好友」一经推出，收到了很多小伙伴的私聊，包括建议、咨询、学习的。

今天干脆把这套代码的思路讲清楚了，也好之后和大家形成更有效深入的讨论。

1.

当初决定自己写这么个机器人有几个原因：

1） 用一个windows客户端工具运营公众号，真的很局限。虽然工具的功能很强大，能自动添加好友，自动拉好友入群，关键字回复等等，但是有一个绕不开的点，它是一款客户端工具，一款exe软件。

2） 我是Mac，为了用这个工具，就要开着虚拟机去操作。

3） 为了能一直自动添加好友，邀请入群，自动回复等一系列操作，电脑就不能合上。

4） 在外面突然想到一个点，想操作群发了，GG，无能为力。

5） 其他……

2.

基于以上的原因，就想着自己来一套算了。毕竟可以定制化的话，之后想要什么就很方便了，而且在服务器端挂个python服务要比开个windows 就为了挂一个exe要很多。

那么首先需要确定需求，wxRobot我是准备长期维护、迭代的，所以显然不可能像网上那些个脚本一样，一个文件打通关。

另外功能自定义，就需要有版本引入，先做什么，痛点是什么都需要明确。我说下自己的选择：

1） 痛点是不能自主化的管理公众号、微信号。

2） 最急迫解决的是之前exe工具用到的功能，毕竟这也是我用这个工具的原因。那么有哪些功能呢？

群发消息

自动添加好友

邀请好友入群

关键字回复

3.

既然是个项目，那么该有的组件一个不能少，看一下我的目录结构，这也是我自己总结出的common structure，大家可以参考一下，如果有好的建议欢迎大佬不吝留言。

app：项目业务模块。如果有多个模块就添加子目录，例如：一个网站下的博客模块、投票模块等。

core：核心组件。例如：数据库组件、类-文件组件等。

doc：文档。存放所有的文档，一般我会有固定的几个：CHANGELOG.md、BUGLIST.md、TODOLIST.md。

etc：配置文件。可以细分基本配置、业务配置等。

static：静态文件。

test：单元测试。

tmp：不需要进入版本控制的东西。

utility：辅助组件。和core相辅。

4.

我把业务分为两块，filehelper算一个，好友相关的算一个。

好友相关的好理解，诸如添加好友、自动回复、邀请入群等。filehelper是什么呢？说白了，我们除了简单的自动回复、添加外，一定还希望做的更多吧？比如交互式指令。那这个filehelper就承担了指令收发的角色。

所有的业务模块都基于一个BaseHandle，这样底层的一些单元我就可以统一管控了：

再来看看FileHandle这个类，这也是当前版本最丰富的模块。这里面有两端逻辑：1.自动更新群组信息。 2.注册群发相关命令。

自动更新群组信息的目的是因为itchat模块会将所有联系人以及群组信息存储在本地的一个pkl文件中（pickle缩写？），如果想提升群发消息前获取群组列表的速度，那么就应该把数据放在内存里（反正也没多少数据），以下我把主要逻辑都罗列出来了，具体的代码太长了，暂时就不放出来了：

注册群发相关命令的思路就是做一个命令注册器，因为群发消息、文章、图片等行为类似，针对不同的用户群组发送不同的消息体。

所以我就把注册器的成员分成了：类型（文字、图片），对象（时间限制群组、无限制群组），行为（群发、单发）。

被装饰器注册的函数就成为了某个具有单独意义的指令了。

classFileHelper(BaseHandle):

...

defupdate_cmd(self, cmd):

# 更新命令，用于动态注册函数

def_register_mass(func):

@functools.wraps(func)

defdecorator(self, msg, *args, **kwargs):

_action, _reply, _obj = func.__name__.split('_')

ifself._meta['action'][_action]:

_to_user = self._meta['obj'][_obj]

for_groupin_to_user:

instance.send_msg(msg, _group['UserName'])

time.sleep(random.randrange(,20))

self._meta['action'][_action] =False

self._current_cmd =None

instance.send_msg('群发消息发送完毕', self._meta['extra']['UserName'])

returndecorator

@_register_mass

defmass_text_ul(self, msg=None):

pass

@_register_mass

defmass_text_l(self, msg):

pass

@_register_mass

defmass_text_test(self, msg):

pass

@_register_mass

defmass_article_ul(self, msg):

pass

@_register_mass

defmass_article_l(self, msg):

pass

对比着效果图来看看：

5.

接下来就是添加好友部分了，目前只支持自动接受好友，根据打招呼自动设置备注，关键字回复。

classFriend(BaseHandle):

_usage ='''

'''

def__init__(self):

super().__init__()

self._meta = {

...

}

defis_biz(self, msg):

# 判断是不是商务合作

看下效果图：

6.

讲完核心代码后，再来讲下中间经历的几个看不到的版本吧。

最一开始就是实现功能咯，没想很多，但是发现代码重复太多了，逻辑都差不多，一堆代码太丑了。优化后的代码就是第一版中的群发注册器函数。

接着原本的BaseHandle基类太重了，想的很好，把itchat方法都重写在基类里，这样就不用在其他地方调用itchat实例了，但是结果就是所有的子类都可以做同样的动作，就变成了filehelper.send_msg(), friend.send_image()了，这样对于同一个方法就会产生歧义了。因此就把基类里所有重写itchat方法的函数都去了，就保留了业务代码，并分别移到对应的类里去，而原本itchat的方法还是用itchat实例去操作。

接着关于itchat实例、FileHelper实例、Friend实例等的共享问题，容易造成重叠，重复使用、互相引用问题。解决办法目前就是把itchat实例单独在配置文件里初始化了，这也同时解决了上一个问题，其他业务类的实例采用单例模式，在类外面暴露一个统一的实例。

7.

好了，这回是真花了功夫把这套代码讲完了，虽然还是相对简陋了，但迫于时间关系，先发出来了。之后会继续优化、健硕它。

今天也和一位大佬讨论了下这个项目，有很多值得思考的地方。

如果你对这份代码也感兴趣的话，欢迎底部留言~

- The End -

「若你有原创文章想与大家分享，欢迎投稿。」

加编辑微信ID，备注#投稿#：

程序 丨 druidlost

小七 丨 duoshangshuang