from sqlalchemy import BIGINT, INTEGER, TEXT
from sqlalchemy.orm import Mapped, mapped_column

from db.base import Base


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)