80 lines
2.5 KiB
Python
80 lines
2.5 KiB
Python
from fastapi import FastAPI
|
||
from fastapi.staticfiles import StaticFiles
|
||
from fastapi.middleware.cors import CORSMiddleware
|
||
from app.core import config
|
||
from app.api import materials, videos, publish, login_helper, auth, admin
|
||
from loguru import logger
|
||
import os
|
||
|
||
settings = config.settings
|
||
|
||
app = FastAPI(title="ViGent TalkingHead Agent")
|
||
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=["*"],
|
||
allow_credentials=True,
|
||
allow_methods=["*"],
|
||
allow_headers=["*"],
|
||
)
|
||
|
||
# Create dirs
|
||
settings.UPLOAD_DIR.mkdir(parents=True, exist_ok=True)
|
||
settings.OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
|
||
(settings.UPLOAD_DIR / "materials").mkdir(exist_ok=True)
|
||
|
||
app.mount("/outputs", StaticFiles(directory=str(settings.OUTPUT_DIR)), name="outputs")
|
||
|
||
# 注册路由
|
||
app.include_router(materials.router, prefix="/api/materials", tags=["Materials"])
|
||
app.include_router(videos.router, prefix="/api/videos", tags=["Videos"])
|
||
app.include_router(publish.router, prefix="/api/publish", tags=["Publish"])
|
||
app.include_router(login_helper.router, prefix="/api", tags=["LoginHelper"])
|
||
app.include_router(auth.router) # /api/auth
|
||
app.include_router(admin.router) # /api/admin
|
||
|
||
|
||
@app.on_event("startup")
|
||
async def init_admin():
|
||
"""
|
||
服务启动时初始化管理员账号
|
||
"""
|
||
admin_email = settings.ADMIN_EMAIL
|
||
admin_password = settings.ADMIN_PASSWORD
|
||
|
||
if not admin_email or not admin_password:
|
||
logger.warning("未配置 ADMIN_EMAIL 和 ADMIN_PASSWORD，跳过管理员初始化")
|
||
return
|
||
|
||
try:
|
||
from app.core.supabase import get_supabase
|
||
from app.core.security import get_password_hash
|
||
|
||
supabase = get_supabase()
|
||
|
||
# 检查是否已存在
|
||
existing = supabase.table("users").select("id").eq("email", admin_email).execute()
|
||
|
||
if existing.data:
|
||
logger.info(f"管理员账号已存在: {admin_email}")
|
||
return
|
||
|
||
# 创建管理员
|
||
supabase.table("users").insert({
|
||
"email": admin_email,
|
||
"password_hash": get_password_hash(admin_password),
|
||
"username": "Admin",
|
||
"role": "admin",
|
||
"is_active": True,
|
||
"expires_at": None # 永不过期
|
||
}).execute()
|
||
|
||
logger.success(f"管理员账号已创建: {admin_email}")
|
||
except Exception as e:
|
||
logger.error(f"初始化管理员失败: {e}")
|
||
|
||
|
||
@app.get("/health")
|
||
def health():
|
||
return {"status": "ok"}
|