a :g@sddlmZmZddlmZmZmZmZmZm Z zddl m Z m Z Wn&e yjdZ GdddeZ Yn0ddlmZmZddlmZdd lmZmZdd lmZGd d d eeZGd ddeZGdddeZGdddeZdS))ABCabstractmethod)Any AwaitableCallableDictOptionalSet)LocaleUnknownLocaleErrorNc@s eZdZdS)r N)__name__ __module__ __qualname__rrz/var/www/vosh/data/www/fastworkle.ru/webshop-tgbot-v.1.0/venv/lib/python3.9/site-packages/aiogram/utils/i18n/middleware.pyr sr )BaseMiddlewareRouter) FSMContext)TelegramObjectUser)I18nc@seZdZdZdeeeeddddZdee ee eedd d Z e e eeefgeefe eeefed d d Zee eeefedddZdS)I18nMiddlewarez# Abstract I18n middleware. i18ni18n_middlewareNri18n_keymiddleware_keyreturncCs||_||_||_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__)r routerexcludercCsD|durt}dh|}|jD]\}}||vr4q"||q"|S)z Register middleware for all events in the Router :param router: :param exclude: :return: Nupdate)setZ observersitemsZouter_middleware)r r"r#Zexclude_eventsZ event_nameZobserverrrrsetup)s   zI18nMiddleware.setup)handlereventdatarc s|j||dIdHp|jj}|jr.|j||j<|jr>|||j<|jZ|j|.|||IdHWdWdS1s0YWdn1s0YdS)Nr)r*) get_localerdefault_localerrcontextZ use_locale)r r(r)r*current_localerrr__call__<s  zI18nMiddleware.__call__r)r*rcsdS)z Detect current user locale based on event and context. **This method must be defined in child classes** :param event: :param data: :return: Nrr r)r*rrrr,Ls zI18nMiddleware.get_locale)rr)N)r r r__doc__rrstrr!rrr r'rrrrrr0rr,rrrrrs*    rcsLeZdZdZd eeeeddfdd Zee ee fedd d Z Z S) SimpleI18nMiddlewarezc Simple I18n middleware. Chooses language code from the User object received in event rrNrcs2tj|||dtdur.tt|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__r1cstdurtt|jd|dd}|dus:|jdurB|jjSztj|jdd}Wnt yp|jjYS0|j |jj vr|jjS|j S)Nr6event_from_user-)sep) r r8r9r getZ language_coderr-parser languageZavailable_locales)r r)r*r<localerrrr,ps  zSimpleI18nMiddleware.get_locale)rr) r r rr3rrr4r!rrrr, __classcell__rrr:rr5Zs r5csNeZdZdZd eeeeeddfdd Zee ee fedd d Z Z S) ConstI18nMiddlewarez< Const middleware chooses statically defined locale rrN)rBrrrrcstj|||d||_dSNr)r7r!rB)r rBrrrr:rrr!szConstI18nMiddleware.__init__r1cs|jS)N)rBr2rrrr,szConstI18nMiddleware.get_locale)rr) r r rr3r4rrr!rrrr,rCrrr:rrDs rDcsdeZdZdZdeeeeeddfdd Zee ee fed fd d Z e edd d dZ ZS)FSMI18nMiddlewarez: This middleware stores locale in the FSM storage rBrrN)rkeyrrrcstj|||d||_dSrE)r7r!rG)r rrGrrr:rrr!szFSMI18nMiddleware.__init__r1csh|d}d}|r.|IdH}||jd}|sdtj||dIdH}|rd|j|j|idIdH|S)Nstater+r*)r?get_datarGr7r, update_data)r r)r*Z fsm_contextrBZfsm_datar:rrr,s zFSMI18nMiddleware.get_locale)rHrBrcs$|j|j|idIdH||j_dS)z Write new locale to the storage :param state: instance of FSMContext :param locale: new locale rIN)rKrGrr/)r rHrBrrr set_localeszFSMI18nMiddleware.set_locale)rBrr)r r rr3rr4rr!rrrr,rrLrCrrr:rrFs  rF)abcrrtypingrrrrrr Zbabelr r ImportError ExceptionZaiogramrrZaiogram.fsm.contextrZ aiogram.typesrrZaiogram.utils.i18n.corerrr5rDrFrrrrs    G+