add loggers headlers

Author: sumaro2101

api_v1/__init__.py

@@ -1,3 +1,7 @@
 from .routers import register_routers
+from exeptions import ValidationError
 
-__all__ = ('register_routers',)
+
+__all__ = ('register_routers',
+           'ValidationError',
+           )

api_v1/exeptions.py

@@ -0,0 +1,9 @@
+from starlette.exceptions import HTTPException
+
+
+class ValidationError(HTTPException):
+    """
+    Исключение вызванное проблемами с валидацией
+    """
+    
+    pass

app_includes/__init__.py

@@ -0,0 +1,79 @@
+import json
+from fastapi import FastAPI, Request
+from fastapi.exceptions import HTTPException
+from starlette.exceptions import HTTPException as StarletteHTTPException
+
+from http import HTTPStatus
+
+from logs.logging import logger
+from api_v1 import ValidationError
+
+
+def register_errors(app: FastAPI) -> None:
+    @app.exception_handler(ValidationError)
+    async def validation_error_handler(
+        request: Request,
+        exc: ValidationError,
+        ):
+        """
+        Логирование всех ValidationError
+        """
+        logger.opt(exception=True).warning(exc)
+        response = dict(
+            status=False,
+            error_code=422,
+            message=exc.detail,
+        )
+        json_response = json.dumps(response).encode(encoding='utf-8')
+        return json_response
+
+    @app.exception_handler(HTTPException)
+    async def http_error_handler(
+        request: Request,
+        exc: HTTPException,
+        ):
+        """
+        Логирование всех HTTPException
+        """
+        logger.opt(exception=True).warning(exc)
+        response = dict(
+            status=False,
+            error_code=exc.status_code,
+            message=exc.detail,
+        )
+        json_response = json.dumps(response).encode(encoding='utf-8')
+        return json_response
+
+    @app.exception_handler(Exception)
+    async def error_handler(
+        request: Request,
+        exc: Exception,
+        ):
+        """
+        Логирование всех StarletteHTTPException
+        """
+        logger.exception(exc)
+        response = dict(
+            status=False,
+            error_code=exc.status_code,
+            message=exc.detail,
+        )
+        json_response = json.dumps(response).encode(encoding='utf-8')
+        return json_response
+
+    @app.exception_handler(ValidationError)
+    async def validation_error_handler(
+        request: Request,
+        exc: ValidationError,
+        ):
+        """
+        Логирование всех ValidationError
+        """
+        logger.opt(exception=True).warning(exc)
+        response = dict(
+            status=False,
+            error_code=500,
+            message=HTTPStatus(500).phrase,
+        )
+        json_response = json.dumps(response).encode(encoding='utf-8')
+        return json_response

app_includes/logs_errors.py

@@ -1,8 +1,13 @@
+from pathlib import Path
 from pydantic_settings import BaseSettings, SettingsConfigDict
 from pydantic import BaseModel
 from starlette.config import Config
 
 
+base_dir = Path(__file__).resolve().parent.parent.parent
+log_dir = base_dir.joinpath('logs')
+
+
 config = Config('.env')
 
 
@@ -47,6 +52,8 @@ class Settings(BaseSettings):
     rabbit: RabbitSettings = RabbitSettings()
     debug: bool = bool(int(config('DEBUG')))
     API_PREFIX: str = '/api/v1'
+    BASE_DIR: Path = base_dir
+    LOG_DIR: Path = log_dir
 
 
 settings = Settings()

config/config.py

@@ -0,0 +1,4 @@
+from .logging import create_log_dirs
+
+
+__all__ = ('create_log_dirs',)

logs/__init__.py

@@ -0,0 +1,64 @@
+import sys
+from loguru import logger
+
+from config import settings
+
+
+log_dir = settings.LOG_DIR
+
+
+def create_log_dirs() -> None:
+    """
+    Создание корневой папки для логов
+    """    
+    log_directory = settings.LOG_DIR
+    log_directory.mkdir(parents=True, exist_ok=True)
+
+
+logger.remove()
+
+logger.add(
+    log_dir.joinpath('access.log'),
+    rotation='15MB',
+    format='{time:YYYY-MM-DD HH:mm:ss} - {level} - {message}',
+    encoding='utf-8',
+    enqueue=True,
+    level='DEBUG',
+    diagnose=False,
+    backtrace=False,
+    colorize=False,
+    filter=lambda record: record['level'].no < 40,
+)
+
+logger.add(
+    log_dir.joinpath('error.log'),
+    rotation='15MB',
+    format='{time:YYYY-MM-DD HH:mm:ss} - {level} - {message}',
+    encoding='utf-8',
+    enqueue=True,
+    level='ERROR',
+    diagnose=False,
+    backtrace=False,
+    colorize=False,
+)
+
+logger.add(
+    sink=sys.stderr,
+    format='{time:YYYY-MM-DD HH:mm:ss} - {level} - {message}',
+    enqueue=True,
+    level='ERROR',
+    diagnose=False,
+    backtrace=False,
+    colorize=False,
+)
+
+logger.add(
+    sink=sys.stdout,
+    format='{time:YYYY-MM-DD HH:mm:ss} - {level} - {message}',
+    enqueue=True,
+    level='DEBUG',
+    diagnose=False,
+    backtrace=False,
+    colorize=False,
+    filter=lambda record: record['level'].no < 40,
+)

logs/logging.py

@@ -1,11 +1,17 @@
 from contextlib import asynccontextmanager
 from fastapi import FastAPI
-from config import db_connection, BaseModel
 
+from config import db_connection, BaseModel
 from api_v1 import register_routers
+from logs import create_log_dirs
+
 
 
 def start_app() -> FastAPI:
+    """
+    Создание приложения со всеми настройками
+    """
+    create_log_dirs()
     app = FastAPI(lifespan=lifespan)
     register_routers(app=app)
     return app