Enhance code quality

trolldb/api/api.py

@@ -14,6 +14,7 @@
 """
 
 import asyncio
+import sys
 import time
 from contextlib import contextmanager
 from multiprocessing import Process
@@ -26,7 +27,7 @@
 from pydantic import FilePath, ValidationError, validate_call
 
 from trolldb.api.routes import api_router
-from trolldb.config.config import AppConfig, Timeout, parse_config_yaml_file
+from trolldb.config.config import AppConfig, Timeout, parse_config
 from trolldb.database.mongodb import mongodb_context
 from trolldb.errors.errors import ResponseError
 
@@ -46,6 +47,7 @@
 """These will appear in the auto-generated documentation and are passed to the ``FastAPI`` class as keyword args."""
 
 
+@logger.catch(onerror=lambda _: sys.exit(1))
 @validate_call
 def run_server(config: Union[AppConfig, FilePath], **kwargs) -> None:
     """Runs the API server with all the routes and connection to the database.
@@ -68,10 +70,6 @@
             take precedence over ``config``. Finally, :obj:`API_INFO`, which are hard-coded information for the API
             server, will be concatenated and takes precedence over all.
 
-    Raises:
-        ValidationError:
-            If the function is not called with arguments of valid type.
-
     Example:
         .. code-block:: python
 
@@ -81,7 +79,7 @@
     """
     logger.info("Attempt to run the API server ...")
     if not isinstance(config, AppConfig):
-        config = parse_config_yaml_file(config)
+        config = parse_config(config)
 
     # Concatenate the keyword arguments for the API server in the order of precedence (lower to higher).
     app = FastAPI(**(config.api_server._asdict() | kwargs | API_INFO))
@@ -89,7 +87,7 @@
     app.include_router(api_router)
 
     @app.exception_handler(ResponseError)
     async def auto_handler_response_errors(_, exc: ResponseError) -> PlainTextResponse:
         """Catches all the exceptions raised as a ResponseError, e.g. accessing non-existing databases/collections."""
         status_code, message = exc.get_error_details()
         info = dict(
@@ -99,13 +97,13 @@
         logger.error(f"Response error caught by the API auto exception handler: {info}")
         return PlainTextResponse(**info)
 
     @app.exception_handler(ValidationError)
     async def auto_handler_pydantic_validation_errors(_, exc: ValidationError) -> PlainTextResponse:
         """Catches all the exceptions raised as a Pydantic ValidationError."""
         logger.error(f"Response error caught by the API auto exception handler: {exc}")
         return PlainTextResponse(str(exc), status_code=status.HTTP_500_INTERNAL_SERVER_ERROR)
 
     async def _serve() -> NoReturn:
         """An auxiliary coroutine to be used in the asynchronous execution of the FastAPI application."""
         async with mongodb_context(config.database):
             logger.info("Attempt to start the uvicorn server ...")
@@ -140,7 +138,7 @@
     """
     logger.info("Attempt to run the API server process in a context manager ...")
     if not isinstance(config, AppConfig):
-        config = parse_config_yaml_file(config)
+        config = parse_config(config)
 
     process = Process(target=run_server, args=(config,))
     try: