python基於itchat實現微信群訊息同步機器人-微信開發-PHP中文網

首頁

微信小程式

微信開發

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

高洛峰

Feb 28, 2017 am 09:03 AM

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

基本想法是，用 Python 模擬微信登陸，接收群組裡訊息後，分別對文字、圖片、分享等各類別訊息類型進行處理，並轉發至其他群組。

前期準備

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

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

pip install itchat

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

初步嘗試

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

執行以下程式碼，會出現一張二維碼，在掃碼登陸之後會傳送訊息給「檔案傳輸助理」。

# 加载包
import itchat
# 登陆
itchat.auto_login()
# 发送文本消息，发送目标是“文件传输助手”
itchat.send(&#39;Hello, filehelper&#39;, toUserName=&#39;filehelper&#39;)

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

import itchat
# 注册消息响应事件，消息类型为itchat.content.TEXT，即文本消息
@itchat.msg_register(itchat.content.TEXT)
def text_reply(msg):
  # 返回同样的文本消息
  return msg[&#39;Text&#39;]

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[&#39;FromUserName&#39;]就是发送者的ID
  # 将消息的类型和文本内容返回给发送者
  itchat.send(&#39;%s: %s&#39; % (msg[&#39;Type&#39;], msg[&#39;Text&#39;]), msg[&#39;FromUserName&#39;])

# 处理多媒体类消息
# 包括图片、录音、文件、视频
@itchat.msg_register([PICTURE, RECORDING, ATTACHMENT, VIDEO])
def download_files(msg):
  # msg[&#39;Text&#39;]是一个文件下载函数
  # 传入文件名，将文件下载下来
  msg[&#39;Text&#39;](msg[&#39;FileName&#39;])
  # 把下载好的文件再发回给发送者
  return &#39;@%s@%s&#39; % ({&#39;Picture&#39;: &#39;img&#39;, &#39;Video&#39;: &#39;vid&#39;}.get(msg[&#39;Type&#39;], &#39;fil&#39;), msg[&#39;FileName&#39;])

# 处理好友添加请求
@itchat.msg_register(FRIENDS)
def add_friend(msg):
  # 该操作会自动将新好友的消息录入，不需要重载通讯录
  itchat.add_friend(**msg[&#39;Text&#39;]) 
  # 加完好友后，给好友打个招呼
  itchat.send_msg(&#39;Nice to meet you!&#39;, msg[&#39;RecommendInfo&#39;][&#39;UserName&#39;])

# 处理群聊消息
@itchat.msg_register(TEXT, isGroupChat=True)
def text_reply(msg):
  if msg[&#39;isAt&#39;]:
    itchat.send(u&#39;@%s\u2005I received: %s&#39; % (msg[&#39;ActualNickName&#39;], msg[&#39;Content&#39;]), msg[&#39;FromUserName&#39;])

# 在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[&#39;FromUserName&#39;]

  # 处理文本消息
  if msg[&#39;Type&#39;] == TEXT:
    # 消息来自于需要同步消息的群聊
    if groups.has_key(source):
      # 转发到其他需要同步消息的群聊
      for item in groups.keys():
        if not item == source:
          # groups[source]: 消息来自于哪个群聊
          # msg[&#39;ActualNickName&#39;]: 发送者的名称
          # msg[&#39;Content&#39;]: 文本消息内容
          # item: 需要被转发的群聊ID
          itchat.send(&#39;%s: %s\n%s&#39; % (groups[source], msg[&#39;ActualNickName&#39;], msg[&#39;Content&#39;]), item)
  # 处理分享消息
  elif msg[&#39;Type&#39;] == SHARING:
    if groups.has_key(source):
      for item in groups.keys():
        if not item == source:
          # msg[&#39;Text&#39;]: 分享的标题
          # msg[&#39;Url&#39;]: 分享的链接
          itchat.send(&#39;%s: %s\n%s\n%s&#39; % (groups[source], msg[&#39;ActualNickName&#39;], msg[&#39;Text&#39;], msg[&#39;Url&#39;]), item)

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

# 处理图片和视频类消息
@itchat.msg_register([PICTURE, VIDEO], isGroupChat=True)
def group_reply_media(msg):
  source = msg[&#39;FromUserName&#39;]

  # 下载图片或视频
  msg[&#39;Text&#39;](msg[&#39;FileName&#39;])
  if groups.has_key(source):
    for item in groups.keys():
      if not item == source:
        # 将图片或视频发送到其他需要同步消息的群聊
        itchat.send(&#39;@%s@%s&#39; % ({&#39;Picture&#39;: &#39;img&#39;, &#39;Video&#39;: &#39;vid&#39;}.get(msg[&#39;Type&#39;], &#39;fil&#39;), msg[&#39;FileName&#39;]), item)

以上程式碼實現了對文字、分享、圖片、影片四類訊息的處理，如果對其他類型的訊息也感興趣，進行相應的處理即可。在前面補上 import 的程式碼，在後面補上登陸和運行的程式碼，就大功告成了。

成果展示

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

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