参加 2018 AI开发者大会，请点击 ↑↑↑
作者|上海小胖，四大咨询的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，这样底层的一些单元我就可以统一管控了：
色。
class BaseHandle:
def __init__(self):
'''
self._meta = {
'obj':{ # 消息发送对象
'ul': [], # unlimit group
'l': [], # limit group
'r': [] # restrict
},
'reply':{
'text': '',
'article': '',
}
}
'''
self._usage = ''
self._meta = {}
self.current_cmd = None
@property
def usage(self):
return self._usage
@property
def meta(self):
return self._meta
再来看看FileHandle这个类，这也是当前版本最丰富的模块。这里面有两端逻辑：1.自动更新群组信息。 2.注册群发相关命令。
自动更新群组信息的目的是因为itchat模块会将所有联系人以及群组信息存储在本地的一个pkl文件中（pickle缩写？），如果想提升群发消息前获取群组列表的速度，那么就应该把数据放在内存里（反正也没多少数据），以下我把主要逻辑都罗列出来了，具体的代码太长了，暂时就不放出来了：
class FileHelper(BaseHandle):
_usage = '''
'''
def __init__(self):
super().__init__()
self._meta = {
...
}
self._th_update = threading.Thread(target=self._update_meta, args=(), daemon=True)
self.auto_update_groups()
def auto_update_groups(self):
# 自动更新群组
self._th_update.start()
def _update_meta(self):
'''
初始化限时推送的群组
'''
def _filter_restrict_groups(group):
# 筛选出不能群发的群组
def _filter_limit_groups(group):
# 筛选出有时间限制的群组
def _filter_unlimit_group(groups, limit_groups):
# 筛选出不受限制的群组
while True:
time.sleep(30)
# 更新群组信息注册群发相关命令的思路就是做一个命令注册器，因为群发消息、文章、图片等行为类似，针对不同的用户群组发送不同的消息体。
所以我就把注册器的成员分成了：类型（文字、图片），对象（时间限制群组、无限制群组），行为（群发、单发）。
被装饰器注册的函数就成为了某个具有单独意义的指令了。
class FileHelper(BaseHandle):
...
def update_cmd(self, cmd):
# 更新命令，用于动态注册函数
def _register_mass(func):
@functools.wraps(func)
def decorator(self, msg, *args, **kwargs):
_action, _reply, _obj = func.__name__.split('_')
if self._meta['action'][_action]:
_to_user = self._meta['obj'][_obj]
for _group in _to_user:
instance.send_msg(msg, _group['UserName'])
time.sleep(random.randrange(0, 20))
self._meta['action'][_action] = False
self._current_cmd = None
instance.send_msg('群发消息发送完毕', self._meta['extra']['UserName'])
return decorator
@_register_mass
def mass_text_ul(self, msg=None):
pass
@_register_mass
def mass_text_l(self, msg):
pass
@_register_mass
def mass_text_test(self, msg):
pass
@_register_mass
def mass_article_ul(self, msg):
pass
@_register_mass
def mass_article_l(self, msg):
pass
对比着效果图来看看：
5
接下来就是添加好友部分了，目前只支持自动接受好友，根据打招呼自动设置备注，关键字回复。
class Friend(BaseHandle):
_usage = '''
'''
def __init__(self):
super().__init__()
self._meta = {
...
}
def is_biz(self, msg):
# 判断是不是商务合作
看下效果图：
6
讲完核心代码后，再来讲下中间经历的几个看不到的版本吧。
最一开始就是实现功能咯，没想很多，但是发现代码重复太多了，逻辑都差不多，一堆代码太丑了。优化后的代码就是第一版中的群发注册器函数。
接着原本的BaseHandle基类太重了，想的很好，把itchat方法都重写在基类里，这样就不用在其他地方调用itchat实例了，但是结果就是所有的子类都可以做同样的动作，就变成了filehelper.send_msg(), friend.send_image()了，这样对于同一个方法就会产生歧义了。因此就把基类里所有重写itchat方法的函数都去了，就保留了业务代码，并分别移到对应的类里去，而原本itchat的方法还是用itchat实例去操作。
接着关于itchat实例、FileHelper实例、Friend实例等的共享问题，容易造成重叠，重复使用、互相引用问题。解决办法目前就是把itchat实例单独在配置文件里初始化了，这也同时解决了上一个问题，其他业务类的实例采用单例模式，在类外面暴露一个统一的实例。
7
好了，这回是真花了功夫把这套代码讲完了，虽然还是相对简陋了，但迫于时间关系，先发出来了。之后会继续优化、健硕它。
今天也和一位大佬讨论了下这个项目，有很多值得思考的地方。
如果你对这份代码也感兴趣的话，欢迎底部留言~
--【完】--
2018 AI开发者大会
AI技术年度盛会即将开启！11月8-9日，来自Google、Amazon、微软、Facebook、LinkedIn、阿里巴巴、百度、腾讯、美团、京东、小米、字节跳动、滴滴、商汤、旷视、思必驰、第四范式、云知声等企业的技术大咖将带来工业界AI应用的最新思维。
如果你是某个AI技术领域的专业人才，或想寻求将AI技术整合至传统企业业务当中，扫码填写大会注册信息表，我们将从中挑选出20名相关性最高的幸运读者，送出单场分论坛入场券。大会嘉宾阵容和议题，请查看文末海报。
此外，如果你想与所有参会大牛充分交流沟通，点击阅读原文购票，使用优惠码：AI2018-DBY 购买两日通票，立减999元；此外大会还推出了1024定制票，主会+分会自由组合，精彩随心。
推荐阅读
唇语识别技术的开源教程，听不见声音我也能知道你说什么！
首发|机器学习未来十年：你需要把握的趋势和热点
学习这么多算法到底在解决哪些问题？深度学习之外，我们要选择谁？
点击「阅读原文」，查看大会更多详情。2018 AI开发者大会——摆脱焦虑，拥抱技术前沿。