o @j3@sddlmZmZddlmZddlmZddlmZm Z m Z m Z m Z m Z mZddlmZe e eefZdZeddGd d d ZGd d d eZGd ddeZGdddeZGdddeZdS))ABCabstractmethod)asynccontextmanager) dataclass)AnyAsyncGeneratorDictLiteralOptionalUnionoverload)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_DESTINYrrrg/var/www/www-root/data/www/ai-bot.pdev.uz/venv/lib/python3.10/site-packages/aiogram/fsm/storage/base.pyr s 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"rrrbuilds zKeyBuilder.buildN) rrr__doc__rrr r rr)rrrrr 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_destinyr0r1r2r3r4r'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 Nr/)r(r0r1r2r3r4rrr__init__5s  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.)r0r2appendrrr3rrrrr4rr ValueErrorr1join)r(r!r"parts error_messagerrrr)Ks$    zDefaultKeyBuilder.buildr*) rrrr+rboolr5rr r r)rrrrr,*s8   r,c @seZdZdZeddededdfddZededee fdd Z eded e e e fddfd d Z edede e e ffd dZedede dee fddZedede de de fddZ ddede dee dee fddZded e e e fde e 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_statekzBaseStorage.set_statecr=)z_ Get key state :param key: storage key :return: current state Nrr(r!rrr get_stateur?zBaseStorage.get_stater$cr=)ze Write data (replace) :param key: storage key :param data: new data Nr)r(r!r$rrrset_datar?zBaseStorage.set_datacr=)zi Get current data for key :param key: storage key :return: current data Nrr@rrrget_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(rDrErrr 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(rDrErrrrrFs cs||IdH}|||Sr*)rCget)r(rDrErr$rrrrFs 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$)rCupdaterBcopy)r(r!r$ current_datarrr update_datas  zBaseStorage.update_datacr=)zC Close storage (database connection, file or etc.) Nrr(rrrcloseszBaseStorage.closer*r'N)rrrr+rr StateTyper>r rrArrrBrCr rFrKrMrrrrr<fs6       & 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 Nrr@rrrr&s zBaseEventIsolation.lockNcsdSr*rrLrrrrMszBaseEventIsolation.closerN) rrrrrrrr&rMrrrrrPs  rPN)abcrr contextlibr dataclassesrtypingrrrr r r r aiogram.fsm.stater rrOrrr r,r<rPrrrrs  $  <_