#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright (C) 2026 tmwgsicp
# Licensed under the GNU Affero General Public License v3.0
# See LICENSE file in the project root for full license text.
# SPDX-License-Identifier: AGPL-3.0-only
"""
微信公众号文章API服务 - FastAPI版本
主应用文件
"""
from contextlib import asynccontextmanager
from dotenv import load_dotenv
load_dotenv()
from fastapi import FastAPI
from fastapi.staticfiles import StaticFiles
from fastapi.responses import FileResponse, HTMLResponse
from fastapi.middleware.cors import CORSMiddleware
from pathlib import Path
# 导入路由
from routes import article, articles, search, admin, login, image, health, stats, rss
from utils.rss_store import init_db
from utils.rss_poller import rss_poller
API_DESCRIPTION = """
微信公众号文章下载 API，支持文章解析、公众号搜索、文章列表获取等功能。
## 快速开始
1. 访问 `/login.html` 扫码登录微信公众号后台
2. 调用 `GET /api/public/searchbiz?query=公众号名称` 搜索目标公众号
3. 从返回结果中取 `fakeid`，调用 `GET /api/public/articles?fakeid=xxx` 获取文章列表
4. 对每篇文章调用 `POST /api/article` 获取完整内容
## 认证说明
所有核心接口都需要先登录。登录后凭证自动保存到 `.env` 文件，服务重启后无需重新登录（有效期约 4 天）。
"""
@asynccontextmanager
async def lifespan(app: FastAPI):
"""应用生命周期：启动和关闭"""
env_file = Path(__file__).parent / ".env"
if not env_file.exists():
print("\n" + "=" * 60)
print("[WARNING] .env file not found")
print("=" * 60)
print("Please configure .env file or login via admin page")
print("Visit: http://localhost:5000/admin.html")
print("=" * 60 + "\n")
else:
print("\n" + "=" * 60)
print("[OK] .env file loaded")
print("=" * 60 + "\n")
init_db()
await rss_poller.start()
yield
await rss_poller.stop()
app = FastAPI(
title="WeChat Download API",
description=API_DESCRIPTION,
version="1.0.0",
docs_url="/api/docs",
redoc_url=None,
openapi_url="/api/openapi.json",
license_info={
"name": "AGPL-3.0",
"url": "https://www.gnu.org/licenses/agpl-3.0.html",
},
lifespan=lifespan,
)
# CORS配置
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
# 注册路由（注意：articles.router 必须在 search.router 之前注册，避免路由冲突）
app.include_router(health.router, prefix="/api", tags=["健康检查"])
app.include_router(stats.router, prefix="/api", tags=["统计信息"])
app.include_router(article.router, prefix="/api", tags=["文章内容"])
app.include_router(articles.router, prefix="/api/public", tags=["文章列表"]) # 必须先注册
app.include_router(search.router, prefix="/api/public", tags=["公众号搜索"]) # 后注册
app.include_router(admin.router, prefix="/api/admin", tags=["管理"])
app.include_router(login.router, prefix="/api/login", tags=["登录"])
app.include_router(image.router, prefix="/api", tags=["图片代理"])
app.include_router(rss.router, prefix="/api", tags=["RSS 订阅"])
# 静态文件
static_dir = Path(__file__).parent / "static"
if static_dir.exists():
app.mount("/static", StaticFiles(directory=str(static_dir)), name="static")
@app.get("/api/redoc", include_in_schema=False)
async def redoc_html():
"""ReDoc 文档（使用 cdnjs 加速）"""
return HTMLResponse("""
WeChat Download API - ReDoc
""")
# 静态页面路由
@app.get("/", include_in_schema=False)
async def root():
"""首页 - 重定向到管理页面"""
return FileResponse(static_dir / "admin.html")
@app.get("/admin.html", include_in_schema=False)
async def admin_page():
"""管理页面"""
return FileResponse(static_dir / "admin.html")
@app.get("/login.html", include_in_schema=False)
async def login_page():
"""登录页面"""
return FileResponse(static_dir / "login.html")
@app.get("/verify.html", include_in_schema=False)
async def verify_page():
"""验证页面"""
return FileResponse(static_dir / "verify.html")
@app.get("/rss.html", include_in_schema=False)
async def rss_page():
"""RSS 订阅管理页面"""
return FileResponse(static_dir / "rss.html")
if __name__ == "__main__":
import os
import uvicorn
from dotenv import load_dotenv
load_dotenv()
host = os.getenv("HOST", "0.0.0.0")
port = int(os.getenv("PORT", "5000"))
debug = os.getenv("DEBUG", "false").lower() in ("true", "1", "yes")
print("=" * 60)
print("Wechat Article API Service - FastAPI Version")
print("=" * 60)
print(f"Admin Page: http://localhost:{port}/admin.html")
print(f"API Docs: http://localhost:{port}/api/docs")
print(f"ReDoc Docs: http://localhost:{port}/api/redoc")
print("First time? Please login via admin page")
print("=" * 60)
uvicorn.run(
"app:app",
host=host,
port=port,
reload=debug,
log_level="debug" if debug else "info",
)