Merge pull request #38 from IcePigZDB/add_transmit_example

add transmit example

examples/transmit.py

@@ -0,0 +1,119 @@
+# -*- 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()