71c74cd658
* Async chalice.core.tenants:tenants_exists, and propagate * rework after review * typofix * fix(chalice): small fixes * fix(api): use a global variable to store singletong... ... that will not work if several POSIX threads are serving from the same POSIX processus. * fix(api): pass database argument as dictionary. ref: https://www.postgresql.org/docs/current/libpq-connect.html#LIBPQ-CONNSTRING * chore(api): ease debugging with simple return statement. * NOTES++ --------- Co-authored-by: Taha Yassine Kraiem <tahayk2@gmail.com>
134 lines
4.1 KiB
Python
134 lines
4.1 KiB
Python
import logging
|
|
import time
|
|
from contextlib import asynccontextmanager
|
|
|
|
import psycopg_pool
|
|
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
|
from decouple import config
|
|
from fastapi import FastAPI, Request
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.middleware.gzip import GZipMiddleware
|
|
from psycopg import AsyncConnection
|
|
from starlette.responses import StreamingResponse
|
|
|
|
from chalicelib.utils import helper
|
|
from chalicelib.utils import pg_client
|
|
from crons import core_crons, core_dynamic_crons
|
|
from routers import core, core_dynamic, additional_routes
|
|
from routers.subs import insights, metrics, v1_api, health
|
|
|
|
loglevel = config("LOGLEVEL", default=logging.WARNING)
|
|
print(f">Loglevel set to: {loglevel}")
|
|
logging.basicConfig(level=loglevel)
|
|
import orpy
|
|
from psycopg.rows import dict_row
|
|
|
|
|
|
class ORPYAsyncConnection(AsyncConnection):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, row_factory=dict_row, **kwargs)
|
|
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
# Startup
|
|
logging.info(">>>>> starting up <<<<<")
|
|
ap_logger = logging.getLogger('apscheduler')
|
|
ap_logger.setLevel(loglevel)
|
|
|
|
app.schedule = AsyncIOScheduler()
|
|
await pg_client.init()
|
|
app.schedule.start()
|
|
|
|
for job in core_crons.cron_jobs + core_dynamic_crons.cron_jobs:
|
|
app.schedule.add_job(id=job["func"].__name__, **job)
|
|
|
|
ap_logger.info(">Scheduled jobs:")
|
|
for job in app.schedule.get_jobs():
|
|
ap_logger.info({"Name": str(job.id), "Run Frequency": str(job.trigger), "Next Run": str(job.next_run_time)})
|
|
|
|
|
|
database = {
|
|
"host": config("pg_host", default="localhost"),
|
|
"dbname": config("pg_dbname", default="orpy"),
|
|
"user": config("pg_user", default="orpy"),
|
|
"password": config("pg_password", default="orpy"),
|
|
"port": config("pg_port", cast=int, default=5432),
|
|
"application_name": config("APP_NAME", default="PY"),
|
|
}
|
|
|
|
database = psycopg_pool.AsyncConnectionPool(kwargs=database, connection_class=ORPYAsyncConnection)
|
|
orpy.set(orpy.Application(
|
|
database,
|
|
))
|
|
|
|
# App listening
|
|
yield
|
|
|
|
# Shutdown
|
|
await database.close()
|
|
logging.info(">>>>> shutting down <<<<<")
|
|
app.schedule.shutdown(wait=False)
|
|
await pg_client.terminate()
|
|
|
|
|
|
app = FastAPI(root_path=config("root_path", default="/api"), docs_url=config("docs_url", default=""),
|
|
redoc_url=config("redoc_url", default=""), lifespan=lifespan)
|
|
app.add_middleware(GZipMiddleware, minimum_size=1000)
|
|
|
|
|
|
@app.middleware('http')
|
|
async def or_middleware(request: Request, call_next):
|
|
if helper.TRACK_TIME:
|
|
now = time.time()
|
|
try:
|
|
response: StreamingResponse = await call_next(request)
|
|
except:
|
|
logging.error(f"{request.method}: {request.url.path} FAILED!")
|
|
raise
|
|
if response.status_code // 100 != 2:
|
|
logging.warning(f"{request.method}:{request.url.path} {response.status_code}!")
|
|
if helper.TRACK_TIME:
|
|
now = time.time() - now
|
|
if now > 2:
|
|
now = round(now, 2)
|
|
logging.warning(f"Execution time: {now} s for {request.method}: {request.url.path}")
|
|
return response
|
|
|
|
|
|
origins = [
|
|
"*",
|
|
]
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=origins,
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
app.include_router(core.public_app)
|
|
app.include_router(core.app)
|
|
app.include_router(core.app_apikey)
|
|
app.include_router(core_dynamic.public_app)
|
|
app.include_router(core_dynamic.app)
|
|
app.include_router(core_dynamic.app_apikey)
|
|
app.include_router(metrics.app)
|
|
app.include_router(insights.app)
|
|
app.include_router(v1_api.app_apikey)
|
|
app.include_router(health.public_app)
|
|
app.include_router(health.app)
|
|
app.include_router(health.app_apikey)
|
|
|
|
app.include_router(additional_routes.public_app)
|
|
app.include_router(additional_routes.app)
|
|
app.include_router(additional_routes.app_apikey)
|
|
|
|
# @app.get('/private/shutdown', tags=["private"])
|
|
# async def stop_server():
|
|
# logging.info("Requested shutdown")
|
|
# await shutdown()
|
|
# import os, signal
|
|
# os.kill(1, signal.SIGTERM)
|