o 4j@sddlmZmZddlmZmZddlmZddlm Z ddl m Z m Z m Z ddlmZeeBdBZdZe d d Gd d d ZGd ddeZGdddeZGdddeZGdddeZdS))ABCabstractmethod)AsyncGeneratorMapping)asynccontextmanager) dataclass)AnyLiteraloverload)StateNdefaultT)frozenc@sReZdZUeed<eed<eed<dZedBed<dZedBed<eZ eed<dS) StorageKeybot_idchat_iduser_idN thread_idbusiness_connection_iddestiny) __name__ __module__ __qualname__int__annotations__rrstrDEFAULT_DESTINYrrrU/var/www/kodo/Anonymous/send/lib/python3.10/site-packages/aiogram/fsm/storage/base.pyrs rc@s6eZdZdZe d dededdBdefddZdS) 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)selfrr rrrbuildszKeyBuilder.buildN) rrr__doc__rrr rr'rrrrrs rc @sbeZdZdZdddddddededed ed ed d f d dZ ddededd Bd 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_destinyr.r/r0r1r2r%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&r.r/r0r1r2rrr__init__5s  zDefaultKeyBuilder.__init__rr 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.)r.r0appendrrr1rrrrr2rr ValueErrorr/join)r&rr parts error_messagerrrr'Ks$    zDefaultKeyBuilder.buildr() rrrr)rboolr3rr r'rrrrr**s8   r*c @seZdZdZeddededdfddZedededBfdd Z eded e ee fddfd d Z edede ee ffd dZededede dBfddZededede de fddZ ddedede dBde dBfddZded e ee fde ee ffddZedddZdS) BaseStoragez) Base class for all FSM storages Nrr#r%cdS)zn Set state for specified key :param key: storage key :param state: new state Nr)r&rr#rrr set_statekzBaseStorage.set_statecr;)z_ Get key state :param key: storage key :return: current state Nrr&rrrr get_statetr=zBaseStorage.get_stater"cr;)ze Write data (replace) :param key: storage key :param data: new data Nr)r&rr"rrrset_data}r=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&rBrCrrr get_valuer=zBaseStorage.get_valuer cr;)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&rBrCr rrrrDr=cs||IdH}|||Sr()rAget)r&rBrCr r"rrrrDs 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 )rN)rr")rAupdater@copy)r&rr" current_datarrr update_datas  zBaseStorage.update_datacr;)zC Close storage (database connection, file or etc.) Nrr&rrrcloser=zBaseStorage.closer(r%N)rrrr)rr StateTyper<rr?rrr@dictrAr rDrIrKrrrrr:fs6   &  r:c@s8eZdZeedededfddZed ddZdS) BaseEventIsolationrr%)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(rrJrrrrKszBaseEventIsolation.closerL) rrrrrrrr$rKrrrrrOs  rO)abcrrcollections.abcrr contextlibr dataclassesrtypingrr r aiogram.fsm.stater rrMrrrr*r:rOrrrrs     <[