104 lines
3.0 KiB
Python
104 lines
3.0 KiB
Python
import asyncio
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.staticfiles import StaticFiles
|
|
|
|
from app.config import settings
|
|
from app.database import Base, engine
|
|
from app import models # noqa: F401
|
|
from app.services.auto_submit import auto_submit_overdue_loop
|
|
from app.services.qrcode_batch_tasks import qrcode_batch_task_loop
|
|
from app.routers import (
|
|
attendance_points,
|
|
auth,
|
|
clock,
|
|
dashboard,
|
|
devices,
|
|
equipment,
|
|
monitor,
|
|
notices,
|
|
people,
|
|
products,
|
|
reconciliation,
|
|
reports,
|
|
reviews,
|
|
usage_stats,
|
|
work_schedule,
|
|
)
|
|
|
|
|
|
def create_app() -> FastAPI:
|
|
app = FastAPI(
|
|
title="嘉恒智能报工后端服务",
|
|
version="0.1.0",
|
|
description="FastAPI service for 嘉恒智能报工小程序",
|
|
)
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
settings.upload_path.mkdir(parents=True, exist_ok=True)
|
|
app.mount("/uploads", StaticFiles(directory=str(settings.upload_path)), name="uploads")
|
|
|
|
def background_tasks_enabled() -> bool:
|
|
if settings.background_tasks_enabled is not None:
|
|
return settings.background_tasks_enabled
|
|
return settings.app_env == "production"
|
|
|
|
@app.on_event("startup")
|
|
async def startup() -> None:
|
|
if settings.auto_create_tables:
|
|
Base.metadata.create_all(bind=engine)
|
|
app.state.auto_submit_task = None
|
|
app.state.qrcode_batch_task = None
|
|
if background_tasks_enabled():
|
|
app.state.auto_submit_task = asyncio.create_task(auto_submit_overdue_loop())
|
|
app.state.qrcode_batch_task = asyncio.create_task(qrcode_batch_task_loop())
|
|
|
|
@app.on_event("shutdown")
|
|
async def shutdown() -> None:
|
|
task = getattr(app.state, "auto_submit_task", None)
|
|
if task:
|
|
task.cancel()
|
|
try:
|
|
await task
|
|
except asyncio.CancelledError:
|
|
pass
|
|
task = getattr(app.state, "qrcode_batch_task", None)
|
|
if task:
|
|
task.cancel()
|
|
try:
|
|
await task
|
|
except asyncio.CancelledError:
|
|
pass
|
|
|
|
@app.get("/health")
|
|
def health() -> dict[str, str]:
|
|
return {"status": "ok"}
|
|
|
|
app.include_router(auth.router)
|
|
app.include_router(attendance_points.router)
|
|
app.include_router(people.router)
|
|
app.include_router(products.router)
|
|
app.include_router(clock.router)
|
|
app.include_router(reports.router)
|
|
app.include_router(reviews.router)
|
|
app.include_router(dashboard.router)
|
|
app.include_router(devices.router)
|
|
app.include_router(equipment.router)
|
|
app.include_router(monitor.router)
|
|
app.include_router(notices.router)
|
|
app.include_router(reconciliation.router)
|
|
app.include_router(work_schedule.router)
|
|
app.include_router(usage_stats.router)
|
|
return app
|
|
|
|
|
|
app = create_app()
|