main.py

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from fastapi.staticfiles import StaticFiles
from fastapi.responses import FileResponse
from routers import auth, company, categories, products, product_certifications, product_media, product_packaging, membership, operation_log, contact, template, customer_relationship, member_level, r2_config, media, statistics
from database import engine, Base, get_db
from sqlalchemy.orm import Session
import json
import os
from pathlib import Path
from dotenv import load_dotenv
from datetime import datetime, timedelta

# 导入中间件
from middleware.trace import TraceMiddleware

# 导入日志模块
from utils.logger import logger

# 加载 .env 文件
load_dotenv()

# 导入模板配置
from config.templates import DEFAULT_TEMPLATES
# 导入会员等级配置
from config.member_levels import DEFAULT_MEMBER_LEVELS
# 导入默认公司和用户配置
from config.default_companies import DEFAULT_COMPANIES, DEFAULT_USERS

# 导入所有模型，确保它们被注册到Base
from models.user import User
from models.company import Company
from models.category import Category
from models.product import Product
from models.membership import Membership
from models.operation_log import OperationLog
from models.contact import Contact
from models.template import Template
from models.customer_relationship import CustomerRelationship
from models.member_level import MemberLevel
from models.company_r2_config import CompanyR2Config
from models.media import Media, MediaTag, MediaTagMapping, MediaDirectory
from models.site_visit import SiteVisit
from utils.auth import get_password_hash

# 数据库表创建已在database.py中处理

app = FastAPI(
    title="外贸独立站 API",
    description="用于外贸独立站的后端 API",
    version="1.0.0"
)

# 配置 CORS
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],  # 在生产环境中应该设置具体的前端域名
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# 添加 TraceMiddleware 中间件
app.add_middleware(TraceMiddleware)

# 挂载静态文件目录（用于访问上传的图片）
BASE_DIR = Path(__file__).parent
IMAGES_DIR = BASE_DIR / "images"
if IMAGES_DIR.exists():
    app.mount("/images", StaticFiles(directory=str(IMAGES_DIR)), name="images")

# 注册路由
# 注意：更具体的路由（如 /{product_number}/media）必须在较不具体的路由（如 /{product_number}）之前注册
app.include_router(auth.router, prefix="/api/auth", tags=["认证"])
app.include_router(product_media.router, prefix="/api/products", tags=["产品媒体"])
app.include_router(product_certifications.router, prefix="/api/products", tags=["产品认证"])
app.include_router(product_packaging.router, prefix="/api/products", tags=["产品包装"])
app.include_router(products.router, prefix="/api/products", tags=["产品"])
app.include_router(company.router, prefix="/api/company", tags=["公司信息"])
app.include_router(categories.router, prefix="/api/categories", tags=["类目"])
app.include_router(membership.router, prefix="/api/membership", tags=["会员充值"])
app.include_router(operation_log.router, prefix="/api/operation-logs", tags=["操作日志"])
app.include_router(contact.router, prefix="/api/contact", tags=["联系信息管理"])
app.include_router(template.router, prefix="/api/templates", tags=["模板管理"])
app.include_router(customer_relationship.router, prefix="/api/customer-relationships", tags=["客户关系管理"])
app.include_router(member_level.router, prefix="/api/member-levels", tags=["会员等级管理"])
app.include_router(r2_config.router, prefix="/api/r2-config", tags=["R2 配置管理"])
app.include_router(media.router, prefix="/api/media", tags=["媒体素材管理"])
app.include_router(statistics.router, prefix="/api/statistics", tags=["统计指标"])

# 初始化超级管理员用户和默认类目
@app.on_event("startup")
def startup_event():
    try:
        db = next(get_db())
        # 检查是否已有超级管理员
        superadmin = db.query(User).filter(User.is_superadmin == True).first()
        if not superadmin:
            # 创建超级管理员
            superadmin = User(
                username="admin",
                email="admin@mail.yomaton.com",
                password=get_password_hash("admin00"),
                is_superadmin=True,
                company_code="0000"
            )
            db.add(superadmin)
            db.commit()
            logger.info("超级管理员已创建: username=admin, password=admin00")
        
        
        # 创建默认模板
        for template_data in DEFAULT_TEMPLATES:
            # 转换features和regions列表为JSON字符串
            template_data["features"] = json.dumps(template_data["features"], ensure_ascii=False)
            template_data["regions"] = json.dumps(template_data["regions"], ensure_ascii=False)
            
            existing_template = db.query(Template).filter(Template.key == template_data["key"]).first()
            if not existing_template:
                new_template = Template(**template_data)
                db.add(new_template)
        db.commit()
        logger.info("默认模板已创建")
        
        # 创建默认会员等级
        for level_data in DEFAULT_MEMBER_LEVELS:
            existing_level = db.query(MemberLevel).filter(MemberLevel.level == level_data["level"]).first()
            if not existing_level:
                new_level = MemberLevel(**level_data)
                db.add(new_level)
        db.commit()
        logger.info("默认会员等级已创建")
        
        # 创建默认公司
        for company_data in DEFAULT_COMPANIES:
            existing_company = db.query(Company).filter(Company.company_code == company_data["company_code"]).first()
            if not existing_company:
                new_company = Company(**company_data)
                db.add(new_company)
        db.commit()
        logger.info("默认公司已创建")
        
        # 创建默认用户和会员信息
        for user_data in DEFAULT_USERS:
            existing_user = db.query(User).filter(User.username == user_data["username"]).first()
            if not existing_user:
                user_data_copy = user_data.copy()
                user_data_copy["password"] = get_password_hash(user_data_copy["password"])
                new_user = User(**user_data_copy)
                db.add(new_user)
                db.flush()
                
                # 为用户创建默认会员信息
                existing_membership = db.query(Membership).filter(Membership.company_code == user_data["company_code"]).first()
                if not existing_membership:
                    new_membership = Membership(
                        company_code = user_data["company_code"],
                        level = 0,  # 默认普通会员
                        start_date = datetime.now(),
                        end_date = datetime.now() + timedelta(days=365)
                    )
                    db.add(new_membership)
        db.commit()
        logger.info("默认用户和会员信息已创建")
    except Exception as e:
        logger.error(f"初始化数据时出错: {e}")

@app.get("/")
def read_root():
    logger.info("访问根路径")
    return {"message": "欢迎使用外贸独立站 API"}

@app.get("/health")
def health_check():
    logger.info("健康检查")
    return {"status": "healthy"}

if __name__ == "__main__":
    import uvicorn
    logger.info("启动应用服务器")
    uvicorn.run(app, host="0.0.0.0", port=7860)