feat: Added edit profile func

2024-03-20 23:12:35 +03:00
parent 3ca80dea1f
commit 2d35f26b29
6 changed files with 197 additions and 7 deletions
@@ -0,0 +1,171 @@
+# type: ignore
+__all__ = ()
+
+from aiogram import F, Router
+from aiogram.exceptions import TelegramBadRequest
+from aiogram.filters import StateFilter
+from aiogram.fsm.context import FSMContext
+from aiogram.types import CallbackQuery, Message, ReplyKeyboardRemove
+
+from app import messages, session
+from app.filters.user_filter import Registered, RegisteredCallback
+from app.keyboards.builders import profile
+from app.keyboards.profile import get
+from app.models.user import User
+from app.utils.states import UserAltering
+
+
+router = Router(name="profile_callback")
+
+
+@router.callback_query(
+    F.data.startswith("profile_change_"),
+    StateFilter(None),
+    RegisteredCallback(),
+)
+async def profile_change_callback(
+    callback: CallbackQuery,
+    state: FSMContext,
+) -> None:
+    if callback.data is None or callback.message is None:
+        return
+
+    column = callback.data.replace("profile_change_", "")
+
+    if column == "username":
+        await callback.message.answer(messages.EDIT_USERNAME)
+    elif column == "age":
+        await callback.message.answer(messages.INPUT_AGE)
+    elif column == "bio":
+        await callback.message.answer(messages.EDIT_BIO)
+    elif column == "sex":
+        await callback.message.answer(
+            messages.INPUT_SEX,
+            reply_markup=profile(["Male", "Female"]),
+        )
+    elif column == "location":
+        await callback.message.answer(messages.INPUT_LOCATION)
+
+    await state.update_data(
+        column=column,
+        message_id=callback.message.message_id,
+    )
+    await state.set_state(UserAltering.value)
+
+    await callback.answer()
+
+
+@router.message(UserAltering.value, F.text, Registered())
+async def profile_change_entered(message: Message, state: FSMContext) -> None:
+    column = (await state.get_data()).get("column")
+    value = message.text.strip()
+
+    if column == "username":
+        try:
+            validated_value = User().validate_username(
+                key="username",
+                value=value,
+            )
+        except AssertionError as e:
+            await message.answer(str(e))
+            return
+
+        await state.update_data(value=validated_value)
+    elif column == "age":
+        try:
+            validated_age = User().validate_age(key="age", value=value)
+        except AssertionError as e:
+            await message.answer(str(e))
+            return
+
+        await state.update_data(value=validated_age)
+    elif column == "bio":
+        if value == "/skip":
+            await state.update_data(value=None)
+        else:
+            try:
+                validated_bio = User().validate_bio(key="bio", value=value)
+            except AssertionError as e:
+                await message.answer(str(e))
+                return
+
+            await state.update_data(value=validated_bio)
+    elif column == "sex":
+        value = value.lower()
+
+        if value not in ["male", "female"]:
+            await message.answer(messages.VALIDATION_ERROR_MESSAGE)
+            return
+
+        await state.update_data(value=value)
+    elif column == "location":
+        location = value.split(", ")
+        if len(location) != 2:
+            await message.answer(messages.VALIDATION_ERROR_MESSAGE)
+            return
+
+        country, city = location
+
+        try:
+            validated_country = User().validate_country(
+                key="country",
+                value=country,
+            )
+        except AssertionError as e:
+            await message.answer(str(e))
+            return
+
+        try:
+            validated_city = User().validate_city(
+                city=city,
+                country=validated_country,
+            )
+        except AssertionError as e:
+            await message.answer(str(e))
+            return
+
+        await state.update_data(value=[validated_country, validated_city])
+
+    state_data = await state.get_data()
+
+    user = User.get_user_queryset_by_telegram_id(message.from_user.id)
+
+    if isinstance(state_data["value"], list):
+        user.update(
+            {
+                "country": state_data["value"][0],
+                "city": state_data["value"][1],
+            },
+        )
+    else:
+        data = {state_data["column"]: state_data["value"]}
+        user.update(data)
+
+    session.commit()
+
+    user = user.first()
+    session.refresh(user)
+
+    try:
+        await message.bot.edit_message_text(
+            messages.PROFILE.format(
+                username=user.username,
+                age=user.age,
+                bio=user.bio if user.bio else messages.NOT_SET,
+                sex=user.sex.capitalize(),
+                country=user.country,
+                city=user.city,
+            ),
+            message.chat.id,
+            state_data["message_id"],
+            reply_markup=get(),
+        )
+    except TelegramBadRequest:
+        pass
+
+    await message.answer(
+        "✅ Profile updated",
+        reply_markup=ReplyKeyboardRemove(),
+    )
+
+    await state.clear()