You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
1.0 KiB
35 lines
1.0 KiB
#!/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}"
|
|
|