Initial commit

backend/app/main.py

@@ -0,0 +1,107 @@
+from __future__ import annotations
+
+from contextlib import asynccontextmanager
+
+from fastapi import FastAPI
+from fastapi import Query
+from fastapi.middleware.cors import CORSMiddleware
+
+from app.models import (
+    Account,
+    ClsNewsDocument,
+    DailyInputDocument,
+    DailyInputUpsertPayload,
+    ReportDocument,
+    ReportListItem,
+)
+from app.services.domain import (
+    get_accounts,
+    get_cls_news,
+    list_reports,
+    load_daily_input,
+    load_report,
+    normalize_daily_input,
+    normalize_date,
+    refresh_cls_news,
+    save_daily_input,
+    save_report,
+    seed_demo_content,
+    generate_report,
+)
+from app.services.storage import init_database
+
+
+@asynccontextmanager
+async def lifespan(_app: FastAPI):
+    init_database()
+    seed_demo_content()
+    yield
+
+
+app = FastAPI(
+    title="WeChat Finance Daily",
+    version="0.1.0",
+    lifespan=lifespan,
+)
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+
+@app.get("/api/health")
+def health() -> dict[str, str]:
+    return {"status": "ok"}
+
+
+@app.get("/api/accounts", response_model=list[Account])
+def accounts() -> list[Account]:
+    return get_accounts()
+
+
+@app.get("/api/daily-inputs/{date_str}", response_model=DailyInputDocument)
+def get_daily_inputs(date_str: str) -> DailyInputDocument:
+    return load_daily_input(normalize_date(date_str))
+
+
+@app.put("/api/daily-inputs/{date_str}", response_model=DailyInputDocument)
+def put_daily_inputs(date_str: str, payload: DailyInputUpsertPayload) -> DailyInputDocument:
+    document = normalize_daily_input(normalize_date(date_str), payload)
+    return save_daily_input(document)
+
+
+@app.post("/api/reports/{date_str}/generate", response_model=ReportDocument)
+def generate_daily_report(date_str: str) -> ReportDocument:
+    normalized_date = normalize_date(date_str)
+    input_document = load_daily_input(normalized_date)
+    report = generate_report(normalized_date, input_document)
+    return save_report(report)
+
+
+@app.get("/api/reports", response_model=list[ReportListItem])
+def get_report_list() -> list[ReportListItem]:
+    return list_reports()
+
+
+@app.get("/api/opinions/{date_str}", response_model=ReportDocument)
+def get_opinion_report(date_str: str) -> ReportDocument:
+    normalized_date = normalize_date(date_str)
+    existing = load_report(normalized_date)
+    if existing is not None:
+        return existing
+    report = generate_report(normalized_date, load_daily_input(normalized_date))
+    return save_report(report)
+
+
+@app.get("/api/cls-news", response_model=ClsNewsDocument)
+def get_cls_news_payload(date: str | None = Query(default=None)) -> ClsNewsDocument:
+    return get_cls_news(date)
+
+
+@app.post("/api/cls-news/refresh", response_model=ClsNewsDocument)
+def refresh_cls_news_payload(date: str | None = Query(default=None)) -> ClsNewsDocument:
+    return refresh_cls_news(date)