o Fhg@sddlmZmZddlmZddlmZddlmZm Z m Z m Z m Z m Z mZmZddlmZe eeefZdZeddGd d d ZGd d d eZGd ddeZGdddeZGdddeZdS))ABCabstractmethod)asynccontextmanager) dataclass)AnyAsyncGeneratorDictLiteralMappingOptionalUnionoverload)StatedefaultT)frozenc@sReZdZUeed<eed<eed<dZeeed<dZeeed<e Z eed<dS) StorageKeybot_idchat_iduser_idN thread_idbusiness_connection_iddestiny) __name__ __module__ __qualname__int__annotations__rr rstrDEFAULT_DESTINYrrrK/var/www/html/venv/lib/python3.10/site-packages/aiogram/fsm/storage/base.pyrs rc@s6eZdZdZe d dedeeddefddZ dS) KeyBuilderzBase class for key builder.NkeypartdatastatelockreturncCsdS)z Build key to be used in storage's db queries :param key: contextual key :param part: part of the record :return: key to be used in storage's db queries Nr)selfr"r#rrr build#s zKeyBuilder.buildN) rrr__doc__rrr r rr*rrrr r! s r!c @sbeZdZdZdddddddededed ed ed d f d dZ ddedee dd efddZ d S)DefaultKeyBuildera$ Simple key builder with default prefix. Generates a colon-joined string with prefix, chat_id, user_id, optional bot_id, business_connection_id, destiny and field. Format: :code:`::::::` fsm:Fprefix separator with_bot_idwith_business_connection_id with_destinyr1r2r3r4r5r(NcCs"||_||_||_||_||_dS)a  :param prefix: prefix for all records :param separator: separator :param with_bot_id: include Bot id in the key :param with_business_connection_id: include business connection id :param with_destiny: include destiny key Nr0)r)r1r2r3r4r5rrr __init__>s  zDefaultKeyBuilder.__init__r"r#r$cCs|jg}|jr|t|j|jr|jr|t|j|t|j|jr0|t|j|t|j |j rB||j n |j t krMd}t ||rT|||j|S)NzDefault key builder is not configured to use key destiny other than the default. Probably, you should set `with_destiny=True` in for DefaultKeyBuilder.)r1r3appendrrr4rrrrr5rr ValueErrorr2join)r)r"r#parts error_messagerrr r*Ts$    zDefaultKeyBuilder.buildr+) rrrr,rboolr6rr r r*rrrr r-3s8   r-c @seZdZdZeddededdfddZededee fdd Z eded e e e fddfd d Z ededee e ffd dZedede dee fddZedede de de fddZ ddede dee dee fddZded e e e fdee e ffddZedddZdS) BaseStoragez) Base class for all FSM storages Nr"r&r(cdS)zn Set state for specified key :param key: storage key :param state: new state Nr)r)r"r&rrr set_statetzBaseStorage.set_statecr>)z_ Get key state :param key: storage key :return: current state Nrr)r"rrr get_state~r@zBaseStorage.get_stater%cr>)ze Write data (replace) :param key: storage key :param data: new data Nr)r)r"r%rrr set_datar@zBaseStorage.set_datacr>)zi Get current data for key :param key: storage key :return: current data NrrArrr get_datar@zBaseStorage.get_data storage_keydict_keycr>)z Get single value from data by key :param storage_key: storage key :param dict_key: value key :return: value stored in key of dict or ``None`` Nr)r)rErFrrr get_values zBaseStorage.get_valuercr>)z Get single value from data by key :param storage_key: storage key :param dict_key: value key :param default: default value to return :return: value stored in key of dict or default Nr)r)rErFrrrr rGs cs||IdH}|||Sr+)rDget)r)rErFrr%rrr rGs cs:|j|dIdH}|||j||dIdH|S)z Update date in the storage for key (like dict.update) :param key: storage key :param data: partial data :return: new data )r"N)r"r%)rDupdaterCcopy)r)r"r% current_datarrr update_datas  zBaseStorage.update_datacr>)zC Close storage (database connection, file or etc.) Nrr)rrr closeszBaseStorage.closer+r(N)rrrr,rr StateTyper?r rrBr rrCrrDr rGrLrNrrrr r=os6       & r=c@s8eZdZeedededfddZed ddZdS) BaseEventIsolationr"r()NNcCs dVdS)z Isolate events with lock. Will be used as context manager :param key: storage key :return: An async generator NrrArrr r's zBaseEventIsolation.lockNcsdSr+rrMrrr rNszBaseEventIsolation.closerO) rrrrrrrr'rNrrrr rQs  rQN)abcrr contextlibr dataclassesrtypingrrrr r r r r aiogram.fsm.staterrrPrrr!r-r=rQrrrr s  (  <_