fix:优化数据

This commit is contained in:
丹尼尔
2026-03-15 16:38:59 +08:00
parent a609f81a36
commit 3aa1a586e5
43 changed files with 14565 additions and 294 deletions

View File

@@ -6,7 +6,19 @@ from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from fastapi.staticfiles import StaticFiles
from backend.app.routers import customers, projects, finance
from backend.app.routers import (
customers,
projects,
finance,
settings,
ai_settings,
email_configs,
cloud_docs,
cloud_doc_config,
portal_links,
)
from backend.app.db import Base, engine
from backend.app import models # noqa: F401 - ensure models are imported
def create_app() -> FastAPI:
@@ -16,6 +28,35 @@ def create_app() -> FastAPI:
version="0.1.0",
)
# Ensure database tables exist (especially when running in Docker)
@app.on_event("startup")
def on_startup() -> None:
Base.metadata.create_all(bind=engine)
# Add new columns to finance_records if they don't exist (Module 6)
try:
from sqlalchemy import text
with engine.connect() as conn:
r = conn.execute(text("PRAGMA table_info(finance_records)"))
cols = [row[1] for row in r]
if "amount" not in cols:
conn.execute(text("ALTER TABLE finance_records ADD COLUMN amount NUMERIC(12,2)"))
if "billing_date" not in cols:
conn.execute(text("ALTER TABLE finance_records ADD COLUMN billing_date DATE"))
conn.commit()
except Exception:
pass
# Add customers.tags if missing (customer tags for project 收纳)
try:
from sqlalchemy import text
with engine.connect() as conn:
r = conn.execute(text("PRAGMA table_info(customers)"))
cols = [row[1] for row in r]
if "tags" not in cols:
conn.execute(text("ALTER TABLE customers ADD COLUMN tags VARCHAR(512)"))
conn.commit()
except Exception:
pass
# CORS
raw_origins = os.getenv("CORS_ORIGINS")
if raw_origins:
@@ -35,6 +76,12 @@ def create_app() -> FastAPI:
app.include_router(customers.router)
app.include_router(projects.router)
app.include_router(finance.router)
app.include_router(settings.router)
app.include_router(ai_settings.router)
app.include_router(email_configs.router)
app.include_router(cloud_docs.router)
app.include_router(cloud_doc_config.router)
app.include_router(portal_links.router)
# Static data mount (for quotes, contracts, finance archives, etc.)
data_dir = Path("data")