mirror of
https://github.com/smallevilbeast/ntchat.git
synced 2025-07-07 20:49:43 +08:00
add transmit example
This commit is contained in:
parent
476e8428cb
commit
99ac6bf007
117
examples/transmit.py
Normal file
117
examples/transmit.py
Normal file
|
@ -0,0 +1,117 @@
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
from email import message
|
||||||
|
from email.mime import image
|
||||||
|
import sys
|
||||||
|
import os.path
|
||||||
|
import time
|
||||||
|
import ntchat
|
||||||
|
import xml.etree.ElementTree as ET
|
||||||
|
import re
|
||||||
|
|
||||||
|
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(ntchat.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()
|
|
@ -23,6 +23,9 @@ MT_RECV_TEXT_MSG = 11046
|
||||||
MT_RECV_IMAGE_MSG = 11047
|
MT_RECV_IMAGE_MSG = 11047
|
||||||
MT_RECV_PICTURE_MSG = 11047
|
MT_RECV_PICTURE_MSG = 11047
|
||||||
|
|
||||||
|
# 聊天记录通知
|
||||||
|
MT_RECV_CHAT_RECORDS_MSG = 11061
|
||||||
|
|
||||||
# 语音消息通知
|
# 语音消息通知
|
||||||
MT_RECV_VOICE_MSG = 11048
|
MT_RECV_VOICE_MSG = 11048
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user