dhgddlmZmZddlmZddlmZddlmZm Z m Z m Z m Z m Z mZmZddlmZe eeefZdZedGd d ZGd d eZGd deZGddeZGddeZy))ABCabstractmethod)asynccontextmanager) dataclass)AnyAsyncGeneratorDictLiteralMappingOptionalUnionoverload)StatedefaultT)frozencbeZdZUeed<eed<eed<dZeeed<dZeeed<e Z eed<y) StorageKeybot_idchat_iduser_idN thread_idbusiness_connection_iddestiny) __name__ __module__ __qualname__int__annotations__rr rstrDEFAULT_DESTINYrR/var/www/netwell/bot/venv/lib/python3.12/site-packages/aiogram/fsm/storage/base.pyrrs8 K L L#Ix}#,0HSM0"GS"r"rc>eZdZdZe ddedeeddefdZ y) KeyBuilderzBase class for key builder.Nkeypartdatastatelockreturncy)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's r#buildzKeyBuilder.build#s r"N) rrr__doc__rrr r rr/r!r"r#r%r% sD%<@    w678      r"r%c beZdZdZdddddddededed ed ed d f d Z ddedee dd efdZ y )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_destinyr7r8r9r:r;r,NcJ||_||_||_||_||_y)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 Nr6)r.r7r8r9r:r;s r#__init__zDefaultKeyBuilder.__init__>s*  "&+F((r"r&r'r(c|jg}|jr$|jt|j|j r0|j r$|jt|j |jt|j|jr$|jt|j|jt|j|jr|j|jn |jtk7r d}t||r|j||jj|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.)r7r9appendrrr:rrrrr;rr ValueErrorr8join)r.r&r'parts error_messages r#r/zDefaultKeyBuilder.buildTs      LLSZZ )  + +0J0J LLS778 9 S%& == LLS]]+ , S%&    LL % [[O +] ]+ +  LL ~~""5))r"r0) rrrr1rboolr=rr r r/r!r"r#r3r33s!,1")) )  ) &* )) )2<@* *w678* *r"r3c DeZdZdZeddededdfdZededee fdZ edede e e fddfd Z ededee e ffd Zed ed e dee fd Zed ed e de de fdZ dd ed e dee dee fdZdede e e fdee e ffdZeddZy) BaseStoragez) Base class for all FSM storages Nr&r*r,c Kyw)zn Set state for specified key :param key: storage key :param state: new state Nr!)r.r&r*s r# set_statezBaseStorage.set_statet  c Kyw)z_ Get key state :param key: storage key :return: current state Nr!r.r&s r# get_statezBaseStorage.get_state~rIrJr)c Kyw)ze Write data (replace) :param key: storage key :param data: new data Nr!)r.r&r)s r#set_datazBaseStorage.set_datarIrJc Kyw)zi Get current data for key :param key: storage key :return: current data Nr!rLs r#get_datazBaseStorage.get_datarIrJ storage_keydict_keyc Kyw)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.rRrSs r# get_valuezBaseStorage.get_values  rJrc Kyw)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.rRrSrs r#rUzBaseStorage.get_values  rJcdK|j|d{}|j||S7wr0)rQget)r.rRrSrr)s r#rUzBaseStorage.get_values/]];//xx'**0s 0.0cK|j|d{}|j||j||d{|jS7@7w)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))rQupdaterOcopy)r.r&r) current_datas r# update_datazBaseStorage.update_datasW"]]s]33 D!mm,m777  ""47s!AA+AAAAc Kyw)zC Close storage (database connection, file or etc.) Nr!r.s r#closezBaseStorage.closes rJr0r,N)rrrr1rr StateTyperHr rrMr rrOr rQrrUr]r`r!r"r#rFrFos : i 4   : (3-   * GCH4E $   * c3h   :  RU    :    s  WZ    PT+%+14+?G}+ #+ #Z #wsCx7H #TRUWZRZ^ #  r"rFcHeZdZeedededfdZeddZy)BaseEventIsolationr&r,)NNcKdyw)z Isolate events with lock. Will be used as context manager :param key: storage key :return: An async generator Nr!rLs r#r+zBaseEventIsolation.locks  s Nc Kywr0r!r_s r#r`zBaseEventIsolation.closes  rJra) rrrrrrrr+r`r!r"r#rdrdsCj^J-G  r"rdN)abcrr contextlibr dataclassesrtypingrrr r r r r raiogram.fsm.staterrrbr rr%r3rFrdr!r"r#rls#*!   $ U3:& '  $###  &9* 9*x\ #\ ~  r"