最近 全棧數據工程師養成攻略 的微信群已經將近500人，開了二群之后為了打通不同微信群之間的消息，花了點時間做了個消息同步機器人，在任意群收到消息時同步到其他群，并且將聊天內容上傳至數據庫，以供進一步分析、統計和展示。
基本思路是，用 Python 模擬微信登陸，接收到群里消息后，對文本、圖片、分享等各類消息類型分別處理，并轉發至其他群。
前期準備
首先得有一個微信號，用于代碼模擬登陸。由于我的微信號得自己留著用，現階段注冊微信又必須要手機號，于是只好特意辦了個電信號，用來申請了一個新的微信，微信號是 honlanbot。雖說似乎可以用阿里小號來注冊微信，不過聽說存在反復回收和安全隱患問題，故不采用。
其次，需要用到一個Python庫 itchat，這個庫已經做好了用代碼調用微信的大多數功能，非常好用，官方文檔在這里，安裝的時候使用 pip 即可。
|
1
|
pip install itchat |
我的手機支持雙卡雙待，于是把兩張卡都裝手機里，再雙開微信，同時保持兩個微信號手機在線，差不多就可以開始寫代碼了。用 itchat 調用微信主要是模擬微信網頁版登陸，所以必須保持微信號手機在線，因為手機端微信一旦退出，其在網頁、PC、MAC、IPAD等相應終端認證的賬號也會隨之退出。
初步嘗試
itchat 提供了一些官方代碼，讓我們在自己的本本或電腦上新建一個 py 文件，初步嘗試一下。
運行以下代碼，會出現出現一張二維碼，掃碼登陸之后將會給“文件傳輸助手”發送一條消息。
|
1
2
3
4
5
6
|
# 加載包import itchat# 登陸itchat.auto_login()# 發送文本消息，發送目標是“文件傳輸助手”itchat.send('Hello, filehelper', toUserName='filehelper') |
以下代碼則注冊了一個消息響應事件，用來定義接收到文本消息后如何處理。在 itchat 里定義了文本、圖片、名片、位置、通知、分享、文件等多種消息類型，可以分別執行不同的處理。
|
1
2
3
4
5
6
7
8
9
10
|
import itchat# 注冊消息響應事件，消息類型為itchat.content.TEXT，即文本消息@itchat.msg_register(itchat.content.TEXT)def text_reply(msg): # 返回同樣的文本消息 return msg['Text']itchat.auto_login()# 綁定消息響應事件后，讓itchat運行起來，監聽消息itchat.run() |
再來看看如何處理其他類型消息，可以把在消息響應事件里把 msg 打印出來，是一個字典，看看有哪些感興趣的字段。
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
import itchat# import全部消息類型from itchat.content import *# 處理文本類消息# 包括文本、位置、名片、通知、分享@itchat.msg_register([TEXT, MAP, CARD, NOTE, SHARING])def text_reply(msg): # 微信里，每個用戶和群聊，都使用很長的ID來區分 # msg['FromUserName']就是發送者的ID # 將消息的類型和文本內容返回給發送者 itchat.send('%s: %s' % (msg['Type'], msg['Text']), msg['FromUserName'])# 處理多媒體類消息# 包括圖片、錄音、文件、視頻@itchat.msg_register([PICTURE, RECORDING, ATTACHMENT, VIDEO])def download_files(msg): # msg['Text']是一個文件下載函數 # 傳入文件名，將文件下載下來 msg['Text'](msg['FileName']) # 把下載好的文件再發回給發送者 return '@%s@%s' % ({'Picture': 'img', 'Video': 'vid'}.get(msg['Type'], 'fil'), msg['FileName'])# 處理好友添加請求@itchat.msg_register(FRIENDS)def add_friend(msg): # 該操作會自動將新好友的消息錄入，不需要重載通訊錄 itchat.add_friend(**msg['Text']) # 加完好友后，給好友打個招呼 itchat.send_msg('Nice to meet you!', msg['RecommendInfo']['UserName'])# 處理群聊消息@itchat.msg_register(TEXT, isGroupChat=True)def text_reply(msg): if msg['isAt']: itchat.send(u'@%s\u2005I received: %s' % (msg['ActualNickName'], msg['Content']), msg['FromUserName'])# 在auto_login()里面提供一個True，即hotReload=True# 即可保留登陸狀態# 即使程序關閉，一定時間內重新開啟也可以不用重新掃碼itchat.auto_login(True)itchat.run() |
開發消息同步機器人
經過以上示例代碼，可以總結出消息同步機器人的開發思路：
- 維護一個字典，叫做 groups 好了，用來存所有需要同步消息的群聊，key 為群聊的ID，value 為群聊的名稱；
- 接收到群聊消息時，如果消息來自于需要同步消息的群聊，就根據消息類型進行處理，同時轉發到其他需要同步的群聊。
直接上代碼好了，首先定義一個消息響應函數，文本類消息我感興趣的是 TEXT 和 SHARING 兩類，使用 isGroupChat=True 指定消息來自于群聊，這個參數默認為 False。
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
@itchat.msg_register([TEXT, SHARING], isGroupChat=True)def group_reply_text(msg): # 獲取群聊的ID，即消息來自于哪個群聊 # 這里可以把source打印出來，確定是哪個群聊后 # 把群聊的ID和名稱加入groups source = msg['FromUserName'] # 處理文本消息 if msg['Type'] == TEXT: # 消息來自于需要同步消息的群聊 if groups.has_key(source): # 轉發到其他需要同步消息的群聊 for item in groups.keys(): if not item == source: # groups[source]: 消息來自于哪個群聊 # msg['ActualNickName']: 發送者的名稱 # msg['Content']: 文本消息內容 # item: 需要被轉發的群聊ID itchat.send('%s: %s\n%s' % (groups[source], msg['ActualNickName'], msg['Content']), item) # 處理分享消息 elif msg['Type'] == SHARING: if groups.has_key(source): for item in groups.keys(): if not item == source: # msg['Text']: 分享的標題 # msg['Url']: 分享的鏈接 itchat.send('%s: %s\n%s\n%s' % (groups[source], msg['ActualNickName'], msg['Text'], msg['Url']), item) |
再來處理下圖片等多媒體類消息。
|
1
2
3
4
5
6
7
8
9
10
11
12
|
# 處理圖片和視頻類消息@itchat.msg_register([PICTURE, VIDEO], isGroupChat=True)def group_reply_media(msg): source = msg['FromUserName'] # 下載圖片或視頻 msg['Text'](msg['FileName']) if groups.has_key(source): for item in groups.keys(): if not item == source: # 將圖片或視頻發送到其他需要同步消息的群聊 itchat.send('@%s@%s' % ({'Picture': 'img', 'Video': 'vid'}.get(msg['Type'], 'fil'), msg['FileName']), item) |
以上代碼實現了對文本、分享、圖片、視頻四類消息的處理，如果對其他類型的消息也感興趣，進行相應的處理即可。在前面補上 import 的代碼，在后面補上登陸和運行的代碼，就大功告成了。
成果展示
目前兩個群之間可以進行消息同步了，一群和二群的小伙伴終于可以暢快地聊了起來（當群主不容易，經常要發很多紅包 = =）。
進一步工作
當然，我不可能一直在筆記本上運行這么個 py 代碼，所以把它部署到服務器上運行就好了，開個 screen 或者用 IPython 都可以。如果賬號偶爾下線了，再運行一下就好。
另外，我還寫了個 API，響應消息的時候會把相應的數據 POST 到我的服務器并存到數據庫，以供進一步的分析、統計和展示，這也是我身為一個群主應盡的職責～
以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持服務器之家。
原文鏈接：http://www.jianshu.com/p/7aeadca0c9bd
