iOJ dZddlZddlmZddlmZmZmZddlm Z m Z ddl m Z m Z mZerddlmZGdd e eeefeeefeeeffZdS) z/This module contains the DictPersistence class.N)deepcopy) TYPE_CHECKINGAnycast)BasePersistencePersistenceInput)CDCDataConversationDictConversationKey)JSONDictcLeZdZdZdZ d7dedzdeded ed ed ed effd Ze de e e e e ffdzfdZ e defdZe de e e e e ffdzfdZe defdZe de e e fdzfdZe defdZe dedzfdZe defdZe de eefdzfdZe defdZde e e eefffdZde e e eefffdZde eeffdZdedzfdZdedefdZdeded edzddfd!Z d"e d#e e e fddfd$Z!d%e d#e e e fddfd&Z"d#e e e fddfd'Z#d#eddfd(Z$d%e ddfd)Z%d"e ddfd*Z&d"e d+e e e fddfd,Z'd%e d-e e e fddfd.Z(d/e e e fddfd0Z)d8d1Z*e+d2e eefdefd3Z,e+d4ede eeffd5Z-e+d#ede e e eefffd6Z.xZ/S)9DictPersistenceay 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_intervalct||d|_d|_d|_d|_d|_d|_d|_d|_ d|_ d|_ |rL | ||_||_n)#ttf$r}td|d}~wwxYw|rL | ||_||_n)#ttf$r}td|d}~wwxYw|rt t!j||_||_ n)#ttf$r}td|d}~wwxYwt%|jt&std|r t!j|} n)#ttf$r}td|d}~wwxYw | d|_n-t)dd| d D| d f|_||_ n)#tt*f$r}td |d}~wwxYw|jSt-d |jd Dr t%|jd t&std |rM |||_||_ dS#ttf$r}td |d}~wwxYwdS)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 c:g|]\}}}|t||fS)float).0onetwothrees ]/Users/shanyulin/my-agent/.venv/lib/python3.11/site-packages/telegram/ext/_dictpersistence.py z,DictPersistence.__init__..s+RRRsC3c E2RRRrz0callback_data_json is not in the required formatc3K|]:}t|dtot|dtV;dS)rN) isinstancedictstr)r&entrys r* z+DictPersistence.__init__..sWuQx..L:eAh3L3Lr,z8Unable to deserialize conversations_json. Not valid JSON)super__init__rrrrrrrrrr _decode_user_chat_data_from_json ValueErrorAttributeError TypeErrorjsonloadsr0r1r IndexErrorall_decode_conversations_from_json) selfrrrrrr r!excdata __class__s r*r6zDictPersistence.__init__^s JPPP""+/+/*./3 /3  a a"&"G"G"W"W'5$$/ a a a VWW]`` a  a a"&"G"G"W"W'5$$/ a a a VWW]`` a  I `!%M!:!:&3##/ ` ` ` UVV\__ `dnd33 I GHHH  T z"455/   N  ]<*.D''*.!RR$q'RRRTXYZT[\++D',>(( + ] ] ] RSSY\\ ]".!%!4Q!7/ "$"5a"8$?? / RSSS   &*&J&JK]&^&^#+=(((/   N    s.!BB6!B11B6G??H%H  H%!J((K9K  Kreturnc|jS)z%:obj:`dict`: The user_data as a dict.)rr@s r* user_datazDictPersistence.user_data r,cP|jr|jStj|jS)z6:obj:`str`: The user_data serialized as a JSON-string.)rr;dumpsrGrFs r*rzDictPersistence.user_data_json*   (' 'z$.)))r,c|jS)z%:obj:`dict`: The chat_data as a dict.)rrFs r* chat_datazDictPersistence.chat_datarHr,cP|jr|jStj|jS)z6:obj:`str`: The chat_data serialized as a JSON-string.)rr;rJrMrFs r*rzDictPersistence.chat_data_jsonrKr,c|jS)z$:obj:`dict`: The bot_data as a dict.)rrFs r*bot_datazDictPersistence.bot_datas ~r,cP|jr|jStj|jS)z5:obj:`str`: The bot_data serialized as a JSON-string.)rr;rJrPrFs r*rzDictPersistence.bot_data_jsons*   '& &z$-(((r,c|jS)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 )rrFs r* callback_datazDictPersistence.callback_datas ""r,cP|jr|jStj|jS)zo:obj:`str`: The metadata on the stored callback data as a JSON-string. .. versionadded:: 13.6 )rr;rJrSrFs r*r z"DictPersistence.callback_data_jsons+  # ,+ +z$,---r,c|jS)z):obj:`dict`: The conversations as a dict.)rrFs r* conversationszDictPersistence.conversationss ""r,c|jr|jS|jr||jStj|jS)z::obj:`str`: The conversations serialized as a JSON-string.)rrV_encode_conversations_to_jsonr;rJrFs r*rz"DictPersistence.conversations_jsonsN  # ,+ +   J55d6HII Iz$,---r,cJK|ji|_t|jS)zReturns the user_data created from the ``user_data_json`` or an empty :obj:`dict`. Returns: :obj:`dict`: The restored user data. )rGrrrFs r* get_user_datazDictPersistence.get_user_data' > ! DO'''r,cJK|ji|_t|jS)zReturns the chat_data created from the ``chat_data_json`` or an empty :obj:`dict`. Returns: :obj:`dict`: The restored chat data. )rMrrrFs r* get_chat_datazDictPersistence.get_chat_datar[r,cJK|ji|_t|jS)zReturns the bot_data created from the ``bot_data_json`` or an empty :obj:`dict`. Returns: :obj:`dict`: The restored bot data. )rPrrrFs r* get_bot_datazDictPersistence.get_bot_datas' = DN &&&r,cNK|j 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)rSrrrFs r*get_callback_dataz!DictPersistence.get_callback_datas/   %"&D 4*+++r,namec|K|ji|_|j|iS)zReturns the conversations created from the ``conversations_json`` or an empty :obj:`dict`. Returns: :obj:`dict`: The restored conversations data. )rVrgetcopy)r@rbs r*get_conversationsz!DictPersistence.get_conversationss>   %"$D !%%dB//44666r,key new_statecK|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 setdefaultrdr)r@rbrgrhs r*update_conversationz#DictPersistence.update_conversation*sl" %"$D    ) )$ 3 3 7 7 < < I I F)2D!#&#'   r,user_idrBcK|ji|_|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)rrdr)r@rlrBs r*update_user_dataz DictPersistence.update_user_data;Q ? " DO ?  w ' '4 / / F#' #r,chat_idcK|ji|_|j||krdS||j|<d|_dS)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)rrdr)r@rprBs r*update_chat_dataz DictPersistence.update_chat_dataIror,c@K|j|krdS||_d|_dS)zWill update the bot_data (if changed). Args: data (:obj:`dict`): The :attr:`telegram.ext.Application.bot_data`. N)rrr@rBs r*update_bot_datazDictPersistence.update_bot_dataWs/ >T ! ! F"r,c@K|j|krdS||_d|_dS)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)rrrts r*update_callback_dataz$DictPersistence.update_callback_databs1  $ & & F"#'   r,c`K|jdS|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@rps r*drop_chat_datazDictPersistence.drop_chat_dataq; ? " F GT***#r,c`K|jdS|j|dd|_dS)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)rryr)r@rls r*drop_user_datazDictPersistence.drop_user_data~r{r,rGc KdS)zDoes nothing. .. versionadded:: 13.6 .. seealso:: :meth:`telegram.ext.BasePersistence.refresh_user_data` Nr$)r@rlrGs r*refresh_user_dataz!DictPersistence.refresh_user_data r,rMc KdS)zDoes nothing. .. versionadded:: 13.6 .. seealso:: :meth:`telegram.ext.BasePersistence.refresh_chat_data` Nr$)r@rprMs r*refresh_chat_dataz!DictPersistence.refresh_chat_datarr,rPc KdS)zDoes nothing. .. versionadded:: 13.6 .. seealso:: :meth:`telegram.ext.BasePersistence.refresh_bot_data` Nr$)r@rPs r*refresh_bot_dataz DictPersistence.refresh_bot_datarr,c KdS)zvDoes nothing. .. versionadded:: 20.0 .. seealso:: :meth:`telegram.ext.BasePersistence.flush` Nr$rFs r*flushzDictPersistence.flushrr,rVci}|D]A\}}i||<|D]"\}}|||tj|<#Btj|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 )itemsr;rJ)rVtmphandlerstatesrgstates r*rXz-DictPersistence._encode_conversations_to_jsonsx$&,2244 6 6OGVCL$llnn 6 6 U05G TZ__-- 6z#r, json_stringc tj|}i}|D]N\}}i||<|D]/\}}|||ttj|<0O|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 )r;r<rtuple)rrrVrrrgrs r*r?z/DictPersistence._decode_conversations_from_jsonsj%%57 "yy{{ G GOGV%'M' "$llnn G G UAF g&uTZ__'='=>> Gr,ci}tj|}|D]a\}}t|}i||<|D]3\}} t|}n#t$r|}YnwxYw||||<4b|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 )r;r<rintr8) rBr decoded_datauserrG int_user_idrgvalue_ids r*r7z0DictPersistence._decode_user_chat_data_from_jsons02z$'' +1133 . .OD)d))K!C 'oo// . . Uc((CC!CCC(-K %%  .  sA.. A=<A=)Nrrrrrr)rDN)0__name__ __module__ __qualname____doc__ __slots__rr2r%r6propertyr1rrrGrrMrrPrr rSr r rVrobjectrZr]r_rarfr rkrnrrrurwrzr}rrrr staticmethodrXr?r7 __classcell__)rCs@r*rr s..` I/3  "$"$!#MM$t+MM M  M  M MMMMMMM^4T#s(^ 34t;X****X* 4T#s(^ 34t;X****X* $sCx.4/X)s)))X) #w~###X#.C...X.#tC)9$9:TA###X#.C...X.(T#tFFN/C*C%D(((((T#tFFN/C*C%D(((('D$8'''' ,4 , , , , 7C 74D 7 7 7 7((-(:@4-( ((((" $c $c3h $D $ $ $ $ $c $c3h $D $ $ $ $ #$sCx. #T # # # # (w (4 ( ( ( ( $C $D $ $ $ $ $C $D $ $ $ $ s tCH~ RV     s tCH~ RV     tCH~ $        T#?O:O5PUX\"ST#GWBW=X\$stCffnAUrs&65 ++++++++++::::::::PPPPPPPPPP/......BBBBBod38nd38nd3PS8n&TUBBBBBr,