基於itchat實現微信群訊息同步機器人

張巨集倫發表於2017-02-26
機器人

最近 全棧資料工程師養成攻略 的微信群已經將近500人，開了二群之後為了打通不同微信群之間的訊息，花了點時間做了個訊息同步機器人，在任意群收到訊息時同步到其他群，並且將聊天內容上傳至資料庫，以供進一步分析、統計和展示。

基本思路是，用 Python 模擬微信網頁版登陸，接收到群裡訊息後，對文字、圖片、分享等各類訊息型別分別處理，並轉發至其他群。

前期準備

首先得有一個微訊號，用於程式碼模擬登陸。由於我的微訊號得自己留著用，現階段註冊微信又必須要手機號，於是只好特意辦了個電訊號，用來申請了一個新的微信，微訊號是 honlanbot。雖說似乎可以用阿里小號來註冊微信，不過聽說存在反覆回收和安全隱患問題，故不採用。

其次，需要用到一個Python庫 itchat，這個庫已經做好了用程式碼呼叫微信的大多數功能，非常好用，官方文件在這裡，安裝的時候使用 pip 即可。

pip install itchat複製程式碼

我的手機支援雙卡雙待，於是把兩張卡都裝手機裡，再雙開微信，同時保持兩個微訊號手機線上，差不多就可以開始寫程式碼了。用 itchat 呼叫微信主要是模擬微信網頁版登陸，所以必須保持微訊號手機線上，因為手機端微信一旦退出，其在網頁、PC、MAC、IPAD等相應終端認證的賬號也會隨之退出。

初步嘗試

itchat 提供了一些官方程式碼，讓我們在自己的本本或電腦上新建一個 py 檔案，初步嘗試一下。

執行以下程式碼，會出現出現一張二維碼，掃碼登陸之後將會給“檔案傳輸助手”傳送一條訊息。

# 載入包
import itchat
# 登陸
itchat.auto_login()
# 傳送文字訊息，傳送目標是“檔案傳輸助手”
itchat.send('Hello, filehelper', toUserName='filehelper')複製程式碼

以下程式碼則註冊了一個訊息響應事件，用來定義接收到文字訊息後如何處理。在 itchat 裡可以定義文字、圖片、名片、位置、通知、分享、檔案等多種訊息型別，從而分別執行不同的處理。

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 列印出來，是一個字典，看看有哪些感興趣的欄位。

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()複製程式碼

開發訊息同步機器人

經過以上示例程式碼，可以總結出訊息同步機器人的開發思路：

  • 登陸後使用get_chatrooms()獲取全部群聊的資料，包括每個群聊的ID和暱稱，可以將需要同步訊息的群聊儲存至通訊錄；
  • 接收到群聊訊息時，如果訊息來自於需要同步訊息的群聊，就根據訊息型別進行處理，同時轉發到其他需要同步的群聊。

直接上程式碼好了，首先定義一個訊息響應函式，文字類訊息我感興趣的是 TEXT 和 SHARING 兩類，使用 isGroupChat=True 指定訊息來自於群聊，這個引數預設為 False。

# 自動回覆文字等類別的群聊訊息
# isGroupChat=True表示為群聊訊息
@itchat.msg_register([TEXT, SHARING], isGroupChat=True)
def group_reply_text(msg):
    # 訊息來自於哪個群聊
    chatroom_id = msg['FromUserName']
    # 傳送者的暱稱
    username = msg['ActualNickName']

    # 訊息並不是來自於需要同步的群
    if not chatroom_id in chatroom_ids:
        return

    if msg['Type'] == TEXT:
        content = msg['Content']
    elif msg['Type'] == SHARING:
        content = msg['Text']

    # 根據訊息型別轉發至其他群
    if msg['Type'] == TEXT:
        for item in chatrooms:
            if not item['UserName'] == chatroom_id:
                itchat.send('%s\n%s' % (username, msg['Content']), item['UserName'])
    elif msg['Type'] == SHARING:
        for item in chatrooms:
            if not item['UserName'] == chatroom_id:
                itchat.send('%s\n%s\n%s' % (username, msg['Text'], msg['Url']), item['UserName'])複製程式碼

再來處理下圖片等多媒體類訊息。

# 自動回覆圖片等類別的群聊訊息
# isGroupChat=True表示為群聊訊息          
@itchat.msg_register([PICTURE, ATTACHMENT, VIDEO], isGroupChat=True)
def group_reply_media(msg):
    # 訊息來自於哪個群聊
    chatroom_id = msg['FromUserName']
    # 傳送者的暱稱
    username = msg['ActualNickName']

    # 訊息並不是來自於需要同步的群
    if not chatroom_id in chatroom_ids:
        return

    # 如果為gif圖片則不轉發
    if msg['FileName'][-4:] == '.gif':
        return

    # 下載圖片等檔案
    msg['Text'](msg['FileName'])
    # 轉發至其他需要同步訊息的群聊
    for item in chatrooms:
        if not item['UserName'] == chatroom_id:
            itchat.send('@%s@%s' % ({'Picture': 'img', 'Video': 'vid'}.get(msg['Type'], 'fil'), msg['FileName']), item['UserName'])複製程式碼

以上程式碼實現了對文字、分享、圖片、視訊四類訊息的處理，如果對其他型別的訊息也感興趣，進行相應的處理即可。在前面補上 import 的程式碼，在後面補上登陸、獲取群聊資料和開始監測的程式碼，就大功告成了。

完整程式碼在這裡：pan.baidu.com/s/1bpAJk0B

成果展示

目前兩個群之間可以進行訊息同步了，一群和二群的小夥伴終於可以暢快地聊了起來（當群主不容易，經常要發很多紅包 = =）。

基於itchat實現微信群訊息同步機器人

基於itchat實現微信群訊息同步機器人

進一步工作

當然，我不可能一直在筆記本上執行這麼個 py 程式碼，所以把它部署到伺服器上執行就好了，開個 screen 或者用 IPython 都可以。如果賬號偶爾下線了，再執行一下就好。

另外，我還寫了個 API，響應訊息的時候會把相應的資料 POST 到我的伺服器並存到資料庫，以供進一步的分析、統計和展示，這也是我身為一個群主應盡的職責～

如果覺得文章不錯，不妨點一下左下方的喜歡～

相關文章