diff options
Diffstat (limited to 'src/main.py')
| -rwxr-xr-x | src/main.py | 20 |
1 files changed, 16 insertions, 4 deletions
diff --git a/src/main.py b/src/main.py index a62d77c..4a1741e 100755 --- a/src/main.py +++ b/src/main.py @@ -1,4 +1,5 @@ import json +import logging import os import sys import time @@ -13,9 +14,12 @@ from pydantic import BaseModel from db.dictionary import DictDB from db.schema import get_index_keys +from db.sql import Scanner +from log import get_logger, log_format -app = FastAPI() +logger = get_logger() +app = FastAPI() app.add_middleware( CORSMiddleware, allow_origins=["http://localhost:8000"], @@ -38,12 +42,12 @@ for i in range(10): try: db = DictDB() except Exception as e: - print(f"Database not responding, will try again soon: {e}") + logger.info(f"Database not responding, will try again soon: {e}") else: break time.sleep(1) else: - print('Database did not respond after 10 attempts, quitting.') + logger.info('Database did not respond after 10 attempts, quitting.') sys.exit(-1) @@ -57,7 +61,7 @@ def get_pubkey(): with open(keypath, "r") as fd: pubkey = fd.read() except FileNotFoundError: - print(f"Could not find JWT certificate in {keypath}") + logger.info(f"Could not find JWT certificate in {keypath}") sys.exit(-1) return pubkey @@ -191,6 +195,9 @@ async def add(data: Request, Authorize: AuthJWT = Depends()): status_code=400, ) + record_id = key["_id"] + logger.info(f"Record {record_id} added") + return JSONResponse(content={"status": "success", "docs": key}) @@ -257,6 +264,11 @@ async def scanner_put(name, data: Request, Authorize: AuthJWT = Depends()): def main(standalone=False): + log_config = uvicorn.config.LOGGING_CONFIG + log_config["formatters"]["access"]["fmt"] = log_format + log_config["formatters"]["default"]["fmt"] = log_format + log_config["loggers"]["uvicorn"]["propagate"] = False + if not standalone: return app |