快捷導(dǎo)航

python基于itchat實現(xiàn)微信群消息同步機器人

更新時間：2017年02月27日 10:07:46 作者：宏倫工作室

本篇文章主要介紹了python基于itchat實現(xiàn)微信群消息同步機器人，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧

最近全棧數(shù)據(jù)工程師養(yǎng)成攻略的微信群已經(jīng)將近500人，開了二群之后為了打通不同微信群之間的消息，花了點時間做了個消息同步機器人，在任意群收到消息時同步到其他群，并且將聊天內(nèi)容上傳至數(shù)據(jù)庫，以供進一步分析、統(tǒng)計和展示。

基本思路是，用 Python 模擬微信登陸，接收到群里消息后，對文本、圖片、分享等各類消息類型分別處理，并轉(zhuǎn)發(fā)至其他群。

前期準(zhǔn)備

首先得有一個微信號，用于代碼模擬登陸。由于我的微信號得自己留著用，現(xiàn)階段注冊微信又必須要手機號，于是只好特意辦了個電信號，用來申請了一個新的微信，微信號是 honlanbot。雖說似乎可以用阿里小號來注冊微信，不過聽說存在反復(fù)回收和安全隱患問題，故不采用。

其次，需要用到一個Python庫 itchat，這個庫已經(jīng)做好了用代碼調(diào)用微信的大多數(shù)功能，非常好用，官方文檔在這里，安裝的時候使用 pip 即可。

pip install itchat

我的手機支持雙卡雙待，于是把兩張卡都裝手機里，再雙開微信，同時保持兩個微信號手機在線，差不多就可以開始寫代碼了。用 itchat 調(diào)用微信主要是模擬微信網(wǎng)頁版登陸，所以必須保持微信號手機在線，因為手機端微信一旦退出，其在網(wǎng)頁、PC、MAC、IPAD等相應(yīng)終端認(rèn)證的賬號也會隨之退出。

初步嘗試

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

運行以下代碼，會出現(xiàn)出現(xiàn)一張二維碼，掃碼登陸之后將會給“文件傳輸助手”發(fā)送一條消息。

# 加載包
import itchat
# 登陸
itchat.auto_login()
# 發(fā)送文本消息，發(fā)送目標(biāo)是“文件傳輸助手”
itchat.send('Hello, filehelper', toUserName='filehelper')

以下代碼則注冊了一個消息響應(yīng)事件，用來定義接收到文本消息后如何處理。在 itchat 里定義了文本、圖片、名片、位置、通知、分享、文件等多種消息類型，可以分別執(zhí)行不同的處理。

import itchat
# 注冊消息響應(yīng)事件，消息類型為itchat.content.TEXT，即文本消息
@itchat.msg_register(itchat.content.TEXT)
def text_reply(msg):
  # 返回同樣的文本消息
  return msg['Text']

itchat.auto_login()
# 綁定消息響應(yīng)事件后，讓itchat運行起來，監(jiān)聽消息
itchat.run()

再來看看如何處理其他類型消息，可以把在消息響應(yīng)事件里把 msg 打印出來，是一個字典，看看有哪些感興趣的字段。

import itchat
# import全部消息類型
from itchat.content import *

# 處理文本類消息
# 包括文本、位置、名片、通知、分享
@itchat.msg_register([TEXT, MAP, CARD, NOTE, SHARING])
def text_reply(msg):
  # 微信里，每個用戶和群聊，都使用很長的ID來區(qū)分
  # msg['FromUserName']就是發(fā)送者的ID
  # 將消息的類型和文本內(nèi)容返回給發(fā)送者
  itchat.send('%s: %s' % (msg['Type'], msg['Text']), msg['FromUserName'])

# 處理多媒體類消息
# 包括圖片、錄音、文件、視頻
@itchat.msg_register([PICTURE, RECORDING, ATTACHMENT, VIDEO])
def download_files(msg):
  # msg['Text']是一個文件下載函數(shù)
  # 傳入文件名，將文件下載下來
  msg['Text'](msg['FileName'])
  # 把下載好的文件再發(fā)回給發(fā)送者
  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
# 即可保留登陸狀態(tài)
# 即使程序關(guān)閉，一定時間內(nèi)重新開啟也可以不用重新掃碼
itchat.auto_login(True)
itchat.run()

開發(fā)消息同步機器人

經(jīng)過以上示例代碼，可以總結(jié)出消息同步機器人的開發(fā)思路：

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

直接上代碼好了，首先定義一個消息響應(yīng)函數(shù)，文本類消息我感興趣的是 TEXT 和 SHARING 兩類，使用 isGroupChat=True 指定消息來自于群聊，這個參數(shù)默認(rèn)為 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):
      # 轉(zhuǎn)發(fā)到其他需要同步消息的群聊
      for item in groups.keys():
        if not item == source:
          # groups[source]: 消息來自于哪個群聊
          # msg['ActualNickName']: 發(fā)送者的名稱
          # msg['Content']: 文本消息內(nèi)容
          # item: 需要被轉(zhuǎn)發(fā)的群聊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']: 分享的標(biāo)題
          # 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:
        # 將圖片或視頻發(fā)送到其他需要同步消息的群聊
        itchat.send('@%s@%s' % ({'Picture': 'img', 'Video': 'vid'}.get(msg['Type'], 'fil'), msg['FileName']), item)

以上代碼實現(xiàn)了對文本、分享、圖片、視頻四類消息的處理，如果對其他類型的消息也感興趣，進行相應(yīng)的處理即可。在前面補上 import 的代碼，在后面補上登陸和運行的代碼，就大功告成了。

成果展示

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