o 5j@sddlmZddlmZmZddlmZmZz ddlm Z m Z Wne y3dZ Gddde Z Ynwddl mZmZerZdd lmZmZdd lmZdd lmZmZdd lmZGd ddeeZGdddeZGdddeZGdddeZdS)) annotations)ABCabstractmethod) TYPE_CHECKINGAny)LocaleUnknownLocaleErrorNc@s eZdZdS)rN)__name__ __module__ __qualname__r r Z/var/www/kodo/Anonymous/send/lib/python3.10/site-packages/aiogram/utils/i18n/middleware.pyr sr)BaseMiddlewareRouter) AwaitableCallable) FSMContext)TelegramObjectUser)I18nc@sFeZdZdZ  d!d"d d Z d#d$ddZd%ddZed&dd Zd S)'I18nMiddlewarez# Abstract I18n middleware. i18ni18n_middlewareri18n_key str | Nonemiddleware_keystrreturnNonecCs||_||_||_dS)z Create an instance of middleware :param i18n: instance of I18n :param i18n_key: context key for I18n instance :param middleware_key: context key for this middleware Nrrr)selfrrrr r r __init__s  zI18nMiddleware.__init__Nr rrouterrexcludeset[str] | NonecCsD|durt}dh|}|jD]\}}||vrq||q|S)z Register middleware for all events in the Router :param router: :param exclude: :return: Nupdate)set observersitemsouter_middleware)r r"r#exclude_events event_nameobserverr r r setup/s   zI18nMiddleware.setuphandler:Callable[[TelegramObject, dict[str, Any]], Awaitable[Any]]eventrdatadict[str, Any]rc s|j||dIdHp|jj}|jr|j||j<|jr |||j<|j.|j||||IdHWdWdS1sFwYWddS1sVwYdS)Nr0r1) get_localerdefault_localerrcontext use_locale)r r.r0r1current_localer r r __call__Ds  RzI18nMiddleware.__call__csdS)z Detect current user locale based on event and context. **This method must be defined in child classes** :param event: :param data: :return: Nr r r0r1r r r r4TszI18nMiddleware.get_localerrrrrrrrrrN)r rr"rr#r$rr)r.r/r0rr1r2rrr0rr1r2rr) r r r __doc__r!r-r9rr4r r r r rs  rcs2eZdZdZ  ddfd d ZdddZZS)SimpleI18nMiddlewarezc Simple I18n middleware. Chooses language code from the User object received in event rrrrrrrrrcs6tj|||dtdurt|jd}t|dS)Nr can be used only when Babel installed Just install Babel (`pip install Babel`) or aiogram with i18n support (`pip install aiogram[i18n]`))superr!rtyper RuntimeError)r rrrmsg __class__r r r!hs zSimpleI18nMiddleware.__init__r0rr1r2cstdurt|jd}t||d}|dus|jdur#|jjSz tj|jdd}Wn t y:|jjYSwt |g}|j rI| |j |D] }||jjvrW|SqK|jjS)NrAevent_from_user-)sep)rrCr rDget language_coderr5parserr territoryappendlanguageavailable_locales)r r0r1rErHlocale candidates candidater r r r4xs*      zSimpleI18nMiddleware.get_localer;r<r>r r r r?r!r4 __classcell__r r rFr r@as  r@cs2eZdZdZ  ddfd d ZdddZZS)ConstI18nMiddlewarez< Const middleware chooses statically defined locale rrrRrrrrrrrcstj|||d||_dSNr)rBr!rR)r rRrrrrFr r r! zConstI18nMiddleware.__init__r0rr1r2cs|jSr=)rRr:r r r r4szConstI18nMiddleware.get_localer;) rRrrrrrrrrrr>rUr r rFr rWs  rWcsBeZdZdZ   ddfd d Zdfdd ZdddZZS)FSMI18nMiddlewarez: This middleware stores locale in the FSM storage rRrrrkeyrrrrrrcstj|||d||_dSrX)rBr!r[)r rr[rrrFr r r!rYzFSMI18nMiddleware.__init__r0rr1r2csj|d}d}|r|IdH}||jd}|s3tj||dIdH}|r3|j|j|idIdH|S)Nstater3r1)rKget_datar[rBr4 update_data)r r0r1 fsm_contextrRfsm_datarFr r r4s zFSMI18nMiddleware.get_localer\rcs&|j|j|idIdH||j_dS)z Write new locale to the storage :param state: instance of FSMContext :param locale: new locale r]N)r_r[rr8)r r\rRr r r set_locales zFSMI18nMiddleware.set_locale)rRrr) rrr[rrrrrrrr>)r\rrRrrr)r r r r?r!r4rbrVr r rFr rZs  rZ) __future__rabcrrtypingrrbabelrr ImportError Exceptionaiogramrrcollections.abcrraiogram.fsm.contextr aiogram.typesrraiogram.utils.i18n.corerrr@rWrZr r r r s&    H4