qBitDownload-Bot/tubot/db/types.py

# -*- coding: utf-8 -*-

#############################
# Types for DataBase module #
#############################

# Imports
from enum import Enum


class CacheDBTypes(Enum):
    """
    Types of CacheDB
    """

    PythonPKL = "python"


class UserStates(Enum):
    """
    Types of User status
    """

    ...


class User(object):
    """
    User class
    """

    tg_id: int
    name: str
    state: UserStates | None = None
    auth: bool = False

    def __init__(
        self, tg_id: int, name: str, state: UserStates | None = None, auth: bool = False
    ) -> None:
        self.tg_id = tg_id
        self.name = name
        self.state = state
        self.auth = auth

    @property
    def dict(self):
        return {
            "tg_id": self.tg_id,
            "name": self.name,
            "state": self.state,
            "auth": self.auth,
        }