From 553f0f6231cc90d010ff86b99e855a3ed28958cf Mon Sep 17 00:00:00 2001
From: bvwl <2201101122@qq.com>
Date: Thu, 6 Nov 2025 11:30:31 +0800
Subject: [PATCH] 0.0.4

---
 README.md |  2 ++
 main.py   | 39 ++++++++++++++++++++++++++++++++-------
 2 files changed, 34 insertions(+), 7 deletions(-)

diff --git a/README.md b/README.md
index 1ce5f7e..4db6f8b 100644
--- a/README.md
+++ b/README.md
@@ -1,3 +1,5 @@
+## 0.0.4
+- 优化测试代码
 ## 0.0.3
 - 移除.idea
 ## 0.0.2
diff --git a/main.py b/main.py
index 772d0cb..b358fd3 100644
--- a/main.py
+++ b/main.py
@@ -1,22 +1,33 @@
+import random
 from re import T
 from DrissionPage import Chromium
+import requests
 
 # 发送请求告知服务器执行完成
-def send_complete():
+def task_push(project_name: str, browser_id: str, status: int = 200):
     print('发送信息告知服务器脚本执行完成')
-    url = ''
+    url = 'http://127.0.0.1:6060/task/info/push'
+    payload = {
+        "project_name": project_name,
+        "browser_id": browser_id,
+        "status": status
+    }
+    res = requests.post(url, json=payload).json()
+    print(res)
 
 
-def main(http: str):
+def main(http: str, browser_id: str):
     """
     使用给定的 http 句柄接管已打开的指纹浏览器并执行示例登录流程
 
     Args:
         http (str): 浏览器的 http 句柄字符串
+        browser_id (str): 浏览器 ID
 
     Returns:
         None
     """
+    project_name = 'auto_test'
     # 启动或接管浏览器，并获取标签页对象
     print(f'使用 http 接管浏览器: {http}')
     tab = Chromium(http).new_tab()
@@ -35,13 +46,18 @@ def main(http: str):
     #     print(1)
     #     time.sleep(5)
 
+    time.sleep(random.randint(5, 10))
+    task_push(project_name, browser_id)
+
 
 if __name__ == '__main__':
     """
-    脚本入口：支持从命令行参数或环境变量读取 http 并调用 main
+    脚本入口：支持从命令行参数或环境变量读取 http 和 browser_id 并调用 main
 
     - 命令行：python project/test/log.py <http>
-    - 环境变量：BROWSER_HTTP=<http> python project/test/log.py
+    - 环境变量：
+      - BROWSER_HTTP=<http> python project/test/log.py
+      - BROWSER_ID=<browser_id> python project/test/log.py (可选，用于提交项目状态)
     """
     import os
     import sys
@@ -52,11 +68,20 @@ if __name__ == '__main__':
         print('缺少 http 参数：请通过命令行传入或设置环境变量 BROWSER_HTTP')
         sys.exit(1)
 
+    # 获取浏览器ID（用于提交项目状态）
+    browser_id = os.getenv('BROWSER_ID') or (sys.argv[2] if len(sys.argv) > 2 else None)
+    if browser_id:
+        print(f'接收到 browser_id: {browser_id}')
+
     print(f'接收到 http: {http}')
     try:
-        main(http)
+        print(f'http --> {http}')
+        print(f'browser_id --> {browser_id}')
+        main(http, browser_id)
         print('脚本执行完成')
+        # 这里可以使用 browser_id 来提交项目状态
+        # if browser_id:
+        #     send_complete(browser_id)
     except Exception as e:
         print(f'脚本执行异常: {e}')
         sys.exit(2)
-