From 13428ac5521ff72138d98dfee236b9deb310a27a Mon Sep 17 00:00:00 2001 From: QingGang Date: Tue, 23 Dec 2025 01:55:59 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E8=BE=93=E5=87=BA=E8=8A=82?= =?UTF-8?q?=E7=82=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- nodes/output.py | 14 ++++++++++++++ nodes/parse_save_urls.py | 20 ++++++++++++++++++++ 2 files changed, 34 insertions(+) create mode 100644 nodes/output.py create mode 100644 nodes/parse_save_urls.py diff --git a/nodes/output.py b/nodes/output.py new file mode 100644 index 0000000..96600c7 --- /dev/null +++ b/nodes/output.py @@ -0,0 +1,14 @@ + +def main(task_id: float, is_new_task, add_urls_result: list[dict], urls: list[str], output: list[dict]): + + # 组建返回结果 + output = { + "task_id": task_id, + "is_new_task": is_new_task, + "add_urls_result": add_urls_result, + "process_urls": urls, + "iteration_result": output, + } + return { + "output": output, + } \ No newline at end of file diff --git a/nodes/parse_save_urls.py b/nodes/parse_save_urls.py new file mode 100644 index 0000000..00968ab --- /dev/null +++ b/nodes/parse_save_urls.py @@ -0,0 +1,20 @@ +def check_status(status_code: float, body: str): + ''' + 检查状态码和约定的返回值 + ''' + if status_code != 200: + raise Exception(f"注册任务失败,状态码:{status_code}") + if "code" not in body or body["code"] != 1: + raise Exception(f"注册任务失败,返回值:{body}") + +def main(status_code: float, body: str): + try: + check_status(status_code, body) + except Exception as e: + raise e + + urls_result = body["data"] + + return { + "add_urls_result": urls_result + }