490 lines
16 KiB
Python
490 lines
16 KiB
Python
import json
|
||
import os
|
||
import socket
|
||
import threading
|
||
import time
|
||
from pathlib import Path
|
||
from queue import Queue
|
||
from typing import Any, Dict
|
||
|
||
from Entity.AnchorModel import AnchorModel
|
||
from Utils.AiUtils import AiUtils
|
||
from Utils.LogManager import LogManager
|
||
import tidevice
|
||
import wda
|
||
from flask import Flask, request
|
||
from flask_cors import CORS
|
||
from Entity.ResultData import ResultData
|
||
from Utils.ControlUtils import ControlUtils
|
||
from Utils.ThreadManager import ThreadManager
|
||
from script.ScriptManager import ScriptManager
|
||
from Entity.Variables import anchorList, prologueList, addModelToAnchorList, removeModelFromAnchorList
|
||
import Entity.Variables as ev
|
||
from Utils.JsonUtils import JsonUtils
|
||
|
||
app = Flask(__name__)
|
||
CORS(app)
|
||
app.config['JSON_AS_ASCII'] = False # Flask jsonify 不转义中文/emoji
|
||
app.config['JSONIFY_MIMETYPE'] = "application/json; charset=utf-8"
|
||
|
||
listData = []
|
||
listLock = threading.Lock()
|
||
|
||
dataQueue = Queue()
|
||
|
||
|
||
def start_socket_listener():
|
||
port = int(os.getenv('FLASK_COMM_PORT', 0))
|
||
LogManager.info(f"Received port from environment: {port}")
|
||
print(f"Received port from environment: {port}")
|
||
if port <= 0:
|
||
LogManager.info("未获取到通信端口，跳过Socket监听")
|
||
print("未获取到通信端口，跳过Socket监听")
|
||
return
|
||
try:
|
||
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
|
||
# 设置端口复用，避免端口被占用时无法绑定
|
||
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||
|
||
# 尝试绑定端口
|
||
try:
|
||
s.bind(('127.0.0.1', port))
|
||
print(f"[INFO] Socket successfully bound to port {port}")
|
||
LogManager.info(f"[INFO] Socket successfully bound to port {port}")
|
||
except Exception as bind_error:
|
||
print(f"[ERROR]端口绑定失败: {bind_error}")
|
||
LogManager.info(f"[ERROR]端口绑定失败: {bind_error}")
|
||
return
|
||
|
||
# 开始监听
|
||
s.listen()
|
||
LogManager.info(f"[INFO] Socket listener started on port {port}, waiting for connections...")
|
||
print(f"[INFO] Socket listener started on port {port}, waiting for connections...")
|
||
while True:
|
||
try:
|
||
conn, addr = s.accept()
|
||
except Exception as e:
|
||
LogManager.error(f"[ERROR] accept 失败: {e}")
|
||
continue
|
||
|
||
# 独立线程处理单条连接，避免单客户端异常拖垮监听线程
|
||
threading.Thread(target=_handle_conn, args=(conn, addr), daemon=True).start()
|
||
# while True:
|
||
# try:
|
||
# LogManager.info(f"[INFO] Waiting for a new connection on port {port}...")
|
||
# print(f"[INFO] Waiting for a new connection on port {port}...")
|
||
# conn, addr = s.accept()
|
||
# LogManager.info(f"[INFO] Connection accepted from: {addr}")
|
||
# print(f"[INFO] Connection accepted from: {addr}")
|
||
#
|
||
# raw_data = conn.recv(1024).decode('utf-8').strip()
|
||
# LogManager.info(f"[INFO] Raw data received: {raw_data}")
|
||
# print(f"[INFO] Raw data received: {raw_data}")
|
||
#
|
||
# data = json.loads(raw_data)
|
||
# LogManager.info(f"[INFO] Parsed data: {data}")
|
||
# print(f"[INFO] Parsed data: {data}")
|
||
# dataQueue.put(data)
|
||
# except Exception as conn_error:
|
||
# LogManager.error(f"[ERROR]连接处理失败: {conn_error}")
|
||
# print(f"[ERROR]连接处理失败: {conn_error}")
|
||
except Exception as e:
|
||
LogManager.error(f"[ERROR]Socket服务启动失败: {e}")
|
||
print(f"[ERROR]Socket服务启动失败: {e}")
|
||
|
||
def _handle_conn(conn: socket.socket, addr):
|
||
try:
|
||
with conn:
|
||
# 1. 循环收包直到拿到完整 JSON
|
||
buffer = ""
|
||
while True:
|
||
data = conn.recv(1024)
|
||
if not data: # 对端关闭
|
||
break
|
||
buffer += data.decode('utf-8', errors='ignore')
|
||
# 2. 尝试切出完整 JSON（简单按行，也可按长度头、分隔符）
|
||
while True:
|
||
line, sep, buffer = buffer.partition('\n')
|
||
if not sep: # 没找到完整行
|
||
break
|
||
line = line.strip()
|
||
if not line: # 空行跳过
|
||
continue
|
||
try:
|
||
obj = json.loads(line)
|
||
except json.JSONDecodeError as e:
|
||
LogManager.warning(f"[WARN] 非法 JSON 丢弃: {line[:100]} {e}")
|
||
continue
|
||
# 3. 收到合法数据，塞进队列
|
||
dataQueue.put(obj)
|
||
LogManager.info(f"[INFO] 收到合法消息: {obj}")
|
||
except Exception as e:
|
||
LogManager.error(f"[ERROR] 连接处理异常: {e}")
|
||
|
||
|
||
# 在独立线程中启动Socket服务
|
||
listener_thread = threading.Thread(target=start_socket_listener, daemon=True)
|
||
listener_thread.start()
|
||
|
||
|
||
# 获取设备列表
|
||
@app.route('/deviceList', methods=['GET'])
|
||
def deviceList():
|
||
try:
|
||
with listLock:
|
||
# 1. 消费完队列
|
||
while not dataQueue.empty():
|
||
obj = dataQueue.get()
|
||
if obj["type"] == 1:
|
||
# 上线：先踢掉同 deviceId 的旧记录（端口可能变）
|
||
listData[:] = [d for d in listData if d.get("deviceId") != obj.get("deviceId")]
|
||
listData.append(obj)
|
||
else:
|
||
# 下线：只要同 deviceId 就删，不管端口
|
||
listData[:] = [d for d in listData if d.get("deviceId") != obj.get("deviceId")]
|
||
|
||
# 2. 兜底：只保留在线
|
||
listData[:] = [d for d in listData if d.get('type') == 1]
|
||
|
||
return ResultData(data=listData.copy()).toJson()
|
||
except Exception as e:
|
||
LogManager.error("获取设备列表失败：", e)
|
||
return ResultData(data=[]).toJson()
|
||
|
||
|
||
# 获取设备应用列表
|
||
@app.route('/deviceAppList', methods=['POST'])
|
||
def deviceAppList():
|
||
param = request.get_json()
|
||
udid = param["udid"]
|
||
apps = ControlUtils.getDeviceAppList(udid)
|
||
return ResultData(data=apps).toJson()
|
||
|
||
|
||
# 打开指定app
|
||
@app.route('/launchApp', methods=['POST'])
|
||
def launchApp():
|
||
body = request.get_json()
|
||
udid = body.get("udid")
|
||
bundleId = body.get("bundleId")
|
||
t = tidevice.Device(udid)
|
||
t.app_start(bundleId)
|
||
return ResultData(data="").toJson()
|
||
|
||
|
||
# 回到首页
|
||
@app.route('/toHome', methods=['POST'])
|
||
def toHome():
|
||
body = request.get_json()
|
||
udid = body.get("udid")
|
||
client = wda.USBClient(udid)
|
||
client.home()
|
||
return ResultData(data="").toJson()
|
||
|
||
|
||
# 点击事件
|
||
@app.route('/tapAction', methods=['POST'])
|
||
def tapAction():
|
||
body = request.get_json()
|
||
udid = body.get("udid")
|
||
client = wda.USBClient(udid)
|
||
session = client.session()
|
||
session.appium_settings({"snapshotMaxDepth": 0})
|
||
x = body.get("x")
|
||
y = body.get("y")
|
||
session.tap(x, y)
|
||
return ResultData(data="").toJson()
|
||
|
||
|
||
# 拖拽事件
|
||
@app.route('/swipeAction', methods=['POST'])
|
||
def swipeAction():
|
||
body = request.get_json()
|
||
udid = body.get("udid")
|
||
direction = body.get("direction")
|
||
client = wda.USBClient(udid)
|
||
session = client.session()
|
||
session.appium_settings({"snapshotMaxDepth": 0})
|
||
|
||
if direction == 1:
|
||
session.swipe_up()
|
||
elif direction == 2:
|
||
session.swipe_left()
|
||
elif direction == 3:
|
||
session.swipe_down()
|
||
else:
|
||
session.swipe_right()
|
||
return ResultData(data="").toJson()
|
||
|
||
|
||
# 长按事件
|
||
@app.route('/longPressAction', methods=['POST'])
|
||
def longPressAction():
|
||
body = request.get_json()
|
||
udid = body.get("udid")
|
||
x = body.get("x")
|
||
y = body.get("y")
|
||
client = wda.USBClient(udid)
|
||
session = client.session()
|
||
session.appium_settings({"snapshotMaxDepth": 5})
|
||
session.tap_hold(x, y, 1.0)
|
||
return ResultData(data="").toJson()
|
||
|
||
|
||
# 养号
|
||
@app.route('/growAccount', methods=['POST'])
|
||
def growAccount():
|
||
body = request.get_json()
|
||
udid = body.get("udid")
|
||
|
||
manager = ScriptManager()
|
||
event = threading.Event()
|
||
# 启动脚本
|
||
thread = threading.Thread(target=manager.growAccount, args=(udid, event))
|
||
thread.start()
|
||
# 添加到线程管理
|
||
ThreadManager.add(udid, thread, event)
|
||
return ResultData(data="").toJson()
|
||
|
||
|
||
# 观看直播
|
||
@app.route("/watchLiveForGrowth", methods=['POST'])
|
||
def watchLiveForGrowth():
|
||
body = request.get_json()
|
||
udid = body.get("udid")
|
||
manager = ScriptManager()
|
||
event = threading.Event()
|
||
thread = threading.Thread(target=manager.watchLiveForGrowth, args=(udid, event))
|
||
thread.start()
|
||
# 添加到线程管理
|
||
ThreadManager.add(udid, thread, event)
|
||
return ResultData(data="").toJson()
|
||
|
||
|
||
# 停止脚本
|
||
@app.route("/stopScript", methods=['POST'])
|
||
def stopScript():
|
||
body = request.get_json()
|
||
udid = body.get("udid")
|
||
LogManager.method_info(f"接口收到 /stopScript udid={udid}", method="task")
|
||
code, msg = ThreadManager.stop(udid)
|
||
return ResultData(code=code, data="", massage=msg).toJson()
|
||
|
||
|
||
# 关注打招呼
|
||
@app.route('/passAnchorData', methods=['POST'])
|
||
def passAnchorData():
|
||
try:
|
||
data: Dict[str, Any] = request.get_json()
|
||
# 设备列表
|
||
idList = data.get("deviceList", [])
|
||
# 主播列表
|
||
acList = data.get("anchorList", [])
|
||
AiUtils.save_aclist_flat_append(acList)
|
||
|
||
# 是否需要回复
|
||
needReply = data.get("needReply", True)
|
||
# 获取打招呼数据
|
||
ev.prologueList = data.get("prologueList", [])
|
||
|
||
# 添加主播数据
|
||
addModelToAnchorList(acList)
|
||
# 启动线程，执行脚本
|
||
for udid in idList:
|
||
manager = ScriptManager()
|
||
event = threading.Event()
|
||
# 启动脚本
|
||
thread = threading.Thread(target=manager.safe_greetNewFollowers, args=(udid, needReply, event))
|
||
thread.start()
|
||
# 添加到线程管理
|
||
ThreadManager.add(udid, thread, event)
|
||
return ResultData(data="").toJson()
|
||
except Exception as e:
|
||
LogManager.error(e)
|
||
|
||
|
||
# 获取私信数据
|
||
@app.route("/getPrologueList", methods=['GET'])
|
||
def getPrologueList():
|
||
import Entity.Variables as Variables
|
||
return ResultData(data=Variables.prologueList).toJson()
|
||
|
||
|
||
# 添加临时数据
|
||
# 批量追加主播到 JSON 文件
|
||
@app.route("/addTempAnchorData", methods=['POST'])
|
||
def addTempAnchorData():
|
||
"""
|
||
请求体支持：
|
||
- 单个对象：{"anchorId": "xxx", "country": "CN"}
|
||
- 对象数组：[{"anchorId": "xxx", "country": "CN"}, {"anchorId": "yyy", "country": "US"}]
|
||
"""
|
||
data = request.get_json()
|
||
if not data:
|
||
return ResultData(code=400, massage="请求数据为空").toJson()
|
||
# 追加到 JSON 文件
|
||
AiUtils.save_aclist_flat_append(data, "log/acList.json")
|
||
return ResultData(data="ok").toJson()
|
||
|
||
|
||
# 获取当前屏幕上的聊天信息
|
||
@app.route("/getChatTextInfo", methods=['POST'])
|
||
def getChatTextInfo():
|
||
data = request.get_json()
|
||
udid = data.get("udid")
|
||
client = wda.USBClient(udid)
|
||
session = client.session()
|
||
xml = session.source()
|
||
try:
|
||
result = AiUtils.extract_messages_from_xml(xml)
|
||
print(result)
|
||
return ResultData(data=result).toJson()
|
||
except Exception as e:
|
||
|
||
LogManager.error(f"获取屏幕翻译出现错误：{e}", "获取屏幕翻译")
|
||
|
||
data = [
|
||
{
|
||
'type': 'massage',
|
||
'dir': 'in',
|
||
'text': '当前页面无法获取聊天记录，请在tiktok聊天页面进行获取！！！'
|
||
},
|
||
{
|
||
'type': 'massage',
|
||
'dir': 'in',
|
||
'text': 'Unable to retrieve chat messages on the current screen. Please navigate to the TikTok chat page and try again!!!'
|
||
}
|
||
]
|
||
return ResultData(data=data, massage="解析失败").toJson()
|
||
|
||
|
||
# 监控消息
|
||
@app.route("/replyMessages", methods=['POST'])
|
||
def monitorMessages():
|
||
body = request.get_json()
|
||
udid = body.get("udid")
|
||
manager = ScriptManager()
|
||
event = threading.Event()
|
||
thread = threading.Thread(target=manager.replyMessages, args=(udid, event))
|
||
thread.start()
|
||
# 添加到线程管理
|
||
ThreadManager.add(udid, thread, event)
|
||
return ResultData(data="").toJson()
|
||
|
||
|
||
@app.route("/setLoginInfo", methods=['POST'])
|
||
def upLoadLogLogs():
|
||
data = request.get_json() # 解析 JSON
|
||
token = data.get("token")
|
||
userId = data.get("userId")
|
||
tenantId = data.get("tenantId")
|
||
ok = LogManager.upload_all_logs("http://47.79.98.113:8101/api/log/upload", token, userId, tenantId)
|
||
if ok:
|
||
return ResultData(data="日志上传成功").toJson()
|
||
else:
|
||
return ResultData(data="", massage="日志上传失败").toJson()
|
||
|
||
|
||
# 获取当前的主播列表数据
|
||
@app.route("/anchorList", methods=['POST'])
|
||
def queryAnchorList():
|
||
file_path = "log/acList.json"
|
||
data = []
|
||
if Path(file_path).exists():
|
||
try:
|
||
with open(file_path, "r", encoding="utf-8") as f:
|
||
data = json.load(f)
|
||
except Exception as e:
|
||
LogManager.error(f"[anchorList] 读取失败: {e}")
|
||
data = []
|
||
return ResultData(data=data).toJson()
|
||
|
||
|
||
# 删除主播
|
||
@app.route("/deleteAnchorWithIds", methods=['POST'])
|
||
def deleteAnchorWithIds():
|
||
ls: list[dict] = request.get_json() # [{"anchorId": "xxx"}, ...]
|
||
ids = [d.get("anchorId") for d in ls if d.get("anchorId")]
|
||
deleted = AiUtils.delete_anchors_by_ids(ids)
|
||
return ResultData(data={"deleted": deleted}).toJson()
|
||
|
||
|
||
@app.route("/aiConfig", methods=['POST'])
|
||
def aiConfig():
|
||
data = request.get_json()
|
||
agentName = data.get("agentName")
|
||
guildName = data.get("guildName")
|
||
contactTool = data.get("contactTool")
|
||
contact = data.get("contact")
|
||
|
||
dict = {
|
||
"agentName": agentName,
|
||
"guildName": guildName,
|
||
"contactTool": contactTool,
|
||
"contact": contact
|
||
}
|
||
|
||
JsonUtils.write_json("aiConfig", dict)
|
||
return ResultData(data="").toJson()
|
||
|
||
|
||
# 查询主播聊天发送的最后一条信息
|
||
@app.route("/select_last_message", methods=['GET'])
|
||
def select_last_message():
|
||
data = JsonUtils.query_all_json_items()
|
||
|
||
return ResultData(data=data).toJson()
|
||
|
||
|
||
@app.route("/update_last_message", methods=['POST'])
|
||
def update_last_message():
|
||
data = request.get_json() # 解析 JSON
|
||
sender = data.get("sender")
|
||
udid = data.get("device")
|
||
text = data.get("text")
|
||
|
||
updated_count = JsonUtils.update_json_items(
|
||
match={"sender": sender, "text": text}, # 匹配条件
|
||
patch={"status": 1}, # 修改内容
|
||
filename="log/last_message.json", # 要修改的文件
|
||
multi=False # 只改第一条匹配的
|
||
)
|
||
if updated_count > 0:
|
||
return ResultData(data=updated_count, massage="修改成功").toJson()
|
||
return ResultData(data=updated_count, massage="修改失败").toJson()
|
||
|
||
|
||
@app.route("/delete_last_message", methods=['POST'])
|
||
def delete_last_message():
|
||
data = request.get_json() # 解析 JSON
|
||
sender = data.get("sender")
|
||
udid = data.get("device")
|
||
text = data.get("text")
|
||
|
||
updated_count = JsonUtils.delete_json_items(
|
||
match={"sender": sender, "text": text}, # 匹配条件
|
||
filename="log/last_message.json", # 要修改的文件
|
||
multi=False # 只改第一条匹配的
|
||
)
|
||
if updated_count > 0:
|
||
return ResultData(data=updated_count, massage="修改成功").toJson()
|
||
return ResultData(data=updated_count, massage="修改失败").toJson()
|
||
|
||
|
||
# @app.route("/killWda", methods=['POST'])
|
||
# def killWda():
|
||
# data = request.get_json() # 解析 JSON
|
||
# udid = data.get("device")
|
||
# print(udid)
|
||
#
|
||
#
|
||
# AiUtils.kill_wda(udid)
|
||
# time.sleep(10)
|
||
# AiUtils.launch_wda(udid)
|
||
#
|
||
# return ResultData(data="", msg="WDA重新启动").toJson()
|
||
|
||
|
||
if __name__ == '__main__':
|
||
app.run("0.0.0.0", port=5000, debug=True, use_reloader=False)
|