Source code for discord_http.object

from datetime import datetime

from . import utils

__all__ = (
    "PartialBase",
    "Snowflake",
)




class Snowflake:
    """
    A class to represent a Discord Snowflake
    """
    def __init__(self, *, id: int):
        if not isinstance(id, int):
            raise TypeError("id must be an integer")
        self.id: int = id

    def __repr__(self) -> str:
        return f"<Snowflake id={self.id}>"

    def __str__(self) -> str:
        return str(self.id)

    def __int__(self) -> int:
        return self.id

    def __eq__(self, other) -> bool:
        match other:
            case x if isinstance(x, Snowflake):
                return self.id == other.id

            case x if isinstance(x, int):
                return self.id == other

            case _:
                return False

    def __gt__(self, other) -> bool:
        match other:
            case x if isinstance(x, Snowflake):
                return self.id > other.id

            case x if isinstance(x, int):
                return self.id > other

            case _:
                raise TypeError(
                    f"Cannot compare 'Snowflake' to '{type(other).__name__}'"
                )

    def __lt__(self, other) -> bool:
        match other:
            case x if isinstance(x, Snowflake):
                return self.id < other.id

            case x if isinstance(x, int):
                return self.id < other

            case _:
                raise TypeError(
                    f"Cannot compare 'Snowflake' to '{type(other).__name__}'"
                )

    def __ge__(self, other) -> bool:
        match other:
            case x if isinstance(x, Snowflake):
                return self.id >= other.id

            case x if isinstance(x, int):
                return self.id >= other

            case _:
                raise TypeError(
                    f"Cannot compare 'Snowflake' to '{type(other).__name__}'"
                )

    def __le__(self, other) -> bool:
        match other:
            case x if isinstance(x, Snowflake):
                return self.id <= other.id

            case x if isinstance(x, int):
                return self.id <= other

            case _:
                raise TypeError(
                    f"Cannot compare 'Snowflake' to '{type(other).__name__}'"
                )

    @property
    def created_at(self) -> datetime:
        """ `datetime`: The datetime of the snowflake """
        return utils.snowflake_time(self.id)





class PartialBase(Snowflake):
    """
    A base class for partial objects.
    This class is based on the Snowflae class standard,
    but with a few extra attributes.
    """
    def __init__(self, *, id: int):
        super().__init__(id=int(id))

    def __repr__(self) -> str:
        return f"<PartialBase id={self.id}>"



    def is_partial(self) -> bool:
        """
        `bool`: Returns True if the object is partial
        This depends on the class name starting with Partial or not.
        """
        return self.__class__.__name__.startswith("Partial")