ehentai-fastapi/main.py

from fastapi import FastAPI, Request, Form
from fastapi.staticfiles import StaticFiles
from fastapi.templating import Jinja2Templates
from fastapi.responses import JSONResponse
import uvicorn
import os
from pydantic import BaseModel
from utils import *

app = FastAPI(title="下载工具", version="1.0.0")

# 在应用启动时检查并创建data文件夹和targets.txt
@app.on_event("startup")
async def startup_event():
    # 检查并创建data文件夹
    data_dir = "data"
    if not os.path.exists(data_dir):
        os.makedirs(data_dir)
        print(f"创建目录: {data_dir}")
    
    # 检查并创建targets.txt文件
    targets_file = os.path.join(data_dir, "targets.txt")
    if not os.path.exists(targets_file):
        with open(targets_file, 'w', encoding='utf-8') as f:
            f.write("# 在这里添加目标URL，每行一个\n")
            f.write("# 示例:\n")
            f.write("# https://example.com/file1.zip\n")
            f.write("# https://example.com/image.jpg\n")
        print(f"创建文件: {targets_file}")
    else:
        print(f"文件已存在: {targets_file}")

# 挂载静态文件和模板
app.mount("/static", StaticFiles(directory="static"), name="static")
templates = Jinja2Templates(directory="templates")

@app.get("/")
async def home(request: Request):
    """主页面"""
    return templates.TemplateResponse("index.html", {"request": request})

@app.post("/load_urls")
async def load_urls():
    """读取 targets.txt 文件中的URL"""
    try:
        file_path = "data/targets.txt"
        
        # 读取文件内容
        with open(file_path, 'r', encoding='utf-8') as f:
            urls = [line.strip() for line in f.readlines() if line.strip()]
        
        # 过滤掉空行和注释行（以#开头的行）
        urls = [url for url in urls if url and not url.startswith('#')]
        
        if not urls:
            return JSONResponse({
                "success": True,
                "message": "targets.txt 文件为空，请在data/targets.txt中添加URL",
                "urls": []
            })
        
        return JSONResponse({
            "success": True,
            "message": f"成功读取 {len(urls)} 个URL",
            "urls": urls
        })
        
    except Exception as e:
        return JSONResponse({
            "success": False,
            "message": f"读取文件时出错: {str(e)}",
            "urls": []
        })

@app.post("/clear")
async def clear_output():
    """清除输出"""
    return JSONResponse({
        "success": True,
        "message": "输出已清除",
        "output": ""
    })

class ProxyRequest(BaseModel):
    ip: str
    port: str

@app.post("/download_urls")
async def download_urls(req: ProxyRequest):
    proxy = f"http://{req.ip}:{req.port}"
    msg = await run_step1(proxy)
    return JSONResponse({"success": True, "message": msg})

@app.post("/download_images")
async def download_images(req: ProxyRequest):
    proxy = f"http://{req.ip}:{req.port}"
    msg = await run_step2(proxy)
    return JSONResponse({"success": True, "message": msg})

if __name__ == "__main__":
    uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=True)