AutoShop-Wednesday/tgbot/database/entities.py

# - *- coding: utf- 8 - *-
from dataclasses import dataclass
from typing import Optional


@dataclass
class Category:
    increment: int
    category_id: int
    category_name: str
    category_unix: int


@dataclass
class Position:
    increment: int
    category_id: int
    position_id: int
    position_name: str
    position_price: float
    position_desc: str
    position_unix: int


@dataclass
class Item:
    increment: int
    user_id: int
    category_id: int
    position_id: int
    item_id: int
    item_unix: int
    item_data: str


@dataclass
class Payments:
    id: int
    cryptobot_token: str
    stars_course: float
    status_cryptobot: str
    status_stars: str

    # Оплата через Lolzteam
    lolzteam_token: str
    lolzteam_merchant_id: Optional[int]

    status_lolzteam: str


@dataclass
class Purchase:
    increment: int
    user_id: int
    user_balance_before: float
    user_balance_after: float
    purchase_receipt: int
    purchase_data: str
    purchase_count: int
    purchase_price: float
    purchase_price_one: float
    purchase_position_id: int
    purchase_position_name: str
    purchase_category_id: int
    purchase_category_name: str
    purchase_unix: int


@dataclass
class Refill:
    increment: int
    user_id: int
    refill_receipt: int
    refill_comment: str
    refill_amount: float
    refill_method: str
    refill_unix: int


@dataclass
class Settings:
    id: int
    status_work: str
    status_refill: str
    status_buy: str
    notification_refill: str
    notification_buy: str
    misc_faq: str
    misc_support: str
    misc_bot: str
    misc_hosting_text: str
    misc_token_telegraph: str
    misc_hide_category: str
    misc_hide_position: str
    misc_method_prod: str
    misc_profit_day: int
    misc_profit_week: int
    misc_profit_month: int

    # Реферальная система
    referral_refill_percent: float
    status_referral: str
    referral_bonus_rub: float


@dataclass
class User:
    increment: int
    user_id: int
    user_login: str
    user_name: str
    user_surname: str
    user_fullname: str
    user_balance: float
    user_refill: float
    user_give: float
    user_unix: int

    # Поля для реферальной системы
    user_referrer_id: Optional[int]
    user_referral_balance: float
    user_referral_hold: float


@dataclass
class Referral:
    increment: int
    referrer_id: int
    referral_id: int
    referral_unix: int


@dataclass
class ReferralTransaction:
    increment: int
    user_id: int
    related_user_id: Optional[int]
    transaction_type: str
    source_id: str
    amount: float
    transaction_unix: int


@dataclass
class ReferralWithdrawal:
    increment: int
    user_id: int
    withdrawal_amount: float
    withdrawal_method: str
    withdrawal_recipient: str
    withdrawal_status: str
    withdrawal_unix: int
    processed_unix: Optional[int]
    processed_admin_id: Optional[int]