chore: minor fixes and formatting
This commit is contained in:
@@ -1,9 +1,7 @@
|
||||
from typing import Optional
|
||||
|
||||
from sqlalchemy import select
|
||||
from db.models import Invoice
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from db.models import Invoice
|
||||
from db.models.invoices import InvoiceStatus
|
||||
|
||||
|
||||
@@ -14,7 +12,7 @@ class InvoiceRepository:
|
||||
*,
|
||||
amount: int,
|
||||
creator_id: int,
|
||||
inline_message_id: Optional[int] = None,
|
||||
inline_message_id: int | None = None,
|
||||
status: InvoiceStatus = InvoiceStatus.PENDING,
|
||||
) -> Invoice:
|
||||
invoice = Invoice(
|
||||
@@ -28,9 +26,7 @@ class InvoiceRepository:
|
||||
await session.commit()
|
||||
return invoice
|
||||
|
||||
async def get_invoice_by_id(
|
||||
self, session: AsyncSession, invoice_id: int
|
||||
) -> Optional[Invoice]:
|
||||
async def get_invoice_by_id(self, session: AsyncSession, invoice_id: int) -> Invoice | None:
|
||||
stmt = select(Invoice).where(Invoice.id == invoice_id)
|
||||
return await session.scalar(stmt)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user