Spaces:
Sleeping
Sleeping
File size: 2,164 Bytes
2636575 d5684b3 f92858b 2636575 d5684b3 a4ec216 2636575 d5684b3 2636575 f92858b d5684b3 f92858b d5684b3 f92858b d5684b3 f92858b d5684b3 f92858b d5684b3 f92858b d5684b3 f92858b a4ec216 2636575 f92858b 2636575 f92858b 2636575 f92858b 2636575 f92858b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 |
import logging
import os
import uvicorn
from contextlib import asynccontextmanager
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import RedirectResponse
from app.api.routers.chat import chat_router
from app.api.routers.user import user_router
from app.api.routers.transaction import transaction_router
from app.settings import init_settings
from fastapi.staticfiles import StaticFiles
from alembic.config import Config
from alembic import command
from app.engine.postgresdb import postgresdb
from config.index import config
logger = logging.getLogger("uvicorn")
def run_migrations(connection, cfg):
cfg.attributes["connection"] = connection
command.upgrade(cfg, "head")
async def run_async_migrations():
logger.info("Running migrations...")
async_engine = postgresdb.get_engine()
async with async_engine.begin() as conn:
await conn.run_sync(run_migrations, Config("alembic.ini"))
logger.info("Finished running migrations")
@asynccontextmanager
async def lifespan(app_: FastAPI):
yield
if postgresdb.get_engine() is not None:
await postgresdb.close()
init_settings()
# init_observability()
def init_app():
app = FastAPI(lifespan=lifespan)
if config.ENVIRONMENT == "dev":
logger.warning("Running in development mode - allowing CORS for all origins")
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
if os.path.exists("data"):
app.mount("/api/data", StaticFiles(directory="data"), name="static")
app.include_router(chat_router, prefix="/api/chat")
app.include_router(user_router)
app.include_router(transaction_router)
return app
app = init_app()
if __name__ == "__main__":
# Redirect to documentation page when accessing base URL
@app.get("/")
async def redirect_to_docs():
return RedirectResponse(url="/docs")
uvicorn.run(app="main:app", host=config.APP_HOST, port=config.APP_PORT, reload=(config.ENVIRONMENT == "dev"))
|