249 lines
12 KiB
Python
249 lines
12 KiB
Python
import json
|
||
import os
|
||
import threading
|
||
import time
|
||
|
||
from flask import request, Flask
|
||
try:
|
||
import xml.etree.cElementTree as ET
|
||
except ImportError:
|
||
import xml.etree.ElementTree as ET
|
||
|
||
import WXBotService
|
||
from planet import planetXml
|
||
from lxml import etree
|
||
import xml.etree.ElementTree as ETree
|
||
|
||
wxbot = Flask(__name__)
|
||
|
||
|
||
@wxbot.route("/wxbot", methods=['get', 'post'])
|
||
def hello():
|
||
data = request.get_json()
|
||
t = threading.Thread(target=kp, args=(data, )) # 通过当前线程开启新的线程去执行
|
||
t.start()
|
||
return json.dumps({"Code": 0})
|
||
|
||
|
||
def kp(event_data):
|
||
# unicode [\\uxxxxx] 转 emoji 表情
|
||
event_data = json.loads(
|
||
WXBotService.unicode_to_emoji(
|
||
json.dumps(event_data, ensure_ascii=False), 'json'))
|
||
|
||
kp_log = '' # 定义鲲鹏日志消息
|
||
event_type = event_data.get('event_type')
|
||
account_wxid = event_data.get('account_wxid')
|
||
data = event_data.get('data')
|
||
msg_type = data.get('msgtype')
|
||
msg = data.get('content')
|
||
to_wxid = data.get('to_wxid')
|
||
to_name = data.get('to_name')
|
||
filepath = data.get("filepath")
|
||
subtype = data.get('subtype')
|
||
|
||
wxid = data.get('final_from_wxid')
|
||
wx_name = data.get('final_from_name')
|
||
group = None if data.get('from_wxid') == wxid else data.get('from_wxid')
|
||
group_name = None if data.get('from_name') == wx_name else data.get(
|
||
'from_name')
|
||
'''event类型判断'''
|
||
if event_type == 1001: # 1001 创建WeChat进程成功，通讯连接成功
|
||
kp_log = f'创建WeChat进程成功，通讯连接成功'
|
||
elif event_type == 1002: # 1002 登录成功
|
||
kp_log = f'登录成功 {data.get("nickname")}（{data.get("wxid")}）'
|
||
elif event_type == 1003: # 1003 账号注销登录
|
||
kp_log = f'账号注销登录 {data.get("wxid")} {data.get("reason")}'
|
||
elif event_type == 1004: # 1004 切换聊天对象事件
|
||
kp_log = f'切换聊天对象事件 {data.get("nickname")}（{data.get("wxid")}）'
|
||
elif event_type == 1005: # 1005 切换联系人事件
|
||
kp_log = f'切换联系人事件 {data.get("nickname")}（{data.get("wxid")}）'
|
||
elif event_type == 1007: # 1005 二维码刷新事件
|
||
kp_log = f'二维码刷新事件 {data.get("qrcode")}'
|
||
elif event_type == 1008: # 1008 连接断开
|
||
kp_log = f'连接断开'
|
||
elif event_type == 1009: # 1009 群员退群通知
|
||
member_list = []
|
||
for member in data.get('memberlist'):
|
||
member_list.append(
|
||
f'{member.get("nickname")}（{member.get("wxid")}）')
|
||
kp_log = f'群员退群通知 {"，".join(member_list)} 退出了群聊 {data.get("roomname")}（{data.get("roomid")}） ' \
|
||
f'人数：{data.get("membercount")}'
|
||
elif event_type == 1010: # 1010 群员进群通知
|
||
member_list = []
|
||
for member in data.get('memberlist'):
|
||
member_list.append(
|
||
f'{member.get("nickName")}（{member.get("userName")}）')
|
||
kp_log = f'群员进群通知 {"，".join(member_list)} 加入了群聊 {data.get("roomname")}（{data.get("roomid")}）'
|
||
elif event_type == 1011: # 1011 收款消息
|
||
pay_msg = ET.fromstring(msg).find('paymsg')
|
||
pay_msg_type = pay_msg.findtext('PayMsgType')
|
||
if pay_msg_type == '9': # 二维码收款
|
||
status = pay_msg.findtext('status')
|
||
username = pay_msg.findtext('username')
|
||
display_name = pay_msg.findtext('displayname')
|
||
fee = pay_msg.findtext('fee')
|
||
if status == '0':
|
||
kp_log = f'收款消息 {display_name}（{username}） 扫描收款二维码'
|
||
elif status == '1':
|
||
kp_log = f'收款消息 {display_name}（{username}） 付款成功 {int(fee) / 100} 元'
|
||
elif status == '2':
|
||
kp_log = f'收款消息 {display_name}（{username}） 取消付款'
|
||
elif pay_msg_type == '45': # 貌似有未收款转账或转账红包过期会触发 42 和 45 类型
|
||
pass # kp_log = f'收款消息 {pay_msg.findtext("fromusername")} {pay_msg.findtext("content")}'
|
||
elif event_type == 1040: # 1040 撤回消息事件
|
||
if not group:
|
||
kp_log = f'撤回消息事件 {wx_name}（{wxid}） 撤回了 {msg}'
|
||
else:
|
||
kp_log = f'撤回消息事件 {wx_name}（{wxid}） 从群 {group_name}（{group}） 撤回了 {msg}'
|
||
elif event_type == 1100: # 1100 同步消息
|
||
if group:
|
||
type_str = f'同步消息 群聊消息 群聊：{group_name}（{group}） 发送者：{wx_name}（{wxid}）'
|
||
elif 'gh_' in wxid:
|
||
type_str = f'同步消息 公众号消息 {wx_name}（{wxid}）'
|
||
elif '@openim' in wxid:
|
||
type_str = f'同步消息 企业微信消息 {wx_name}（{wxid}）'
|
||
else:
|
||
type_str = f'同步消息 私聊消息 {wx_name}（{wxid}）'
|
||
|
||
if data.get('isSender') == 1:
|
||
# 手机发送还是电脑发送这里判断不是很准确
|
||
if to_wxid == account_wxid:
|
||
type_str = f'机器人手机端 发送的 ' + type_str
|
||
else:
|
||
type_str = f'机器人电脑端 发送的 ' + type_str
|
||
|
||
if msg_type == 1: # 1 文本消息/自带表情
|
||
kp_log = f'{type_str} 文本消息 内容：{msg}'
|
||
elif msg_type == 3: # 3 图片消息
|
||
kp_log = f'{type_str} 图片消息 内容：{filepath}'
|
||
# 本地dat图片解密
|
||
decrypt_path = WXBotService.local_decrypt_image(
|
||
filepath, os.path.join(os.getcwd(), '解密图片'))
|
||
print(f'{type_str} 图片消息 解密路径：{decrypt_path}')
|
||
elif msg_type == 34: # 34 语言消息（需要在设置中打开语言自动转文字）
|
||
xml_msg = ET.fromstring(msg)
|
||
kp_log = f'{type_str} 语音消息 内容：{xml_msg.find("voicetrans").get("transtext")}'
|
||
elif msg_type == 37: # 37 好友请求
|
||
xml_msg = ET.fromstring(msg)
|
||
kp_log = f'同步消息 好友请求 {xml_msg.get("fromnickname")}（{xml_msg.get("fromusername")}） ' \
|
||
f'验证内容：{xml_msg.get("content")}'
|
||
# 同意好友请求
|
||
WXBotService.agree_friend_request(account_wxid,
|
||
xml_msg.get('alias'),
|
||
xml_msg.get('encryptusername'),
|
||
xml_msg.get('ticket'),
|
||
xml_msg.get('scene'))
|
||
elif msg_type == 42: # 42 名片消息
|
||
xml_msg = ET.fromstring(msg)
|
||
kp_log = f'{type_str} 企业微信名片消息 {xml_msg.get("nickname")}（{xml_msg.get("username")}）'
|
||
elif msg_type == 43: # 43 视频消息
|
||
kp_log = f'{type_str} 视频消息 内容：{filepath}'
|
||
elif msg_type == 47: # 47 表情包
|
||
kp_log = f'{type_str} 表情包消息 网址：{ET.fromstring(msg).find("emoji").get("cdnurl".replace("amp;", ""))}'
|
||
# 本地下载图片
|
||
down_path = WXBotService.local_download_image(
|
||
msg, os.path.join(os.getcwd(), '下载图片'))
|
||
print(f'{type_str} 表情包消息 下载路径：{down_path}')
|
||
elif msg_type == 48: # 48 地理位置
|
||
kp_log = f'{type_str} 地理位置 内容：{msg}'
|
||
|
||
|
||
elif msg_type == 49: # 49 xml消息
|
||
# 获取根节点
|
||
root=ET.fromstring(msg)
|
||
xml_appmsg = root.find('appmsg')
|
||
|
||
xml_title = xml_appmsg.findtext('title')
|
||
xml_type = xml_appmsg.findtext('type')
|
||
xml_des = xml_appmsg.findtext('des')
|
||
xml_url = xml_appmsg.findtext('url')
|
||
|
||
if xml_type == '1': # 1 文本
|
||
kp_log = f'收到 {wxid} xml文本消息'
|
||
elif xml_type == '3': # 3 歌曲
|
||
kp_log = f'{type_str} 音乐消息 歌曲：{xml_title} 歌手：{xml_des} 网址：{xml_url}'
|
||
elif xml_type == '5': # 5 链接
|
||
kp_log = f'{type_str} 链接消息 标题：{xml_title} 内容：{xml_des} 网址：{xml_url}'
|
||
print("kp_log--------------------------------",msg)
|
||
# 本身也是系统消息-link,会无线循环
|
||
# 判断是不是自己发的消息
|
||
# if data.get('isSender') != 1:
|
||
# WXBotService.send_link_message(
|
||
# account_wxid, "wxid_w2imh7ki5s4q12", "title", "dec",
|
||
# "https://www.baidu.com",
|
||
# "https://p1-q.mafengwo.net/s17/M00/E3/9C/CoUBXl_cawCAQWMcAAGGhEoBcRU77.jpeg")
|
||
|
||
|
||
elif xml_type == '6': # 6 文件
|
||
kp_log = f'{type_str} 文件消息 文件名：{xml_title} 文件大小：{xml_des} 文件路径：{filepath}'
|
||
elif xml_type == '19': # 19 合并转发聊天记录
|
||
kp_log = f'{type_str} 合并转发聊天记录 文件名：{xml_title} 文件大小：{xml_des} 文件路径：{filepath}'
|
||
elif xml_type == '33': # 33 小程序
|
||
kp_log = f'{type_str} 小程序消息 小程序名：{xml_appmsg.findtext("sourcedisplayname")} 标题：{xml_title} ' \
|
||
f'appid：{xml_appmsg.findtext("weappinfo/appid")} ' \
|
||
f'pagepath：{xml_appmsg.findtext("weappinfo/pagepath")}'
|
||
print("---------------------------kp_log",msg)
|
||
# print("---------------------------kp_log",planetXml)
|
||
# print('msg---', msg)
|
||
# 发送小程序
|
||
if data.get('isSender') != 1:
|
||
root=ET.fromstring(msg)
|
||
|
||
# 微信好友的wxid
|
||
wxid='wxid_w2imh7ki5s4q12'
|
||
for pagepath in root.iter('pagepath'):
|
||
print (pagepath.text) # 打印名字
|
||
pagepath.text =pagepath.text +f'?wxid={wxid}'
|
||
print ('修改path',pagepath.text) # 打印名字
|
||
|
||
# 修改后的xml转成string , method='xml'
|
||
msg =ET.tostring(root, encoding='utf-8').decode('utf-8')
|
||
|
||
WXBotService.send_xml_message(
|
||
account_wxid, "wxid_w2imh7ki5s4q12", msg
|
||
# account_wxid, "wxid_w2imh7ki5s4q12", planetXml
|
||
|
||
)
|
||
|
||
|
||
elif xml_type == '57': # 引用消息
|
||
kp_log = f'{type_str} 引用消息 内容：{msg}'
|
||
elif xml_type == '2000': # 2000 转账消息
|
||
money = xml_appmsg.findtext('wcpayinfo/feedesc').replace(
|
||
'￥', '')
|
||
pay_subtype = xml_appmsg.findtext('wcpayinfo/paysubtype')
|
||
transfer_id = xml_appmsg.findtext('wcpayinfo/transferid')
|
||
if pay_subtype == '1':
|
||
kp_log = f'{type_str} 转账消息 {money} 元'
|
||
# 自动接收转账
|
||
WXBotService.transfer_receipt(account_wxid, wxid,
|
||
transfer_id)
|
||
elif pay_subtype == '3':
|
||
kp_log = f'{type_str} 已收款 {money} 元 转账'
|
||
elif msg_type == 66: # 66 企业微信名片消息
|
||
xml_msg = ET.fromstring(msg)
|
||
kp_log = f'企业微信名片消息 {xml_msg.get("nickname")}（{xml_msg.get("username")}）'
|
||
elif msg_type == 10000: # 10000 系统消息
|
||
kp_log = f'系统消息 {wx_name}（{wxid}） {msg}'
|
||
|
||
#获取通讯录
|
||
# allFriends=WXBotService.get_address_list(account_wxid)
|
||
# print('allFriends',allFriends)
|
||
# 发送H5链接
|
||
|
||
if not kp_log:
|
||
print(
|
||
'===================================================================================='
|
||
)
|
||
print(event_data)
|
||
print(
|
||
'===================================================================================='
|
||
)
|
||
else:
|
||
print(kp_log)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
wxbot.run('127.0.0.1', 28369)
|