soj ddlZddlmZddlmZmZddlmZddlmZm Z ddl m Z ddl m Z  dd ejd e d ed ed edef dZ dd ejd e d eded eddf dZ dd ejd e d ed eddf dZ dd ed ed eddfdZ dd ed ed ed edef dZ dd ed eded eddf dZ dd ed ed ed edef dZ dd ed eded eddf dZd edefdZdS)N)datetime)CallbackContext ContextTypes)phrases)ContextDataKeyContextDataType)CommonChatException)get_random_stringTcontext data_typekey inner_keytolerate_key_exceptionreturnc"|tjur|jn|j} |||d|dS||dS#t$r<}|r3t jd|dttj d|d}~wwxYw)a! Get the bot context data :param context: The context :param data_type: The type :param key: The key :param inner_key: The inner key :param tolerate_key_exception: If the key is not found, raise CommonChatException instead of KeyError :return: The data NvaluezKey zF not found in context user data, this might be due to a system restartT) only_message) rBOTbot_data user_dataKeyErrorloggingdebugr r RESTART_BOT)r r r rrdataes /app/src/utils/context_utils.pyget_context_datar s$ )O,???7  WEVD   9W%i09 99W% %  !  Mbsbbb   &g&9MMM MGsA A B7B  Brc |tjur|jn|j}||vrdii||<|=d|i||d|<t j||d|d<dS|||d<t j||d<dS)z Set the bot context data :param context: The context :param data_type: The type :param key: The key :param value: The value :param inner_key: The inner key :return: None rN last_updated)rrrrrnow)r r r rrrs rset_context_datar"/s" )O,???7  WEVD $bMS )0%(8S '9%8@ S '9%n555"S '$,LNNS .!!!c|tjur|jn|j}||vrdS|#||d|dS||dS)z Remove the bot context data :param context: The context :param data_type: The type :param key: The key :param inner_key: The inner key :return: None Nr)rrrrpop)r r r rrs rremove_context_datar&Msk )O,???7  WEVD $ S 'y)))))  r#c>t|tj||dS)z Remove the user context data :param context: The context :param key: The key :param inner_key: The inner key :return: None N)r&rUSER)r r rs rremove_user_context_datar)gs!!5sIFFFFFr#c<t|tj|||S)a Get the bot context data :param context: The context :param key: The key :param inner_key: The inner key :param tolerate_key_exception: If the key is not found, raise CommonChatException instead of KeyError :return: The data )rrrr r rrs rget_bot_context_datar,us G_%8#yJ` a aar#c@t|tj|||dS)z Set the bot context data :param context: The context :param key: The key :param value: The value :param inner_key: The inner key :return: None N)r"rrr r rrs rset_bot_context_datar/s#Wo13yIIIIIr#c<t|tj|||S)a Get the user context data :param context: The context :param key: The key :param inner_key: The inner key :param tolerate_key_exception: If the key is not found, raise CommonChatException instead of KeyError :return: The data )rrr(r+s rget_user_context_datar1s G_%93 Ka b bbr#c@t|tj|||dS)z Set the user context data :param context: The context :param key: The key :param value: The value :param inner_key: The inner key :return: None N)r"rr(r.s rset_user_context_datar3s#Wo2C JJJJJr#ctdD]B}t} t|tj|d/#t $r|cYcSwxYwt jdtd)zu Get a random inner key for the user context data :param context: The context :return: The inner key dF)rrz;Could not find a random inner key for the user context datazFailed to generate an inner key) ranger r1r INLINE_QUERYrrerrorr )r irs r'get_random_user_context_inner_query_keyr:s3ZZ  %''   !+#',                MOPPP ? @ @@s? AA)NT)N)rrsrc.tg_compat.extrr resourcesrsrc.model.enums.ContextDataKeyrrsrc.model.error.CommonChatErrorr src.utils.download_utilsr DEFAULT_TYPEstrboolanyrr"r&r)r,r/r1r3r:r#rrEs_;;;;;;;;JJJJJJJJ??????666666#'    &      !      P 33  &33 3  3  3  3333D   &    6EI G G  G#1 G>A G  G G G G"#' bb b bb! b  bbbb(QU J J  J#1 J:= JJM J  J J J J$#' cc c cc! c  cccc(QU K K  K#1 K:= KJM K  K K K KA_AAAAAAAr#