o "hJ @sdZddlZddlmZddlmZmZmZmZddl m Z m Z ddl m Z mZmZer2ddlmZGdd d e eeefeeefeeeffZdS) z/This module contains the DictPersistence class.N)deepcopy) TYPE_CHECKINGAnyOptionalcast)BasePersistencePersistenceInput)CDCDataConversationDictConversationKey)JSONDictcs0eZdZdZdZ       dUdeededed ed ed ed effd d Z e dee e e e e fffddZe defddZe dee e e e e fffddZe defddZe dee e e ffddZe defddZe deefddZe defddZe dee eeffd d!Ze defd"d#Zde e e eefffd$d%Zde e e eefffd&d'Zde eeffd(d)Zdeefd*d+Zd,edefd-d.Zd,ed/e d0eeddfd1d2Z!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/Z0S)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)rr z4Unable 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 JSONr cSs g|] \}}}|t||fqS)float).0onetwothreer!r!P/var/www/html/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__rrrrrrrrrr _decode_user_chat_data_from_json ValueErrorAttributeError TypeErrorjsonloadsr+r,r IndexErrorall_decode_conversations_from_json) selfrrrrrrr excdata __class__r!r'r1]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.)rr6dumpsrCrBr!r!r'r zDictPersistence.user_data_jsoncCrA)z%:obj:`dict`: The chat_data as a dict.)rrBr!r!r' chat_datarDzDictPersistence.chat_datacCrE)z6:obj:`str`: The chat_data serialized as a JSON-string.)rr6rFrHrBr!r!r'rrGzDictPersistence.chat_data_jsoncCrA)z$:obj:`dict`: The bot_data as a dict.)rrBr!r!r'bot_datarDzDictPersistence.bot_datacCrE)z5:obj:`str`: The bot_data serialized as a JSON-string.)rr6rFrIrBr!r!r'rrGzDictPersistence.bot_data_jsoncCrA)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 )rrBr!r!r' callback_dataszDictPersistence.callback_datacCrE)zo:obj:`str`: The metadata on the stored callback data as a JSON-string. .. versionadded:: 13.6 )rr6rFrJrBr!r!r'rs z"DictPersistence.callback_data_jsoncCrA)z):obj:`dict`: The conversations as a dict.)rrBr!r!r' conversationsrDzDictPersistence.conversationscCs*|jr|jS|jr||jSt|jS)z::obj:`str`: The conversations serialized as a JSON-string.)rrK_encode_conversations_to_jsonr6rFrBr!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)rCrrrBr!r!r' get_user_data  zDictPersistence.get_user_datacrM)zReturns the chat_data created from the ``chat_data_json`` or an empty :obj:`dict`. Returns: :obj:`dict`: The restored chat data. N)rHrrrBr!r!r' get_chat_datarOzDictPersistence.get_chat_datacrM)zReturns the bot_data created from the ``bot_data_json`` or an empty :obj:`dict`. Returns: :obj:`dict`: The restored bot data. N)rIrrrBr!r!r' get_bot_datarOzDictPersistence.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)rJrrrBr!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)rKrgetcopy)r;rSr!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 setdefaultrTr)r;rSrWrXr!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)rrTr)r;r[r=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)rrTr)r;r_r=r!r!r'update_chat_dataHr^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)rrr;r=r!r!r'update_bot_dataVs   zDictPersistence.update_bot_datacra)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)rrrbr!r!r'update_callback_dataas   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_datap   zDictPersistence.drop_chat_datacre)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)rrfr)r;r[r!r!r'drop_user_data}rhzDictPersistence.drop_user_datarCcdS)zDoes nothing. .. versionadded:: 13.6 .. seealso:: :meth:`telegram.ext.BasePersistence.refresh_user_data` Nr!)r;r[rCr!r!r'refresh_user_dataz!DictPersistence.refresh_user_datarHcrj)zDoes nothing. .. versionadded:: 13.6 .. seealso:: :meth:`telegram.ext.BasePersistence.refresh_chat_data` Nr!)r;r_rHr!r!r'refresh_chat_datarlz!DictPersistence.refresh_chat_datarIcrj)zDoes nothing. .. versionadded:: 13.6 .. seealso:: :meth:`telegram.ext.BasePersistence.refresh_bot_data` Nr!)r;rIr!r!r'refresh_bot_datarlz DictPersistence.refresh_bot_datacrj)zvDoes nothing. .. versionadded:: 20.0 .. seealso:: :meth:`telegram.ext.BasePersistence.flush` Nr!rBr!r!r'flushrlzDictPersistence.flushrKcCsLi}|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 )itemsr6rF)rKtmphandlerstatesrWstater!r!r'rLs  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 )r6r7rptuple)rurqrKrrrsrWrtr!r!r'r:s 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 )r6r7rpintr3) r=rq decoded_datauserrC int_user_idrWvalue_idr!r!r'r2s   z0DictPersistence._decode_user_chat_data_from_json)Nrrrrrr)r@N)1__name__ __module__ __qualname____doc__ __slots__rrr-r"r1propertyr,rwrrCrrHrrIrr rJrr rKrobjectrNrPrQrRrVr rZr]r`rcrdrgrirkrmrnro staticmethodrLr:r2 __classcell__r!r!r>r'r s0O$$          ,r )rr6rUrtypingrrrr telegram.extrrtelegram.ext._utils.typesr r r telegram._utils.typesr r,r r!r!r!r's  6