200行程式碼,一行行教你自制微信機器人
寫機器人的原因
筆者當初決定自己寫這麼個機器人,有幾個原因:
1) 用一個Windows客戶端工具運營公眾號,真的很侷限。雖然工具的功能很強大,能自動新增好友,自動拉好友入群,關鍵字回覆等等,但是有一個繞不開的點,它是一款客戶端工具,一款EXE軟體。
2) 我是Mac,為了用這個工具,就要開著虛擬機器去操作。
3) 為了能一直自動新增好友,邀請入群,自動回覆等一系列操作,電腦就不能合上。
4) 在外面突然想到一個點,想操作群發了,GG,無能為力。
5) 其他……
確定需求
基於以上的原因,就想著自己來一套算了。畢竟可以定製化的話,之後想要什麼就很方便了,而且在伺服器端掛個Python服務要比開個Windows就為了掛一個EXE要很多。
那麼首先需要確定需求,wxRobot我是準備長期維護、迭代的,所以顯然不可能像網上那些個指令碼一樣,一個檔案打通關。
另外功能自定義,就需要有版本引入,先做什麼,痛點是什麼都需要明確。我說下自己的選擇:
1) 痛點是不能自主化的管理公眾號、微信號。
2) 最急迫解決的是之前EXE工具用到的功能,畢竟這也是我用這個工具的原因。那麼有哪些功能呢?
群發訊息 自動新增好友 邀請好友入群 關鍵字回覆
元件構成
既然是個專案,那麼該有的元件一個不能少,看一下我的目錄結構,這也是我自己總結出的Common structure,大家可以參考一下。
App:專案業務模組。如果有多個模組就新增子目錄,例如:一個網站下的部落格模組、投票模組等。
Core:核心元件。例如:資料庫元件、類-檔案元件等。
Doc:文件。存放所有的文件,一般我會有固定的幾個:CHANGELOG.md、BUGLIST.md、TODOLIST.md。
ETC:配置檔案。可以細分基本配置、業務配置等。
static:靜態檔案。
test:單元測試。
TMP:不需要進入版本控制的東西。
Utility:輔助元件。和core相輔。
業務分塊
我把業務分為兩塊,filehelper算一個,好友相關的算一個。
好友相關的好理解,諸如新增好友、自動回覆、邀請入群等。filehelper是什麼呢?說白了,我們除了簡單的自動回覆、新增外,一定還希望做的更多吧?比如互動式指令。那這個filehelper就承擔了指令收發的角色。
所有的業務模組都基於一個BaseHandle,這樣底層的一些單元我就可以統一管控了:
classBaseHandle: 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 defusage(self): returnself._usage @property defmeta(self): returnself._meta
再來看看FileHandle這個類,這也是當前版本最豐富的模組。這裡面有兩端邏輯:1.自動更新群組資訊。 2.註冊群發相關命令。
自動更新群組資訊的目的是因為itchat模組會將所有聯絡人以及群組資訊儲存在本地的一個PKI檔案中(一個用公鑰概念和技術實施和提供安全服務的具有普適性的安全基礎設施)。
如果想提升群發訊息前獲取群組列表的速度,那麼就應該把資料放在記憶體裡(反正也沒多少資料),以下我把主要邏輯都羅列出來了,具體的程式碼太長了,暫時就不放出來了:
classFileHelper(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() defauto_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): # 篩選出不受限制的群組 whileTrue: time.sleep( 30) # 更新群組資訊
註冊群發相關命令的思路就是做一個命令註冊器,因為群發訊息、文章、圖片等行為類似,針對不同的使用者群組傳送不同的訊息體。
所以我就把註冊器的成員分成了:型別(文字、圖片)、物件(時間限制群組、無限制群組)、行為(群發、單發)。
被裝飾器註冊的函式就成為了某個具有單獨意義的指令了。
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_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']) 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
對比著效果圖來看看:
新增好友
接下來就是新增好友部分了,目前只支援自動接受好友,根據打招呼自動設定備註,關鍵字回覆。
classFriend(BaseHandle): _usage = ''' ''' def__init__(self): super().__init__() self._meta = { ... } defis_biz(self, msg): # 判斷是不是商務合作
看下效果圖:
不同版本
講完核心程式碼後,再來講下中間經歷的幾個看不到的版本吧。
最一開始就是實現功能咯,沒想很多,但是發現程式碼重複太多了,邏輯都差不多,一堆程式碼太醜了。優化後的程式碼就是第一版中的群發註冊器函式。
接著原本的BaseHandle基類太重了,想的很好,把itchat方法都重寫在基類裡,這樣就不用在其他地方呼叫itchat例項了,但是結果就是所有的子類都可以做同樣的動作,就變成了filehelper.send_msg(), friend.send_image()了,這樣對於同一個方法就會產生歧義了。
因此就把基類裡所有重寫itchat方法的函式都去了,就保留了業務程式碼,並分別移到對應的類裡去,而原本itchat的方法還是用itchat例項去操作。
接著關於itchat例項、FileHelper例項、Friend例項等的共享問題,容易造成重疊,重複使用、互相引用問題。
解決辦法目前就是把itchat例項單獨在配置檔案裡初始化了,這也同時解決了上一個問題,其他業務類的例項採用單例模式,在類外面暴露一個統一的例項。
作者簡介:上海小胖,四大諮詢的Tech Lead,MongoDB Professional獲得者。「Python專欄」專注Python領域的各種技術:爬蟲、DevOps、人工智慧、Web開發等。還有「大航海計劃」,各種內推活動。