jddlmZmZddlmZmZddlmZddlm Z ddl m Z m Z m Z ddlmZeezdzZdZe d Gd d ZGd deZGddeZGddeZGddeZy))ABCabstractmethod)AsyncGeneratorMapping)asynccontextmanager) dataclass)AnyLiteraloverload)StateNdefaultT)frozencbeZdZUeed<eed<eed<dZedzed<dZedzed<eZ eed<y) StorageKeybot_idchat_iduser_idN thread_idbusiness_connection_iddestiny) __name__ __module__ __qualname__int__annotations__rrstrDEFAULT_DESTINYrK/var/www/html/venv/lib/python3.12/site-packages/aiogram/fsm/storage/base.pyrrs8 K L L IsTz )-C$J-"GS"rrc>eZdZdZe ddededdzdefdZy) 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.buildsrN) rrr__doc__rrr rr,rrr r"r"sC%9=    -.5      rr"c beZdZdZdddddddededed ed ed d f d Z ddededd zd 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_destinyr4r5r6r7r8r)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 Nr3)r+r4r5r6r7r8s r __init__zDefaultKeyBuilder.__init__5s*  "&+F((rr#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.)r4r6appendrrr7rrrrr8rr ValueErrorr5join)r+r#r$parts error_messages r r,zDefaultKeyBuilder.buildKs      LLSZZ )  + +0J0J LLS778 9 S%& == LLS]]+ , S%&    LL % [[O +] ]+ +  LL ~~""5))rr-) rrrr.rboolr:rr r,rrr r0r0*s!,1")) )  ) &* )) )29=* *-.5* *rr0c DeZdZdZeddededdfdZedededzfdZ edede ee fddfd Z edede ee ffd Zed ed ede dzfd Zed ed ede de fdZ dd ed ede dzde dzfdZdede ee fde ee 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_statekc Kyw)z_ Get key state :param key: storage key :return: current state Nrr+r#s r get_statezBaseStorage.get_statetrFrGr&c Kyw)ze Write data (replace) :param key: storage key :param data: new data Nr)r+r#r&s r set_datazBaseStorage.set_data}rFrGc Kyw)zi Get current data for key :param key: storage key :return: current data NrrIs r get_datazBaseStorage.get_datarFrG 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+rOrPs r get_valuezBaseStorage.get_valuerFrGr c 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+rOrPr s r rRzBaseStorage.get_valuerFrGcdK|j|d{}|j||S7wr-)rNget)r+rOrPr r&s r rRzBaseStorage.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&)rNupdaterLcopy)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.) Nrr+s r closezBaseStorage.closerFrGr-r)N)rrrr.rr StateTyperErrJrr rLdictrNr rRrZr]rrr rCrCfs : i 4   : #*   * GCH4E $   * c3h   :  t   :  s WZ  # +++t + t + #Z #wsCx7H #TRUWZRZ^ #  rrCcHeZdZeedededfdZeddZy)BaseEventIsolationr#r))NNcKdyw)z Isolate events with lock. Will be used as context manager :param key: storage key :return: An async generator NrrIs r r(zBaseEventIsolation.locks  s Nc Kywr-rr\s r r]zBaseEventIsolation.closes  rGr^) rrrrrrrr(r]rrr rbrbsCj^J-G  rrb)abcrrcollections.abcrr contextlibr dataclassesrtypingr r r aiogram.fsm.stater rr_rrr"r0rCrbrrr rks#3*!))# %K$   $###  $9* 9*xX #X v  r