ehentai-fastapi/utils.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
工具函数模块
"""
from typing import Optional

from logger import get_logger
from step1 import main as step1_main
from step2 import main as step2_main

# 设置日志
logger = get_logger("utils")

async def run_step1(proxy: Optional[str] = None) -> str:
    """执行第一步：抓取画廊链接"""
    try:
        logger.info("开始执行画廊链接抓取")
        await step1_main(proxy)
        logger.info("画廊链接抓取完成")
        return "画廊链接抓取完成！"
    except Exception as e:
        logger.exception("step1 执行失败")
        return f"抓取失败：{e}"

async def run_step2(proxy: Optional[str] = None) -> str:
    """执行第二步：下载图片"""
    try:
        logger.info("开始执行图片下载")
        await step2_main(proxy)
        logger.info("图片下载完成")
        return "图片下载完成！"
    except Exception as e:
        logger.exception("step2 执行失败")
        return f"下载失败：{e}"