Merge pull request #238 from grillazz/rotoger-one-more-time

rotoger-one-more-time

Author: grillazz

app/api/stuff.py

@@ -1,4 +1,8 @@
+from collections.abc import Callable
+from typing import Annotated, Any
+
 from fastapi import APIRouter, Depends, HTTPException, Request, status
+from pydantic import ValidationError, WrapValidator
 from rotoger import get_logger
 from sqlalchemy.exc import SQLAlchemyError
 from sqlalchemy.ext.asyncio import AsyncSession
@@ -22,12 +26,26 @@ async def create_random_stuff(
     return {"id": str(random_stuff.id)}
 
 
+failed_items: list[dict] = []  # Global or pass via context
+
+def catch_invalid(v: Any, handler: Callable[[Any], Any] ) -> Any:
+    try:
+        return handler(v)
+    except ValidationError:
+        failed_items.append(v)  # Intercept here!
+        return None  # Or raise if needed
+
 @router.post("/add_many", status_code=status.HTTP_201_CREATED)
 async def create_multi_stuff(
-    payload: list[StuffSchema], db_session: AsyncSession = Depends(get_db)
+    payload: list[Annotated[StuffSchema, WrapValidator(catch_invalid)]], db_session: AsyncSession = Depends(get_db)
 ):
+    await logger.ainfo(f">>>{failed_items}")
     try:
-        stuff_instances = [Stuff(**stuff.model_dump()) for stuff in payload]
+        await logger.ainfo(f">>>{failed_items}")
+        await logger.ainfo(f">>>{payload}")
+        stuff_instances = [
+            Stuff(**stuff.model_dump()) for stuff in payload if stuff is not None
+        ]
         db_session.add_all(stuff_instances)
         await db_session.commit()
     except SQLAlchemyError as ex:
@@ -39,6 +57,7 @@ async def create_multi_stuff(
         await logger.ainfo(
             f"{len(stuff_instances)} Stuff instances inserted into the database."
         )
+        return {"inserted": len(stuff_instances)}
         return True
 
 

app/main.py

@@ -21,12 +21,12 @@
 from app.redis import get_redis
 from app.services.auth import AuthBearer
 
-logger = get_logger()
 templates = Jinja2Templates(directory=Path(__file__).parent.parent / "templates")
 
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
+    app.logger = get_logger()
     app.redis = await get_redis()
     postgres_dsn = global_settings.postgres_url.unicode_string()
     try:
@@ -35,12 +35,12 @@ async def lifespan(app: FastAPI):
             min_size=5,
             max_size=20,
         )
-        await logger.ainfo(
+        await app.logger.ainfo(
             "Postgres pool created", idle_size=app.postgres_pool.get_idle_size()
         )
         yield
     except Exception as e:
-        await logger.aerror("Error during app startup", error=repr(e))
+        await app.logger.aerror("Error during app startup", error=repr(e))
         raise
     finally:
         await app.redis.close()

app/schemas/stuff.py

@@ -5,7 +5,6 @@
 
 config = ConfigDict(from_attributes=True)
 
-
 class RandomStuff(BaseModel):
     chaos: dict[str, Any] = Field(
         ..., description="Pretty chaotic JSON data can be added here..."

app/server.py

@@ -0,0 +1,20 @@
+from granian import Granian
+
+
+def startup():
+    print("Server starting up...")
+
+def shutdown():
+    print("Server shutting down...")
+
+server = Granian(
+    "main:app",
+    host="0.0.0.0",  # Bind to all interfaces
+    port=8000,
+    workers=4,
+    interface="asgi",
+    blocking_threads=8  # Optional: threads per worker for blocking ops
+)
+server.on_startup(startup)
+server.on_shutdown(shutdown)
+server.serve_forever()

pyproject.toml

@@ -22,14 +22,14 @@ dependencies = [
     "redis==7.1.0",
     "bcrypt==5.0.0",
     "polars[pyarrow]==1.36.1",
-    "python-multipart==0.0.20",
+    "python-multipart==0.0.22",
     "fastexcel==0.18.0",
     "inline-snapshot==0.31.1",
     "dirty-equals==0.11",
     "polyfactory==3.1.0",
     "granian==2.6.0",
     "apscheduler[redis,sqlalchemy]>=4.0.0a6",
-    "rotoger==0.2.1",
+    "rotoger==0.3.0",
     "pyinstrument>=5.1.2",
 ]