get income_statement report by report_id

app/api/routers/income_statement.py

@@ -47,3 +47,19 @@ async def get_income_statements(
     if len(all_rows) == 0:
         raise HTTPException(status_code=status.HTTP_204_NO_CONTENT, detail="No income statements found for this user")
     return all_rows
+
+
+@r.get(
+    "/{report_id}",
+    response_model=IncomeStatementResponse,
+    responses={
+        200: {"description": "Income statement found"},
+        404: {"description": "Income statement not found"},
+        500: {"description": "Internal server error"},
+    },
+)
+async def get_income_statement(report_id: int, db: AsyncSession = Depends(get_db_session)) -> IncomeStatementResponse:
+    income_statement = await IncomeStatementModel.get(db, id=report_id)
+    if not income_statement:
+        raise HTTPException(status_code=404, detail="Income statement not found")
+    return income_statement

app/model/income_statement.py

@@ -34,3 +34,9 @@ class IncomeStatement(Base, BaseModel):
         query = sql.select(cls).where(cls.user_id == user_id)
         income_statements = await db.scalars(query)
         return income_statements
+
+    @classmethod
+    async def get(cls: "type[IncomeStatement]", db: AsyncSession, id: int) -> "IncomeStatement":
+        query = sql.select(cls).where(cls.id == id)
+        income_statement = await db.scalar(query)
+        return income_statement