- Implemented SubscriptionRepository with methods to get, create, and update subscriptions. - Added BalanceTXRepository for creating and retrieving balance transactions. - Introduced a test script for simulating Pally webhook for local testing.
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from typing import TYPE_CHECKING
|
|
|
|
from sqlalchemy import BIGINT, INTEGER, TEXT
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from db.base import Base
|
|
|
|
if TYPE_CHECKING:
|
|
from db.models.bills import Bill
|
|
from db.models.subscriptions import Subscription
|
|
from db.models.transactions import BalanceTransaction
|
|
|
|
|
|
class User(Base):
|
|
__tablename__ = "users"
|
|
|
|
id: Mapped[int] = mapped_column(BIGINT, nullable=False, unique=True, primary_key=True)
|
|
referal_id: Mapped[int] = mapped_column(BIGINT, nullable=True)
|
|
subscription_link: Mapped[str] = mapped_column(TEXT, nullable=True, unique=True)
|
|
balance: Mapped[int] = mapped_column(INTEGER, nullable=False, default=0)
|
|
|
|
bills: Mapped[list["Bill"]] = relationship(
|
|
"Bill",
|
|
back_populates="user",
|
|
cascade="all, delete",
|
|
)
|
|
subscription: Mapped["Subscription"] = relationship(
|
|
"Subscription", back_populates="user", cascade="all, delete-orphan"
|
|
)
|
|
balance_transactions: Mapped[list["BalanceTransaction"]] = relationship(
|
|
"BalanceTransaction", back_populates="user", cascade="all, delete"
|
|
)
|