wiki_crawler/nodes/iteration_output.py

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(iteration_output: list[dict], status_code: float, body: str):
    '''
    迭代器运行结果记录
    iteration_output: 迭代器运行结果记录
    status_code: 从URL获取的响应状态码
    body: 从URL获取的响应体
    '''
    try:
        check_status(status_code, body)
        data = body["data"]
        iteration_output.append(data)
    except Exception as e:
        raise e
    
    return {
        "iteration_output": iteration_output,
    }