357 lines
15 KiB
Python
357 lines
15 KiB
Python
from fastapi import Depends, FastAPI, HTTPException
|
||
from sqlalchemy.orm import Session
|
||
|
||
# from . import crud, models, schemas
|
||
import crud, models, schemas
|
||
from database import SessionLocal, engine
|
||
# from .database import SessionLocal, engine
|
||
|
||
import json
|
||
import os
|
||
import threading
|
||
import time
|
||
from pydantic import BaseModel
|
||
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.dom.minidom
|
||
from pydantic import BaseModel
|
||
|
||
models.Base.metadata.create_all(bind=engine)
|
||
|
||
app = FastAPI()
|
||
|
||
|
||
# Dependency
|
||
def get_db():
|
||
db = SessionLocal()
|
||
try:
|
||
yield db
|
||
finally:
|
||
db.close()
|
||
|
||
|
||
# 默认首页
|
||
@app.get("/")
|
||
async def root():
|
||
return {"message": "Hello World"}
|
||
|
||
|
||
# 创建用户
|
||
@app.post("/users/")
|
||
# schemas.UserCreate决定了swagger文档
|
||
def create_user(user: schemas.User, db: Session = Depends(get_db)):
|
||
print('user', user)
|
||
print('typeuser', type(user))
|
||
|
||
# 检测是否存在wxid
|
||
db_user = crud.get_user_by_wxid(db, user.wxid)
|
||
print('db_user', db_user)
|
||
|
||
if db_user:
|
||
raise HTTPException(status_code=400, detail="wxid already registered")
|
||
return crud.create_user(db=db, user=user)
|
||
|
||
|
||
# 分页查询-用户
|
||
@app.get("/users/")
|
||
def read_users(skip: int = 0, limit: int = 100, db: Session = Depends(get_db)):
|
||
users = crud.get_users(db, skip=skip, limit=limit)
|
||
return users
|
||
|
||
|
||
# 根据openId查询用户
|
||
@app.post("/openId")
|
||
def read_user(user: schemas.User, db: Session = Depends(get_db)):
|
||
db_user = crud.get_user_by_openId(db, openId=user.openId)
|
||
if db_user is None:
|
||
raise HTTPException(status_code=404, detail="openId not found")
|
||
return db_user
|
||
|
||
|
||
# 根据wxid查询用户
|
||
@app.get("/users")
|
||
def read_user(wxid, db: Session = Depends(get_db)):
|
||
db_user = crud.get_user_by_wxid(db, wxid=wxid)
|
||
if db_user is None:
|
||
raise HTTPException(status_code=404, detail="User-wxid not found")
|
||
return db_user
|
||
|
||
|
||
# 监听机器人http-get消息
|
||
@app.get("/wxbot")
|
||
def helloget(q: dict):
|
||
data = q
|
||
t = threading.Thread(target=kp, args=(data, )) # 通过当前线程开启新的线程去执行
|
||
t.start()
|
||
return json.dumps({"Code": 0})
|
||
|
||
|
||
# 监听机器人post日志消息
|
||
@app.post("/wxbot")
|
||
# def hellopost(requestData:schemas.User):
|
||
def hellopost(requestData: dict):
|
||
print('requestData---', requestData)
|
||
data = requestData
|
||
t = threading.Thread(target=kp, args=(data, )) # 通过当前线程开启新的线程去执行
|
||
t.start()
|
||
return json.dumps({"Code": 0})
|
||
|
||
|
||
# 保存好友信息
|
||
def saveWxidInfo(account_wxid, wxid):
|
||
allFriend = WXBotService.get_address_list(account_wxid)
|
||
for friendItem in allFriend['data']["contactList"]:
|
||
if friendItem['wxid'] == wxid:
|
||
friendItemObj = models.User(**friendItem)
|
||
# 当前的路由函数
|
||
create_user(db=SessionLocal(), user=friendItemObj)
|
||
|
||
|
||
# 发送小程序模版,建议远程更新
|
||
def sendminiProgram(data, account_wxid, wxid, msgXml):
|
||
# 发送小程序
|
||
if data.get('isSender') != 1:
|
||
root = ET.fromstring(msgXml) #xml-字符串形式
|
||
|
||
#打开xml文档
|
||
# current_dir = os.path.dirname(os.path.abspath(__file__))
|
||
# print('current_dir+/planet.xml',current_dir+'\planet.xml')
|
||
# tree = ET.parse(current_dir+'\planet.xml')
|
||
# root = tree.getroot()
|
||
|
||
# 微信好友的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'
|
||
msgXml = ET.tostring(root, encoding='utf-8').decode('utf-8')
|
||
WXBotService.send_xml_message(
|
||
account_wxid, wxid, msgXml
|
||
# account_wxid, "wxid_w2imh7ki5s4q12", planetXml
|
||
)
|
||
|
||
|
||
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') #机器人id
|
||
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') #类型
|
||
thumbpath = data.get('thumbpath') #图片,存在电脑绝对路径
|
||
|
||
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}'
|
||
# 将好友信息=>保存到数据库
|
||
newFriendDict = WXBotService.get_friend_info(account_wxid, wxid)
|
||
# print('newFriendDict[data]---',newFriendDict['data'])
|
||
userClass = models.User(**(newFriendDict['data']))
|
||
# print('userClass====',type(userClass))
|
||
# crud.create_user(db=SessionLocal(),user=userClass)
|
||
|
||
# 保存好友信息
|
||
saveWxidInfo(account_wxid, wxid)
|
||
|
||
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'))
|
||
newFriendWxid = xml_msg.get('alias')
|
||
# 将好友信息=>保存到数据库
|
||
# newFriendDict=WXBotService.get_friend_info(account_wxid, newFriendWxid)
|
||
# userClass=models.User(**(newFriendDict['data']))
|
||
# crud.create_user(db= SessionLocal(),user=userClass)
|
||
|
||
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)
|
||
# 发送小程序
|
||
sendminiProgram(data, account_wxid, wxid, 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)
|