1 files changed, 21 insertions, 1 deletions

diff --git a/handlers/user/pay_invoice.py b/handlers/user/pay_invoice.py
index 98f80a6..db75f47 100644
--- a/handlers/user/pay_invoice.py
+++ b/handlers/user/pay_invoice.py
@@ -20,7 +20,15 @@ from sqlalchemy.ext.asyncio import AsyncSession
 
 from libs.fsm import get_data, set_data
 from libs.user import mention
-from models import Payment, PaymentStatus, ReceiptFile, ReceiptFileType, User, UserRole
+from models import (
+    Invoice,
+    Payment,
+    PaymentStatus,
+    ReceiptFile,
+    ReceiptFileType,
+    User,
+    UserRole,
+)
 from models.callback_data import PayInvoiceClb, PaymentStatusClb
 
 router = Router(name="pay_invoice")
@@ -44,7 +52,19 @@ async def button(
     state: FSMContext,
     callback_data: PayInvoiceClb,
     session: AsyncSession,
+    user: User,
 ) -> None:
+    if user.is_admin():
+        await clb.answer("Администраторы не могут оплачивать счета", show_alert=True)
+        return
+
+    invoice = await session.get(Invoice, callback_data.invoice_id)
+    assert invoice is not None
+
+    if user.datetime > invoice.datetime:
+        await clb.answer("Вы не можете оплатить данный счёт", show_alert=True)
+        return
+
     payment = await session.scalar(
         select(Payment).where(
             and_(