Minor improvements

plugins/language.py

@@ -1,16 +1,20 @@
 from pykeyboard import InlineButton, InlineKeyboard
 from pyrogram import filters
-from pyrogram.client import Client
 from pyrogram.types import CallbackQuery, Message
 
+from classes.callbacks import CallbackLanguage
 from classes.pyroclient import PyroClient
+from classes.pyrouser import PyroUser
 
 
-@Client.on_message(
+@PyroClient.on_message(
     ~filters.scheduled & filters.private & filters.command(["language"], prefixes=["/"])  # type: ignore
 )
 async def command_language(app: PyroClient, message: Message):
-    user = await app.find_user(message.from_user)
+    user = await PyroUser.find(
+        message.from_user.id, locale=message.from_user.language_code
+    )
+
     keyboard = InlineKeyboard(row_width=2)
     buttons = []
 
@@ -27,16 +31,18 @@ async def command_language(app: PyroClient, message: Message):
     )
 
 
-@Client.on_callback_query(filters.regex(r"language:[\s\S]*"))  # type: ignore
+@PyroClient.on_callback_query(filters.regex(r"language:[\s\S]*"))  # type: ignore
 async def callback_language(app: PyroClient, callback: CallbackQuery):
-    user = await app.find_user(callback.from_user)
-    language = str(callback.data).split(":")[1]
+    user = await PyroUser.find(
+        callback.from_user.id, locale=callback.from_user.language_code
+    )
+    parsed = CallbackLanguage.from_callback(callback)
 
-    await user.update_locale(language)
+    await user.update_locale(parsed.language)
 
     await callback.answer(
-        app._("locale_set", "callbacks", locale=language).format(
-            locale=app._("name", "metadata", locale=language)
+        app._("locale_set", "callbacks", locale=parsed.language).format(
+            locale=app._("name", "metadata", locale=parsed.language)
         ),
         show_alert=True,
     )