mirror of
https://github.com/smallevilbeast/ntchat.git
synced 2025-05-23 02:26:11 +08:00
119 lines
4.1 KiB
Python
119 lines
4.1 KiB
Python
# -*- coding: utf-8 -*-
|
||
from email import message
|
||
from email.mime import image
|
||
import sys
|
||
import os.path
|
||
import time
|
||
import ntchat
|
||
import re
|
||
|
||
# 聊天记录通知
|
||
MT_RECV_CHAT_RECORDS_MSG = 11061
|
||
|
||
wechat = ntchat.WeChat()
|
||
|
||
# 要监听的wxids，可以通过获取contact接口获取wxid，也可以开启后从debug信息中看出来
|
||
from_wxids = ["xxxxx", "xxxxxx"]
|
||
# 要转发的目标群
|
||
target_wxids = ["xxxxxxxx@chatroom"]
|
||
# 检查文件等待时长，单位s
|
||
wait_limit = 10
|
||
|
||
# 打开pc微信, smart: 是否管理已经登录的微信
|
||
wechat.open(smart=True)
|
||
|
||
|
||
@wechat.msg_register(ntchat.MT_RECV_TEXT_MSG)
|
||
def on_recv_text_msg(wechat_instance: ntchat.WeChat, message):
|
||
data = message["data"]
|
||
from_wxid = data["from_wxid"]
|
||
self_wxid = wechat_instance.get_login_info()["wxid"]
|
||
|
||
# 判断消息不是自己发的，且来自于想要转发的用户列表，并发给target用户
|
||
if from_wxid != self_wxid and from_wxid in from_wxids:
|
||
for target_wxid in target_wxids:
|
||
wechat_instance.send_text(to_wxid=target_wxid,
|
||
content=f"{data['msg']}")
|
||
|
||
|
||
# 等待file_path的文件被下载,超过等待次数后返回true
|
||
def wait_for_file(file_path) -> bool:
|
||
cnt = 0
|
||
while not os.path.exists(file_path):
|
||
time.sleep(1)
|
||
cnt = cnt + 1
|
||
if cnt > wait_limit:
|
||
print(
|
||
f"wait for {wait_limit} second, but file cannot be downloaded, forgive."
|
||
)
|
||
return False
|
||
return True
|
||
|
||
|
||
@wechat.msg_register(ntchat.MT_RECV_IMAGE_MSG)
|
||
def on_recv_img_msg(wechat_instance: ntchat.WeChat, message):
|
||
data = message["data"]
|
||
from_wxid = data["from_wxid"]
|
||
img_path = data["image"]
|
||
# img_path = "D:\\a.png"
|
||
self_wxid = wechat_instance.get_login_info()["wxid"]
|
||
|
||
# 判断消息不是自己发的，且来自于想要转发的用户列表，并发给target用户
|
||
if from_wxid != self_wxid and from_wxid in from_wxids:
|
||
if wait_for_file(file_path=img_path):
|
||
for target_wxid in target_wxids:
|
||
wechat_instance.send_image(to_wxid=target_wxid, file_path=img_path)
|
||
|
||
|
||
@wechat.msg_register(ntchat.MT_RECV_FILE_MSG)
|
||
def on_recv_img_msg(wechat_instance: ntchat.WeChat, message):
|
||
data = message["data"]
|
||
from_wxid = data["from_wxid"]
|
||
file_path = data["file"]
|
||
self_wxid = wechat_instance.get_login_info()["wxid"]
|
||
|
||
# 判断消息不是自己发的，且来自于想要转发的用户列表，并发给target用户
|
||
if from_wxid != self_wxid and from_wxid in from_wxids:
|
||
if wait_for_file(file_path=file_path):
|
||
for target_wxid in target_wxids:
|
||
wechat_instance.send_file(to_wxid=target_wxid, file_path=file_path)
|
||
|
||
|
||
def update_wxid_in_xml(xml, from_wxid, target_wxid):
|
||
patten = re.compile(from_wxid)
|
||
return patten.sub(target_wxid, xml)
|
||
|
||
|
||
@wechat.msg_register(MT_RECV_CHAT_RECORDS_MSG)
|
||
def on_recv_chat_record_msg(wechat_instance: ntchat.WeChat, message):
|
||
data = message["data"]
|
||
from_wxid = data["from_wxid"]
|
||
raw_msg = data["raw_msg"]
|
||
self_wxid = wechat_instance.get_login_info()["wxid"]
|
||
xml = update_wxid_in_xml(raw_msg, from_wxid, self_wxid)
|
||
# 判断消息不是自己发的，且来自于想要转发的用户列表，并发给target用户
|
||
if from_wxid != self_wxid and from_wxid in from_wxids:
|
||
for target_wxid in target_wxids:
|
||
wechat_instance.send_xml(to_wxid=target_wxid, xml=xml)
|
||
|
||
|
||
@wechat.msg_register(ntchat.MT_RECV_LINK_MSG)
|
||
def on_recv_link_msg(wechat_instance: ntchat.WeChat, message):
|
||
data = message["data"]
|
||
from_wxid = data["from_wxid"]
|
||
raw_msg = data["raw_msg"]
|
||
# xml中的fromusername改成自己的wxid 再发
|
||
self_wxid = wechat_instance.get_login_info()["wxid"]
|
||
xml = update_wxid_in_xml(raw_msg, from_wxid, self_wxid)
|
||
# 判断消息不是自己发的，且来自于想要转发的用户列表，并发给target用户
|
||
if from_wxid != self_wxid and from_wxid in from_wxids:
|
||
for target_wxid in target_wxids:
|
||
wechat_instance.send_xml(to_wxid=target_wxid, xml=xml)
|
||
|
||
|
||
try:
|
||
while True:
|
||
pass
|
||
except KeyboardInterrupt:
|
||
ntchat.exit_()
|
||
sys.exit() |