a y{Jh @sddlmZmZddlmZddlmZddlmZm Z m Z m Z m Z ddl mZe e eefZdZeddGd d d ZGd d d eZGd ddeZdS))ABCabstractmethod)asynccontextmanager) dataclass)AnyAsyncGeneratorDictOptionalUnion)StatedefaultT)frozenc@sBeZdZUeed<eed<eed<dZeeed<eZe ed<dS) StorageKeyZbot_idZchat_idZuser_idN thread_iddestiny) __name__ __module__ __qualname__int__annotations__rr DEFAULT_DESTINYrstrrrJ/var/www/shaz/venv/lib/python3.9/site-packages/aiogram/fsm/storage/base.pyr s rc@seZdZdZedeeddddZeeee dddZ eee e e fdd d d Z eee e e fdd d Zee e e fe e e fd ddZeddddZdS) BaseStoragez) Base class for all FSM storages N)keystatereturncsdS)zn Set state for specified key :param key: storage key :param state: new state Nr)selfrrrrr set_stateszBaseStorage.set_staterrcsdS)z_ Get key state :param key: storage key :return: current state Nrrrrrr get_state%szBaseStorage.get_state)rdatarcsdS)ze Write data (replace) :param key: storage key :param data: new data Nr)rrr#rrrset_data/szBaseStorage.set_datacsdS)zi Get current data for key :param key: storage key :return: current data Nrr!rrrget_data9szBaseStorage.get_datacs8|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#)r%updater$copy)rrr#Z current_datarrr update_dataCs zBaseStorage.update_datarcsdS)zC Close storage (database connection, file or etc.) NrrrrrclosePszBaseStorage.close)N)rrr__doc__rr StateTyperr rr"rrr$r%r(r+rrrrrs    " rc@s:eZdZeeeeddddZeddddZdS) BaseEventIsolation)NNr cCs dVdS)z Isolate events with lock. Will be used as context manager :param key: storage key :return: An async generator Nrr!rrrlockYs zBaseEventIsolation.lockNr)csdS)Nrr*rrrr+eszBaseEventIsolation.close) rrrrrrrr/r+rrrrr.Xs  r.N)abcrr contextlibr dataclassesrtypingrrrr r Zaiogram.fsm.stater rr-rrrr.rrrrs   B