backedn dev
This commit is contained in:
17
src/main.py
17
src/main.py
@@ -13,9 +13,9 @@ from typing import Optional, Literal, List, Union
|
||||
from fastapi import FastAPI
|
||||
|
||||
|
||||
from database import setup_db_conn, get_inventory_from_redis, read_config_from_db
|
||||
from initialize import initialize_config
|
||||
from utils import setup_logging
|
||||
from database import setup_db_conn, get_inventory_from_redis, get_config_from_db
|
||||
from src.initialize import initialize_config
|
||||
from utils import setup_logging
|
||||
|
||||
|
||||
@asynccontextmanager
|
||||
@@ -23,12 +23,17 @@ async def lifespan(app: FastAPI):
|
||||
''' make loading it async'''
|
||||
log = logging.getLogger('uvicorn')
|
||||
cfg_init_result = initialize_config()
|
||||
|
||||
shared_redis_conn = setup_db_conn(os.getenv('redis_host'), os.getenv('redis_port'))
|
||||
if not shared_redis_conn:
|
||||
log.error("Cannot connect to Redis DB. Exiting...")
|
||||
exit(1)
|
||||
|
||||
inv_check = read_config_from_db(setup_db_conn(os.getenv('redis_host'), os.getenv('redis_port')))
|
||||
log.info(f"Data validity check (DEVELOPER MODE): {inv_check}")
|
||||
inv_check = get_config_from_db(shared_redis_conn)
|
||||
log.info(f"[DEBUG] Data validity healthcheck (DEVELOPER MODE): {inv_check}")
|
||||
if not cfg_init_result:
|
||||
log.error("Configuration initialization failed. Exiting...")
|
||||
exit(1)
|
||||
# exit(1)
|
||||
|
||||
yield
|
||||
log.info("Shutting down FastAPI app...")
|
||||
|
||||
Reference in New Issue
Block a user