52 lines
1.2 KiB
Python
52 lines
1.2 KiB
Python
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.openapi.utils import get_openapi
|
|
|
|
from app.api.routes import health, api, messages
|
|
from app.db.session import engine
|
|
from app.models import base
|
|
|
|
# Create database tables
|
|
base.Base.metadata.create_all(bind=engine)
|
|
|
|
app = FastAPI(
|
|
title="General Communication Service",
|
|
description="A general purpose communication service API",
|
|
version="0.1.0",
|
|
)
|
|
|
|
# Enable CORS
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# Include routes
|
|
app.include_router(health.router)
|
|
app.include_router(api.router, prefix="/api/v1")
|
|
app.include_router(messages.router, prefix="/api/v1/messages")
|
|
|
|
|
|
# Custom OpenAPI schema
|
|
def custom_openapi():
|
|
if app.openapi_schema:
|
|
return app.openapi_schema
|
|
openapi_schema = get_openapi(
|
|
title=app.title,
|
|
version=app.version,
|
|
description=app.description,
|
|
routes=app.routes,
|
|
)
|
|
app.openapi_schema = openapi_schema
|
|
return app.openapi_schema
|
|
|
|
|
|
app.openapi = custom_openapi
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=True) |