Vladless
40 lines
1.5 KiB
Python
40 lines
1.5 KiB
Python
import uuid
|
|
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import BigInteger, Boolean, Column, DateTime, ForeignKey, Integer, String
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from ._base import Base, DictLikeMixin
|
|
|
|
|
|
class Gift(DictLikeMixin, Base):
|
|
__tablename__ = "gifts"
|
|
|
|
gift_id = Column(String, primary_key=True, default=lambda: uuid.uuid4().hex)
|
|
sender_user_id = Column(BigInteger, nullable=True)
|
|
recipient_user_id = Column(BigInteger, nullable=True)
|
|
sender_tg_id = Column(BigInteger, ForeignKey("users.tg_id"), nullable=True, index=True)
|
|
recipient_tg_id = Column(BigInteger, ForeignKey("users.tg_id"), nullable=True, index=True)
|
|
selected_months = Column(Integer)
|
|
expiry_time = Column(DateTime)
|
|
gift_link = Column(String)
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
is_used = Column(Boolean, default=False)
|
|
is_unlimited = Column(Boolean, default=False)
|
|
max_usages = Column(Integer, nullable=True)
|
|
tariff_id: Mapped[int | None] = mapped_column(ForeignKey("tariffs.id"))
|
|
|
|
selected_device_limit = Column(Integer, nullable=True)
|
|
selected_traffic_gb = Column(Integer, nullable=True)
|
|
selected_price_rub = Column(Integer, nullable=True)
|
|
|
|
|
|
class GiftUsage(DictLikeMixin, Base):
|
|
__tablename__ = "gift_usages"
|
|
|
|
gift_id = Column(String, ForeignKey("gifts.gift_id"), primary_key=True)
|
|
user_id = Column(BigInteger, nullable=False, primary_key=True)
|
|
tg_id = Column(BigInteger, nullable=True, index=True)
|
|
used_at = Column(DateTime, default=datetime.utcnow)
|