fastapi-armor-lib 0.1.0__py3-none-any.whl

fastapi_armor/__init__.py

@@ -0,0 +1,3 @@
+from .armor import Armor
+
+__all__ = ["Armor"]

fastapi_armor/armor.py

@@ -0,0 +1,27 @@
+from fastapi import FastAPI
+
+from .error_handler import add_exception_handlers
+from .logger import setup_logger
+from .sentry import init_sentry
+
+
+class Armor:
+    def __init__(
+        self,
+        app: FastAPI,
+        log_file: str = "logs/app.log",
+        log_max_bytes: int = 5_000_000,
+        log_backup_count: int = 3,
+        sentry_dsn: str = "",
+        environment: str = "development",
+    ):
+        self.app = app
+        self.logger = setup_logger(
+            log_file=log_file,
+            max_bytes=log_max_bytes,
+            backup_count=log_backup_count,
+        )
+        add_exception_handlers(app=self.app, logger=self.logger)
+
+        if sentry_dsn:
+            init_sentry(dsn=sentry_dsn, environment=environment)

fastapi_armor/error_handler.py

@@ -0,0 +1,36 @@
+import logging
+
+from fastapi import FastAPI, HTTPException, Request
+from fastapi.exceptions import RequestValidationError
+from fastapi.responses import JSONResponse
+
+JSON_MEDIA_TYPE = "application/json; charset=utf-8"
+
+
+def add_exception_handlers(app: FastAPI, logger: logging.Logger) -> None:
+    @app.exception_handler(HTTPException)
+    async def http_exception_handler(_: Request, exc: HTTPException) -> JSONResponse:
+        return JSONResponse(
+            status_code=exc.status_code,
+            content={"detail": exc.detail},
+            media_type=JSON_MEDIA_TYPE,
+        )
+
+    @app.exception_handler(RequestValidationError)
+    async def validation_exception_handler(
+        _: Request, exc: RequestValidationError
+    ) -> JSONResponse:
+        return JSONResponse(
+            status_code=422,
+            content={"error": "Validation failed", "detail": exc.errors()},
+            media_type=JSON_MEDIA_TYPE,
+        )
+
+    @app.exception_handler(Exception)
+    async def generic_exception_handler(_: Request, exc: Exception) -> JSONResponse:
+        logger.exception("Unhandled exception occurred", exc_info=exc)
+        return JSONResponse(
+            status_code=500,
+            content={"error": "Internal server error"},
+            media_type=JSON_MEDIA_TYPE,
+        )

fastapi_armor/logger.py

@@ -0,0 +1,37 @@
+import logging
+from logging.handlers import RotatingFileHandler
+from pathlib import Path
+
+
+def setup_logger(
+    log_file: str = "logs/app.log",
+    max_bytes: int = 5_000_000,
+    backup_count: int = 3,
+) -> logging.Logger:
+    logger = logging.getLogger("fastapi_armor")
+    logger.setLevel(logging.INFO)
+
+    if logger.handlers:
+        return logger
+
+    log_path = Path(log_file)
+    log_path.parent.mkdir(parents=True, exist_ok=True)
+
+    formatter = logging.Formatter("%(asctime)s | %(levelname)s | %(message)s")
+
+    file_handler = RotatingFileHandler(
+        filename=log_path,
+        maxBytes=max_bytes,
+        backupCount=backup_count,
+        encoding="utf-8",
+    )
+    file_handler.setFormatter(formatter)
+
+    console_handler = logging.StreamHandler()
+    console_handler.setFormatter(formatter)
+
+    logger.addHandler(file_handler)
+    logger.addHandler(console_handler)
+    logger.propagate = False
+
+    return logger

fastapi_armor/sentry.py

@@ -0,0 +1,19 @@
+from typing import Optional
+
+import sentry_sdk
+from sentry_sdk.integrations.fastapi import FastApiIntegration
+from sentry_sdk.integrations.starlette import StarletteIntegration
+
+
+def init_sentry(dsn: str, environment: str = "development") -> Optional[None]:
+    if not dsn:
+        return None
+
+    sentry_sdk.init(
+        dsn=dsn,
+        environment=environment,
+        integrations=[FastApiIntegration(), StarletteIntegration()],
+        traces_sample_rate=1.0,
+        send_default_pii=False,
+    )
+    return None

fastapi_armor_lib-0.1.0.dist-info/METADATA

@@ -0,0 +1,140 @@
+Metadata-Version: 2.4
+Name: fastapi-armor-lib
+Version: 0.1.0
+Summary: Production-ready error handling, logging, and monitoring for FastAPI apps
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: fastapi>=0.100.0
+Requires-Dist: sentry-sdk[fastapi]>=1.40.0
+Provides-Extra: dev
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: httpx; extra == "dev"
+Requires-Dist: uvicorn; extra == "dev"
+
+# fastapi-armor 🛡️
+
+> Production-ready error handling, logging, and Sentry monitoring for FastAPI — in one line.
+
+[![Python](https://img.shields.io/badge/python-3.9+-blue.svg)](https://www.python.org/)
+[![FastAPI](https://img.shields.io/badge/FastAPI-0.100+-009688.svg)](https://fastapi.tiangolo.com/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![PyPI version](https://img.shields.io/pypi/v/fastapi-armor.svg)](https://pypi.org/project/fastapi-armor/)
+
+---
+
+## Why fastapi-armor?
+
+Every production FastAPI app needs the same boilerplate: structured logging, clean error responses, and error monitoring. **fastapi-armor** wires all of that up automatically — no copy-pasting, no missing edge cases.
+
+**Before:**
+```python
+from fastapi import FastAPI, HTTPException, Request
+from fastapi.responses import JSONResponse
+import logging, logging.handlers, sentry_sdk
+# ... 60+ lines of setup code
+
+app = FastAPI()
+```
+
+**After:**
+```python
+from fastapi import FastAPI
+from fastapi_armor import Armor
+
+app = FastAPI()
+Armor(app)  # ✅ Done.
+```
+
+---
+
+## Installation
+
+```bash
+pip install fastapi-armor
+```
+
+---
+
+## Quick Start
+
+```python
+from fastapi import FastAPI
+from fastapi_armor import Armor
+
+app = FastAPI()
+
+Armor(
+    app,
+    log_file="logs/app.log",
+    sentry_dsn="https://your-dsn@sentry.io/123",  # optional
+    environment="production",
+)
+
+@app.get("/")
+def root():
+    return {"status": "running"}
+```
+
+That's it. Your app now has:
+- ✅ Rotating file logs + console output
+- ✅ Clean JSON error responses for all exception types
+- ✅ Sentry integration (if DSN is provided)
+
+---
+
+## What It Does
+
+### 🔴 Error Handling
+
+| Exception Type | Response |
+|---|---|
+| `HTTPException` | `{"detail": "..."}` with original status code |
+| `RequestValidationError` | `{"error": "Validation failed", "detail": [...]}` with 422 |
+| Any other `Exception` | `{"error": "Internal server error"}` with 500 (full traceback logged internally) |
+
+All responses use `Content-Type: application/json; charset=utf-8`.
+
+### 📋 Logging
+
+- **Rotating file logs** — auto-creates the `logs/` directory
+- **Console output** — for local development
+- **Format:** `2024-01-01 12:00:00 | ERROR | Traceback (most recent call last): ...`
+- Named logger: `fastapi_armor`
+
+### 🔍 Sentry (optional)
+
+Initializes `sentry_sdk` with `FastApiIntegration` and `StarletteIntegration` when a DSN is provided. Private data is not sent by default.
+
+---
+
+## Parameters
+
+| Parameter | Type | Default | Description |
+|---|---|---|---|
+| `app` | `FastAPI` | required | Your FastAPI application instance |
+| `log_file` | `str` | `"logs/app.log"` | Path to the rotating log file |
+| `log_max_bytes` | `int` | `5_000_000` | Max size per log file (5 MB) |
+| `log_backup_count` | `int` | `3` | Number of backup log files to keep |
+| `sentry_dsn` | `str` | `""` | Sentry DSN — leave empty to disable |
+| `environment` | `str` | `"development"` | Environment tag sent to Sentry |
+
+---
+
+## Requirements
+
+- Python 3.9+
+- FastAPI 0.100+
+- sentry-sdk 1.40+ *(installed automatically)*
+
+---
+
+## Development
+
+```bash
+git clone https://github.com/raghadalb-tech/fastapi-armor.git
+cd fastapi-armor
+pip install -e ".[dev]"
+pytest tests/ -v
+```
+
+Test output:

fastapi_armor_lib-0.1.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+fastapi_armor/__init__.py,sha256=L--H3LWEWrvAxV7XEYKxPlVooWggAckMw98A5Gbut20,49
+fastapi_armor/armor.py,sha256=JVYS6vIt0pcmwVmuz3Mg2d07txxjBAvA7i2ER-saJ14,768
+fastapi_armor/error_handler.py,sha256=ik28lhUvI65TP1AyIsmNrb5nCWv0LNGRTqipRyGZuC0,1332
+fastapi_armor/logger.py,sha256=WY5OgP6VGVed02tbMdz8t5OZK9nJL4rq8tb9T544Hxo,975
+fastapi_armor/sentry.py,sha256=UiUQr_nbsowZDVCakjz4HTg2ClVW82Pceazy3Q4ZRUw,540
+fastapi_armor_lib-0.1.0.dist-info/METADATA,sha256=9Ftgb_TJjsxNmOfWINtwfOfVSkr01Z1BOxcxHR1O4M4,3849
+fastapi_armor_lib-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+fastapi_armor_lib-0.1.0.dist-info/top_level.txt,sha256=I-0oX8mMbAJwIsoxfh2aGYzOC3ADD3bapAshG0cjEMs,14
+fastapi_armor_lib-0.1.0.dist-info/RECORD,,

fastapi_armor_lib-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

fastapi_armor_lib-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+fastapi_armor