aaaa

nodes/parse_pending_urls.py

@@ -1,19 +1,27 @@
-def check_status(status_code: float, body: str):
+import json
+
+def parse_response(status_code: float, body: str):
     '''
     检查状态码和约定的返回值
+    并且返回正确的body
     '''
     if status_code != 200:
         raise Exception(f"注册任务失败，状态码：{status_code}")
-    if "code" not in body or body["code"] != 1:
+    
+    data = json.loads(body)
+
+    if "code" not in data or data["code"] != 1:
         raise Exception(f"注册任务失败，返回值：{body}")
     
+    return data["data"]
+    
 def main(status_code: float, body: str):
     try:
-        check_status(status_code, body)
+        data = parse_response(status_code, body)
     except Exception as e:
         raise e
     
-    urls = body["data"]["urls"]
+    urls = data["urls"]
 
     return {
         "urls": urls,