product management, unification of kb and so on.

repositories/invoices.py

@@ -1,6 +1,6 @@
 from typing import Optional
 
-from sqlalchemy import select, update
+from sqlalchemy import select
 from db.models import Invoice
 from sqlalchemy.ext.asyncio import AsyncSession
 
@@ -37,19 +37,15 @@ class InvoiceRepository:
     async def update_invoice_msg(
         self, session: AsyncSession, *, invoice_id: int, inline_message_id: int
     ):
-        stmt = (
-            update(Invoice.__table__)
-            .where(Invoice.id == invoice_id)
-            .values(inline_message_id=inline_message_id)
-        )
-        await session.execute(stmt)
+        invoice = await self.get_invoice_by_id(session, invoice_id)
+        invoice.inline_message_id = inline_message_id
+
+        await session.commit()
 
     async def update_invoice_status(
         self, session: AsyncSession, *, invoice_id: int, status: InvoiceStatus
     ):
-        stmt = (
-            update(Invoice.__table__)
-            .where(Invoice.id == invoice_id)
-            .values(status=status)
-        )
-        await session.execute(stmt)
+        invoice = await self.get_invoice_by_id(session, invoice_id)
+        invoice.status = status
+
+        await session.commit()