Notes

#
# Copyright (c) 2024 The Capital Group Companies, Inc.
#

from contextlib import asynccontextmanager
from datetime import datetime
from typing import Any

import orjson
from fastapi import FastAPI, Request
from fastapi.middleware.gzip import GZipMiddleware
from fastapi.responses import JSONResponse
from starlette.middleware import Middleware
from starlette.middleware.base import BaseHTTPMiddleware
from starlette.middleware.cors import CORSMiddleware

from .database import create_database_pool, destroy_database_pool
from .middleware import error_middleware, log_middleware
from .router import exposures, investors, melt_v1beta1, portfolios, sedols
from .router.investors import get_investor


class ORJSONResponse(JSONResponse):
media_type = "application/json"

def render(self, content: Any) -> bytes:
return orjson.dumps(content)


@asynccontextmanager
async def lifespan(app: FastAPI):
app.state.db_conn_pool = await create_database_pool()

yield

await destroy_database_pool(app.state.db_conn_pool)


middleware = [
Middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
]

app = FastAPI(
title="Onyx API",
contact={
"name": "Onyx",
"url": "https://quanthub.capgroup.com/",
"email": "[email protected]",
},
lifespan=lifespan,
middleware=middleware,
default_response_class=ORJSONResponse,
version="0.1.0",
)

app.add_middleware(GZipMiddleware, minimum_size=1000)
app.add_middleware(BaseHTTPMiddleware, dispatch=error_middleware)
app.add_middleware(BaseHTTPMiddleware, dispatch=log_middleware)

app.include_router(
investors.router,
prefix="/api/v1",
tags=["Investors"],
)

app.include_router(
portfolios.router,
prefix="/api/v1",
tags=["Portfolios"],
)

app.include_router(
melt_v1beta1.router,
prefix="/api/v1beta1",
tags=["Portfolios"],
)

app.include_router(
sedols.router,
prefix="/api/v1",
tags=["Security Data by SEDOL"],
)

app.include_router(
exposures.router,
prefix="/api/v1",
tags=["Exposures"],
)


@app.get("/")
async def root():
return {"message": "Onyx API is running!!"}


@app.get("/health")
async def health():
return {"message": f"Current Time : {datetime.now()}"}


@app.get("/healthCheck")
async def health_check():
res = await get_investor(Request,'01122','2024-01-31')
print('RES-->',res)


breakpoint()

result = {
"message": "API health check verified"
}
return result