handling 404 properly

Author: BaseMax

app/api/snippets.py

@@ -1,4 +1,4 @@
-from fastapi import APIRouter, Depends
+from fastapi import HTTPException, APIRouter, Depends
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy import select
 from app.db.session import SessionLocal
@@ -25,7 +25,10 @@ async def create_snippet(data: SnippetCreate, db: AsyncSession = Depends(get_db)
     return snippet
 
 
-@router.get("/{snippet_id}", response_model=SnippetOut)
+@router.get("/{snippet_id}")
 async def get_snippet(snippet_id: int, db: AsyncSession = Depends(get_db)):
     q = await db.execute(select(Snippet).where(Snippet.id == snippet_id))
-    return q.scalar_one()
+    snippet = q.scalar_one_or_none()  # <-- returns None if not found
+    if snippet is None:
+        raise HTTPException(status_code=404, detail="Snippet not found")
+    return snippet

app/main.py

@@ -1,7 +1,17 @@
+from contextlib import asynccontextmanager
 from fastapi import FastAPI
+from app.db.models import Base
+from app.db.session import engine
 from app.api import snippets, execute
 
-app = FastAPI(title="Code Snippet Execution Platform")
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    async with engine.begin() as conn:
+        await conn.run_sync(Base.metadata.create_all)
+    yield
+    await engine.dispose()
+
+app = FastAPI(title="Code Snippet Execution Platform", lifespan=lifespan)
 
 app.include_router(snippets.router)
 app.include_router(execute.router)