1from aiogram import F, Router
2from aiogram.filters.callback_data import CallbackData
3from aiogram.types import CallbackQuery
4from aiogram_i18n import I18nContext
5 
6 
7class LanguageSelectionCD(CallbackData, prefix="lang"):
8    code: str
9    update_in: str
10 
11 
12class LocaleStore:
13    async def save_user_locale(self, user_id: int, locale: str) -> None:
14        raise NotImplementedError
15 
16 
17router = Router()
18 
19 
20@router.callback_query(LanguageSelectionCD.filter(F.update_in == "pm"))
21async def set_new_language(
22    callback: CallbackQuery,
23    callback_data: LanguageSelectionCD,
24    i18n: I18nContext,
25    locale_store: LocaleStore,
26) -> None:
27    await locale_store.save_user_locale(
28        user_id=callback.from_user.id,
29        locale=callback_data.code,
30    )
31    await i18n.set_locale(locale=callback_data.code)
32    await callback.answer(i18n.language.changed_success())
33 
34    await callback.message.edit_text(
35        i18n.language.select_prompt(),
36        reply_markup=None,
37    )
38