o ZzjOJ @sdZddlZddlmZddlmZmZmZddlm Z m Z ddl m Z m Z mZer0ddlmZGdd d e eeefeeefeeeffZdS) z/This module contains the DictPersistence class.N)deepcopy) TYPE_CHECKINGAnycast)BasePersistencePersistenceInput)CDCDataConversationDictConversationKey)JSONDictcs0eZdZdZdZ       dUdedBdeded ed ed ed effd d Ze de e e e e ffdBfddZ e defddZe de e e e e ffdBfddZe defddZe de e e fdBfddZe defddZe dedBfddZe defddZe de eefdBfd d!Ze defd"d#Zde e e eefffd$d%Zde e e eefffd&d'Zde eeffd(d)ZdedBfd*d+Zd,edefd-d.Zd,ed/ed0edBddfd1d2Z d3e d4e e e fddfd5d6Z!d7e d4e e e fddfd8d9Z"d4e e e fddfd:d;Z#d4eddfdd?Z%d3e ddfd@dAZ&d3e dBe e e fddfdCdDZ'd7e dEe e e fddfdFdGZ(dHe e e fddfdIdJZ)dVdKdLZ*e+dMe eefdefdNdOZ,e+dPede eeffdQdRZ-e+d4ede e e eefffdSdTZ.Z/S)WDictPersistenceay Using Python's :obj:`dict` and :mod:`json` for making your bot persistent. Attention: The interface provided by this class is intended to be accessed exclusively by :class:`~telegram.ext.Application`. Calling any of the methods below manually might interfere with the integration of persistence into :class:`~telegram.ext.Application`. Note: * Data managed by :class:`DictPersistence` is in-memory only and will be lost when the bot shuts down. This is, because :class:`DictPersistence` is mainly intended as starting point for custom persistence classes that need to JSON-serialize the stored data before writing them to file/database. * This implementation of :class:`BasePersistence` does not handle data that cannot be serialized by :func:`json.dumps`. .. seealso:: :wiki:`Making Your Bot Persistent ` .. versionchanged:: 20.0 The parameters and attributes ``store_*_data`` were replaced by :attr:`store_data`. Args: store_data (:class:`~telegram.ext.PersistenceInput`, optional): Specifies which kinds of data will be saved by this persistence instance. By default, all available kinds of data will be saved. user_data_json (:obj:`str`, optional): JSON string that will be used to reconstruct user_data on creating this persistence. Default is ``""``. chat_data_json (:obj:`str`, optional): JSON string that will be used to reconstruct chat_data on creating this persistence. Default is ``""``. bot_data_json (:obj:`str`, optional): JSON string that will be used to reconstruct bot_data on creating this persistence. Default is ``""``. conversations_json (:obj:`str`, optional): JSON string that will be used to reconstruct conversation on creating this persistence. Default is ``""``. callback_data_json (:obj:`str`, optional): JSON string that will be used to reconstruct callback_data on creating this persistence. Default is ``""``. .. versionadded:: 13.6 update_interval (:obj:`int` | :obj:`float`, optional): The :class:`~telegram.ext.Application` will update the persistence in regular intervals. This parameter specifies the time (in seconds) to wait between two consecutive runs of updating the persistence. Defaults to 60 seconds. .. versionadded:: 20.0 Attributes: store_data (:class:`~telegram.ext.PersistenceInput`): Specifies which kinds of data will be saved by this persistence instance. ) _bot_data_bot_data_json_callback_data_callback_data_json _chat_data_chat_data_json_conversations_conversations_json _user_data_user_data_jsonN< store_datauser_data_jsonchat_data_json bot_data_jsonconversations_jsoncallback_data_jsonupdate_intervalc sJtj||dd|_d|_d|_d|_d|_d|_d|_d|_ d|_ d|_ |rGz | ||_||_Wnt tfyF}ztd|d}~ww|rhz | ||_||_Wnt tfyg}ztd|d}~ww|rz t||_||_ Wnt tfy}ztd|d}~wwt|jtstd|rzt|} Wnt tfy}ztd|d}~wwz| durd|_ntddd | d D| d f|_||_ Wnt tfy}ztd |d}~ww|jdurtd d|jd Drt|jd tstd |r#z |||_||_ WdSt tfy"}ztd|d}~wwdS)N)rrz4Unable to deserialize user_data_json. Not valid JSONz4Unable to deserialize chat_data_json. Not valid JSONz3Unable to deserialize bot_data_json. Not valid JSONz%bot_data_json must be serialized dictz8Unable to deserialize callback_data_json. Not valid JSONrcSs g|] \}}}|t||fqS)float).0onetwothreer r X/var/www/kodo/tsheker/venv/lib/python3.10/site-packages/telegram/ext/_dictpersistence.py s z,DictPersistence.__init__..rz0callback_data_json is not in the required formatcss,|]}t|dtot|dtVqdS)rN) isinstancedictstr)r"entryr r r& s  z+DictPersistence.__init__..z8Unable to deserialize conversations_json. Not valid JSON)super__init__rrr rrrrrrr _decode_user_chat_data_from_json ValueErrorAttributeError TypeErrorjsonloadsr*r+r IndexErrorall_decode_conversations_from_json) selfrrrrrrrexcdata __class__r r&r0^s                zDictPersistence.__init__returncC|jS)z%:obj:`dict`: The user_data as a dict.)rr:r r r& user_datazDictPersistence.user_datacC|jr|jSt|jS)z6:obj:`str`: The user_data serialized as a JSON-string.)rr5dumpsrBrAr r r&r zDictPersistence.user_data_jsoncCr@)z%:obj:`dict`: The chat_data as a dict.)rrAr r r& chat_datarCzDictPersistence.chat_datacCrD)z6:obj:`str`: The chat_data serialized as a JSON-string.)rr5rErGrAr r r&rrFzDictPersistence.chat_data_jsoncCr@)z$:obj:`dict`: The bot_data as a dict.)r rAr r r&bot_datarCzDictPersistence.bot_datacCrD)z5:obj:`str`: The bot_data serialized as a JSON-string.)rr5rErHrAr r r&rrFzDictPersistence.bot_data_jsoncCr@)ztuple[list[tuple[:obj:`str`, :obj:`float`, dict[:obj:`str`, :class:`object`]]], dict[:obj:`str`, :obj:`str`]]: The metadata on the stored callback data. .. versionadded:: 13.6 )rrAr r r& callback_dataszDictPersistence.callback_datacCrD)zo:obj:`str`: The metadata on the stored callback data as a JSON-string. .. versionadded:: 13.6 )rr5rErIrAr r r&rs z"DictPersistence.callback_data_jsoncCr@)z):obj:`dict`: The conversations as a dict.)rrAr r r& conversationsrCzDictPersistence.conversationscCs*|jr|jS|jr||jSt|jS)z::obj:`str`: The conversations serialized as a JSON-string.)rrJ_encode_conversations_to_jsonr5rErAr r r&rs   z"DictPersistence.conversations_jsonc|jdur i|_t|jS)zReturns the user_data created from the ``user_data_json`` or an empty :obj:`dict`. Returns: :obj:`dict`: The restored user data. N)rBrrrAr r r& get_user_data  zDictPersistence.get_user_datacrL)zReturns the chat_data created from the ``chat_data_json`` or an empty :obj:`dict`. Returns: :obj:`dict`: The restored chat data. N)rGrrrAr r r& get_chat_datarNzDictPersistence.get_chat_datacrL)zReturns the bot_data created from the ``bot_data_json`` or an empty :obj:`dict`. Returns: :obj:`dict`: The restored bot data. N)rHr rrAr r r& get_bot_datarNzDictPersistence.get_bot_datacs |jdur d|_dSt|jS)adReturns the callback_data created from the ``callback_data_json`` or :obj:`None`. .. versionadded:: 13.6 Returns: tuple[list[tuple[:obj:`str`, :obj:`float`, dict[:obj:`str`, :class:`object`]]], dict[:obj:`str`, :obj:`str`]]: The restored metadata or :obj:`None`, if no data was stored. N)rIrrrAr r r&get_callback_datas   z!DictPersistence.get_callback_datanamecs$|jdur i|_|j|iS)zReturns the conversations created from the ``conversations_json`` or an empty :obj:`dict`. Returns: :obj:`dict`: The restored conversations data. N)rJrgetcopy)r:rRr r r&get_conversationss z!DictPersistence.get_conversationskey new_statecsB|jsi|_|j|i||krdS||j||<d|_dS)aWill update the conversations for the given handler. Args: name (:obj:`str`): The handler's name. key (:obj:`tuple`): The key the state is changed for. new_state (:obj:`tuple` | :class:`object`): The new state for the given key. N)r setdefaultrSr)r:rRrVrWr r r&update_conversation*s  z#DictPersistence.update_conversationuser_idr<c:|jdur i|_|j||krdS||j|<d|_dS)zWill update the user_data (if changed). Args: user_id (:obj:`int`): The user the data might have been changed for. data (:obj:`dict`): The :attr:`telegram.ext.Application.user_data` ``[user_id]``. N)rrSr)r:rZr<r r r&update_user_data;   z DictPersistence.update_user_datachat_idcr[)zWill update the chat_data (if changed). Args: chat_id (:obj:`int`): The chat the data might have been changed for. data (:obj:`dict`): The :attr:`telegram.ext.Application.chat_data` ``[chat_id]``. N)rrSr)r:r^r<r r r&update_chat_dataIr]z DictPersistence.update_chat_datac |j|krdS||_d|_dS)zWill update the bot_data (if changed). Args: data (:obj:`dict`): The :attr:`telegram.ext.Application.bot_data`. N)r rr:r<r r r&update_bot_dataWs   zDictPersistence.update_bot_datacr`)aKWill update the callback_data (if changed). .. versionadded:: 13.6 Args: data (tuple[list[tuple[:obj:`str`, :obj:`float`, dict[:obj:`str`, :class:`object`]]], dict[:obj:`str`, :obj:`str`]]): The relevant data to restore :class:`telegram.ext.CallbackDataCache`. N)rrrar r r&update_callback_databs   z$DictPersistence.update_callback_datac(|jdurdS|j|dd|_dS)zWill delete the specified key from the :attr:`chat_data`. .. versionadded:: 20.0 Args: chat_id (:obj:`int`): The chat id to delete from the persistence. N)rpopr)r:r^r r r&drop_chat_dataq   zDictPersistence.drop_chat_datacrd)zWill delete the specified key from the :attr:`user_data`. .. versionadded:: 20.0 Args: user_id (:obj:`int`): The user id to delete from the persistence. N)rrer)r:rZr r r&drop_user_data~rgzDictPersistence.drop_user_datarBcdS)zDoes nothing. .. versionadded:: 13.6 .. seealso:: :meth:`telegram.ext.BasePersistence.refresh_user_data` Nr )r:rZrBr r r&refresh_user_dataz!DictPersistence.refresh_user_datarGcri)zDoes nothing. .. versionadded:: 13.6 .. seealso:: :meth:`telegram.ext.BasePersistence.refresh_chat_data` Nr )r:r^rGr r r&refresh_chat_datarkz!DictPersistence.refresh_chat_datarHcri)zDoes nothing. .. versionadded:: 13.6 .. seealso:: :meth:`telegram.ext.BasePersistence.refresh_bot_data` Nr )r:rHr r r&refresh_bot_datarkz DictPersistence.refresh_bot_datacri)zvDoes nothing. .. versionadded:: 20.0 .. seealso:: :meth:`telegram.ext.BasePersistence.flush` Nr rAr r r&flushrkzDictPersistence.flushrJcCsLi}|D]\}}i||<|D] \}}|||t|<qqt|S)afHelper method to encode a conversations dict (that uses tuples as keys) to a JSON-serializable way. Use :meth:`self._decode_conversations_from_json` to decode. Args: conversations (:obj:`dict`): The conversations dict to transform to JSON. Returns: :obj:`str`: The JSON-serialized conversations dict )itemsr5rE)rJtmphandlerstatesrVstater r r&rKs  z-DictPersistence._encode_conversations_to_json json_stringcCsTt|}i}|D]\}}i||<|D]\}}|||tt|<qq |S)aQHelper method to decode a conversations dict (that uses tuples as keys) from a JSON-string created with :meth:`self._encode_conversations_to_json`. Args: json_string (:obj:`str`): The conversations dict as JSON string. Returns: :obj:`dict`: The conversations dict after decoding )r5r6rotuple)rtrprJrqrrrVrsr r r&r9s z/DictPersistence._decode_conversations_from_jsonc Csvi}t|}|D]-\}}t|}i||<|D]\}}zt|}Wn ty0|}Ynw||||<qq |S)aHelper method to decode chat or user data (that uses ints as keys) from a JSON-string. Args: data (:obj:`str`): The user/chat_data dict as JSON string. Returns: :obj:`dict`: The user/chat_data defaultdict after decoding )r5r6rointr2) r<rp decoded_datauserrB int_user_idrVvalue_idr r r&r1s   z0DictPersistence._decode_user_chat_data_from_json)Nrrrrrr)r?N)0__name__ __module__ __qualname____doc__ __slots__rr,r!r0propertyr+rvrrBrrGrrHrrrIrr rJrobjectrMrOrPrQrUr rYr\r_rbrcrfrhrjrlrmrn staticmethodrKr9r1 __classcell__r r r=r&r s0O$$          ,r )rr5rTrtypingrrr telegram.extrrtelegram.ext._utils.typesrr r telegram._utils.typesr r+r r r r r&s  6