o Fh@sddlmZmZddlmZmZmZmZmZm Z z ddl m Z m Z Wne y5dZ GdddeZ YnwddlmZmZddlmZdd lmZmZdd lmZGd d d eeZGd ddeZGdddeZGdddeZdS))ABCabstractmethod)Any AwaitableCallableDictOptionalSet)LocaleUnknownLocaleErrorNc@s eZdZdS)r N)__name__ __module__ __qualname__rrP/var/www/html/venv/lib/python3.10/site-packages/aiogram/utils/i18n/middleware.pyr sr )BaseMiddlewareRouter) FSMContext)TelegramObjectUser)I18nc @seZdZdZ  ddedeededdfdd Z dd ed e d ee edefd dZ de e eeefgeefde deeefdefddZede deeefdefddZdS)I18nMiddlewarez# Abstract I18n middleware. i18ni18n_middlewarei18n_keymiddleware_keyreturnNcCs||_||_||_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 Nrrrselfrrrrrr__init__s  zI18nMiddleware.__init__rrouterexcludecCsD|durt}dh|}|jD]\}}||vrq||q|S)z Register middleware for all events in the Router :param router: :param exclude: :return: Nupdate)set observersitemsouter_middleware)rr!r"exclude_events event_nameobserverrrrsetup)s   zI18nMiddleware.setuphandlereventdatac s|j||dIdHp|jj}|jr|j||j<|jr |||j<|j.|j||||IdHWdWdS1sFwYWddS1sVwYdS)Nr-r.) get_localerdefault_localerrcontext use_locale)rr,r-r.current_localerrr__call__<s  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: Nrrr-r.rrrr0Ls zI18nMiddleware.get_localerrN)r r r__doc__rrstrr rrr r+rrrrrr5rr0rrrrrsD     $rc sZeZdZdZ  ddedeededdffdd Zd ed e ee fdefd d Z Z S)SimpleI18nMiddlewarezc Simple I18n middleware. Chooses language code from the User object received in event rrrrrNcs2tj|||dtdurtt|jddS)Nr can be used only when Babel installed Just install Babel (`pip install Babel`) or aiogram with i18n support (`pip install aiogram[i18n]`))superr r RuntimeErrortyper r __class__rrr as zSimpleI18nMiddleware.__init__r-r.cstdurtt|jd|dd}|dus|jdur"|jjSz tj|jdd}Wn t y9|jjYSw|j |jj vrE|jjS|j S)Nr<event_from_user-)sep) r r>r?r get language_coderr1parser languageavailable_locales)rr-r.rBlocalerrrr0ps    zSimpleI18nMiddleware.get_localer7) r r rr9rrr:r rrrr0 __classcell__rrr@rr;Zs &r;c s^eZdZdZ  ddededeededdf fd d Zd ed e ee fdefd dZ Z S)ConstI18nMiddlewarez< Const middleware chooses statically defined locale rrrJrrrNcstj|||d||_dSNr)r=r rJ)rrJrrrr@rrr  zConstI18nMiddleware.__init__r-r.cs|jSr8)rJr6rrrr0szConstI18nMiddleware.get_localer7) r r rr9r:rrr rrrr0rKrrr@rrLs & rLc szeZdZdZ   ddededeededd f fd d Zd ed e ee fdeffdd Z de dedd fddZ ZS)FSMI18nMiddlewarez: This middleware stores locale in the FSM storage rJrrkeyrrrNcstj|||d||_dSrM)r=r rP)rrrPrrr@rrr rNzFSMI18nMiddleware.__init__r-r.csj|d}d}|r|IdH}||jd}|s3tj||dIdH}|r3|j|j|idIdH|S)Nstater/r.)rEget_datarPr=r0 update_data)rr-r. fsm_contextrJfsm_datar@rrr0s zFSMI18nMiddleware.get_localerQcs&|j|j|idIdH||j_dS)z Write new locale to the storage :param state: instance of FSMContext :param locale: new locale rRN)rTrPrr4)rrQrJrrr set_locales zFSMI18nMiddleware.set_locale)rJrr)r r rr9rr:rr rrrr0rrWrKrrr@rrOs$"  rO)abcrrtypingrrrrrr babelr r ImportError Exceptionaiogramrraiogram.fsm.contextr aiogram.typesrraiogram.utils.i18n.corerrr;rLrOrrrrs     G+