#!/usr/bin/env python # # A library that provides a Python interface to the Telegram Bot API # Copyright (C) 2015-2021 # Leandro Toledo de Souza # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Lesser Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Lesser Public License for more details. # # You should have received a copy of the GNU Lesser Public License # along with this program. If not, see [http://www.gnu.org/licenses/]. """Base class for Telegram Objects.""" try: import ujson as json except ImportError: import json # type: ignore[no-redef] import warnings from typing import TYPE_CHECKING, List, Optional, Tuple, Type, TypeVar from telegram.utils.types import JSONDict from telegram.utils.deprecate import set_new_attribute_deprecated if TYPE_CHECKING: from telegram import Bot TO = TypeVar('TO', bound='TelegramObject', covariant=True) class TelegramObject: """Base class for most Telegram objects.""" _id_attrs: Tuple[object, ...] = () # Adding slots reduces memory usage & allows for faster attribute access. # Only instance variables should be added to __slots__. # We add __dict__ here for backward compatibility & also to avoid repetition for subclasses. __slots__ = ('__dict__',) def __str__(self) -> str: return str(self.to_dict()) def __getitem__(self, item: str) -> object: return getattr(self, item, None) def __setattr__(self, key: str, value: object) -> None: set_new_attribute_deprecated(self, key, value) @staticmethod def _parse_data(data: Optional[JSONDict]) -> Optional[JSONDict]: return None if data is None else data.copy() @classmethod def de_json(cls: Type[TO], data: Optional[JSONDict], bot: 'Bot') -> Optional[TO]: """Converts JSON data to a Telegram object. Args: data (Dict[:obj:`str`, ...]): The JSON data. bot (:class:`telegram.Bot`): The bot associated with this object. Returns: The Telegram object. """ data = cls._parse_data(data) if data is None: return None if cls == TelegramObject: return cls() return cls(bot=bot, **data) # type: ignore[call-arg] @classmethod def de_list(cls: Type[TO], data: Optional[List[JSONDict]], bot: 'Bot') -> List[Optional[TO]]: """Converts JSON data to a list of Telegram objects. Args: data (Dict[:obj:`str`, ...]): The JSON data. bot (:class:`telegram.Bot`): The bot associated with these objects. Returns: A list of Telegram objects. """ if not data: return [] return [cls.de_json(d, bot) for d in data] def to_json(self) -> str: """Gives a JSON representation of object. Returns: :obj:`str` """ return json.dumps(self.to_dict()) def to_dict(self) -> JSONDict: """Gives representation of object as :obj:`dict`. Returns: :obj:`dict` """ data = {} # We want to get all attributes for the class, using self.__slots__ only includes the # attributes used by that class itself, and not its superclass(es). Hence we get its MRO # and then get their attributes. The `[:-2]` slice excludes the `object` class & the # TelegramObject class itself. attrs = {attr for cls in self.__class__.__mro__[:-2] for attr in cls.__slots__} for key in attrs: if key == 'bot' or key.startswith('_'): continue value = getattr(self, key, None) if value is not None: if hasattr(value, 'to_dict'): data[key] = value.to_dict() else: data[key] = value if data.get('from_user'): data['from'] = data.pop('from_user', None) return data def __eq__(self, other: object) -> bool: if isinstance(other, self.__class__): if self._id_attrs == (): warnings.warn( f"Objects of type {self.__class__.__name__} can not be meaningfully tested for" " equivalence." ) if other._id_attrs == (): warnings.warn( f"Objects of type {other.__class__.__name__} can not be meaningfully tested" " for equivalence." ) return self._id_attrs == other._id_attrs return super().__eq__(other) # pylint: disable=no-member def __hash__(self) -> int: if self._id_attrs: return hash((self.__class__, self._id_attrs)) # pylint: disable=no-member return super().__hash__()