59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from core.config import API_VERSION, ALLOWED_ORIGINS
|
|
from database.connection import engine
|
|
from database.models import Base
|
|
from api.routers.system_router import router as system_router
|
|
from api.routers.upload_file_router import router as upload_router
|
|
from api.routers.auth_router import router as auth_router
|
|
from contextlib import asynccontextmanager
|
|
from utils.qgis_init import init_qgis
|
|
|
|
app = FastAPI(
|
|
title="ETL Geo Upload Service",
|
|
version=API_VERSION,
|
|
description="Upload Automation API"
|
|
)
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=ALLOWED_ORIGINS,
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# Base.metadata.create_all(bind=engine)
|
|
|
|
# qgis setup
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
global qgs
|
|
qgs = init_qgis()
|
|
print("QGIS initialized")
|
|
|
|
yield
|
|
|
|
# SHUTDOWN (optional)
|
|
print("Shutting down...")
|
|
|
|
app = FastAPI(lifespan=lifespan)
|
|
|
|
@app.get("/qgis/status")
|
|
def qgis_status():
|
|
try:
|
|
from qgis.core import Qgis
|
|
return {
|
|
"qgis_status": "connected",
|
|
"qgis_version": Qgis.QGIS_VERSION
|
|
}
|
|
except Exception as e:
|
|
return {
|
|
"qgis_status": "error",
|
|
"error": str(e)
|
|
}
|
|
|
|
# Register routers
|
|
app.include_router(system_router, tags=["System"])
|
|
app.include_router(auth_router, prefix="/auth", tags=["Auth"])
|
|
app.include_router(upload_router, prefix="/upload", tags=["Upload"]) |