python基于itchat實現微信群消息同步機器人

2019-11-25 16:20:31

字體：大中小

來源：轉載

供稿：網友

最近全棧數據工程師養成攻略的微信群已經將近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()

開發消息同步機器人

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

維護一個字典，叫做 groups 好了，用來存所有需要同步消息的群聊，key 為群聊的ID，value 為群聊的名稱；
接收到群聊消息時，如果消息來自于需要同步消息的群聊，就根據消息類型進行處理，同時轉發到其他需要同步的群聊。

直接上代碼好了，首先定義一個消息響應函數，文本類消息我感興趣的是 TEXT 和 SHARING 兩類，使用 isGroupChat=True 指定消息來自于群聊，這個參數默認為 False。

@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)

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

# 處理圖片和視頻類消息@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 的代碼，在后面補上登陸和運行的代碼，就大功告成了。

成果展示

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