临时提交

Flask/FlaskService.py

@@ -5,6 +5,8 @@ import threading
 import warnings
 from queue import Queue
 from typing import Any, Dict
+from Utils.AiUtils import AiUtils
+from Utils.Requester import Requester
 
 import tidevice
 import wda
@@ -72,6 +74,11 @@ listener_thread.start()
 def passToken():
     data = request.get_json()
     accountToken = data['token']
+    print(accountToken)
+
+    Requester.requestComments()
+
+
     return ResultData(data="").toJson()
 
 # 获取设备列表
@@ -230,5 +237,16 @@ def addTempAnchorData():
     addModelToAnchorList(data)
     return ResultData(data="").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()
+    result = AiUtils.extract_messages_from_xml(xml)
+    return ResultData(data=result).toJson()
+
 if __name__ == '__main__':
     app.run("0.0.0.0", port=5000, debug=True, use_reloader=False)