From 99ac6bf0078c14be4cff21e21871509e70411216 Mon Sep 17 00:00:00 2001 From: daobing zhu Date: Fri, 23 Sep 2022 08:42:30 +0800 Subject: [PATCH 1/3] add transmit example --- examples/transmit.py | 117 ++++++++++++++++++++++++++++++++++++ ntchat/const/notify_type.py | 3 + 2 files changed, 120 insertions(+) create mode 100644 examples/transmit.py diff --git a/examples/transmit.py b/examples/transmit.py new file mode 100644 index 0000000..cbba65f --- /dev/null +++ b/examples/transmit.py @@ -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() \ No newline at end of file diff --git a/ntchat/const/notify_type.py b/ntchat/const/notify_type.py index e1d017f..60e748c 100644 --- a/ntchat/const/notify_type.py +++ b/ntchat/const/notify_type.py @@ -23,6 +23,9 @@ MT_RECV_TEXT_MSG = 11046 MT_RECV_IMAGE_MSG = 11047 MT_RECV_PICTURE_MSG = 11047 +# 聊天记录通知 +MT_RECV_CHAT_RECORDS_MSG = 11061 + # 语音消息通知 MT_RECV_VOICE_MSG = 11048 From cd51f463bc50ccd1287ec73275208a1251cd15ba Mon Sep 17 00:00:00 2001 From: daobing zhu Date: Fri, 23 Sep 2022 08:43:35 +0800 Subject: [PATCH 2/3] remove unuse import --- examples/transmit.py | 1 - 1 file changed, 1 deletion(-) diff --git a/examples/transmit.py b/examples/transmit.py index cbba65f..cb5f347 100644 --- a/examples/transmit.py +++ b/examples/transmit.py @@ -5,7 +5,6 @@ import sys import os.path import time import ntchat -import xml.etree.ElementTree as ET import re wechat = ntchat.WeChat() From cdaacf186b7aa316e9640abc91636061daac91a4 Mon Sep 17 00:00:00 2001 From: daobing zhu Date: Fri, 23 Sep 2022 17:23:25 +0800 Subject: [PATCH 3/3] address comment --- examples/transmit.py | 5 ++++- ntchat/const/notify_type.py | 3 --- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/examples/transmit.py b/examples/transmit.py index cb5f347..9062f84 100644 --- a/examples/transmit.py +++ b/examples/transmit.py @@ -7,6 +7,9 @@ import time import ntchat import re +# 聊天记录通知 +MT_RECV_CHAT_RECORDS_MSG = 11061 + wechat = ntchat.WeChat() # 要监听的wxids，可以通过获取contact接口获取wxid，也可以开启后从debug信息中看出来 @@ -81,7 +84,7 @@ def update_wxid_in_xml(xml, from_wxid, target_wxid): return patten.sub(target_wxid, xml) -@wechat.msg_register(ntchat.MT_RECV_CHAT_RECORDS_MSG) +@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"] diff --git a/ntchat/const/notify_type.py b/ntchat/const/notify_type.py index 60e748c..e1d017f 100644 --- a/ntchat/const/notify_type.py +++ b/ntchat/const/notify_type.py @@ -23,9 +23,6 @@ MT_RECV_TEXT_MSG = 11046 MT_RECV_IMAGE_MSG = 11047 MT_RECV_PICTURE_MSG = 11047 -# 聊天记录通知 -MT_RECV_CHAT_RECORDS_MSG = 11061 - # 语音消息通知 MT_RECV_VOICE_MSG = 11048