feat(sub): handling upgrade/downgrade

rw integration and /me endpoints are left

db/models/invoice.py

@@ -7,7 +7,7 @@ from db.base import Base
 from schemas.invoices import InvoiceStatus
 
 if TYPE_CHECKING:
-    from db.models import User
+    from db.models import Order, User
 
 
 class Invoice(Base):
@@ -17,9 +17,11 @@ class Invoice(Base):
         INTEGER, autoincrement=True, unique=True, nullable=False, primary_key=True
     )
     creator_id: Mapped[int] = mapped_column(ForeignKey("users.id"), nullable=False)
+    order_id: Mapped[int | None] = mapped_column(ForeignKey("orders.id"), nullable=True)
     amount: Mapped[float] = mapped_column(FLOAT, nullable=False)
     status: Mapped[InvoiceStatus] = mapped_column(
         Enum(InvoiceStatus, name="invoicestatus"), nullable=False, default=InvoiceStatus.ACTIVE
     )
 
     creator: Mapped["User"] = relationship("User", lazy="selectin")
+    order: Mapped["Order | None"] = relationship("Order", lazy="selectin")

db/models/orders.py

@@ -1,7 +1,8 @@
+import datetime
 from enum import StrEnum
 from typing import TYPE_CHECKING
 
-from sqlalchemy import INTEGER, Enum, ForeignKey
+from sqlalchemy import FLOAT, INTEGER, DateTime, Enum, ForeignKey
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from db.base import Base
@@ -38,9 +39,13 @@ class Order(Base):
 
     devices: Mapped[int] = mapped_column(INTEGER, nullable=False)
     duration_days: Mapped[int] = mapped_column(INTEGER, nullable=False)
+    total_amount: Mapped[float] = mapped_column(FLOAT, nullable=False)
+    balance_amount: Mapped[float] = mapped_column(FLOAT, nullable=False, default=0)
     status: Mapped[OrderStatus] = mapped_column(
         Enum(OrderStatus, name="orderstatus"), nullable=False, default=OrderStatus.PENDING
     )
+    applies_at: Mapped[datetime.datetime | None] = mapped_column(DateTime(True), nullable=True)
+    applied_at: Mapped[datetime.datetime | None] = mapped_column(DateTime(True), nullable=True)
 
     user: Mapped["User"] = relationship("User", back_populates="orders", lazy="selectin")
     addons: Mapped[list["OrderAddon"]] = relationship(