o LDih@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 i/var/www/www-root/data/www/77.83.87.30/venv/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:|jjYSw|j |jj vrF|jjS|j S)NrAevent_from_user-)sep) rrCr rDget language_coderr5parserlanguageavailable_locales)r r0r1rErHlocaler 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 rrrPrrrrrrrcstj|||d||_dSNr)rBr!rP)r rPrrrrFr r r! zConstI18nMiddleware.__init__r0rr1r2cs|jSr=)rPr:r r r r4szConstI18nMiddleware.get_localer;) rPrrrrrrrrrr>rQr r rFr rSs  rScsBeZdZdZ   ddfd d Zdfdd ZdddZZS)FSMI18nMiddlewarez: This middleware stores locale in the FSM storage rPrrrkeyrrrrrrcstj|||d||_dSrT)rBr!rW)r rrWrrrFr r r!rUzFSMI18nMiddleware.__init__r0rr1r2csj|d}d}|r|IdH}||jd}|s3tj||dIdH}|r3|j|j|idIdH|S)Nstater3r1)rKget_datarWrBr4 update_data)r r0r1 fsm_contextrPfsm_datarFr r r4s zFSMI18nMiddleware.get_localerXrcs&|j|j|idIdH||j_dS)z Write new locale to the storage :param state: instance of FSMContext :param locale: new locale rYN)r[rWrr8)r rXrPr r r set_locales zFSMI18nMiddleware.set_locale)rPrr) rrrWrrrrrrrr>)rXrrPrrr)r r r r?r!r4r^rRr r rFr rVs  rV) __future__rabcrrtypingrrbabelrr ImportError Exceptionaiogramrrcollections.abcrraiogram.fsm.contextr aiogram.typesrraiogram.utils.i18n.corerrr@rSrVr r r r s&    H-