Система режимов для разделения работы с академией и базовыми режимами (#35)

* Обновил оновное меню admin бота

* ruff

* hot hix

Author: BratLaym

db/models/invitation.py

@@ -4,7 +4,7 @@
 from tortoise.fields import OnDelete
 
 from db.models.base import TimestampedModel, UuidModel
-from utils.role import Role
+from utils.enums import Role
 
 
 class Invitation(TimestampedModel, UuidModel):

db/models/participation.py

@@ -1,6 +1,6 @@
 from tortoise import fields
 
-from utils.role import Role
+from utils.enums import Role
 
 from .base import TimestampedModel, UuidModel
 

handlers/admin/campaign_list.py

@@ -5,10 +5,11 @@
 from aiogram import Router
 from aiogram.types import CallbackQuery
 from aiogram_dialog import Dialog, DialogManager, Window
-from aiogram_dialog.widgets.kbd import Button, ScrollingGroup, Select, Start
+from aiogram_dialog.widgets.kbd import Button, Row, ScrollingGroup, Select
 from aiogram_dialog.widgets.text import Const, Format
 
 from db.models.participation import Participation
+from utils.enums import Mode
 from utils.redirect import redirect
 
 from . import states
@@ -22,8 +23,24 @@
 # === Гетеры ===
 async def get_campaigns_data(dialog_manager: DialogManager, **kwargs):
     user: User = dialog_manager.middleware_data["user"]
-    campaigns = await Participation.filter(user=user).prefetch_related("campaign").all()
-    return {"campaigns": campaigns, "is_admin": user.admin, "has_campaigns": len(campaigns) > 0}
+    if "current_mode" not in dialog_manager.dialog_data:
+        value = Mode.Base
+        if user.admin:
+            value = Mode.Academy
+        dialog_manager.dialog_data["current_mode"] = value
+
+    campaigns = (
+        await Participation.filter(user=user)
+        .prefetch_related("campaign")
+        .filter(campaign__verified=dialog_manager.dialog_data["current_mode"] == Mode.Academy)
+        .all()
+    )
+    return {
+        "campaigns": campaigns,
+        "is_admin": user.admin,
+        "has_campaigns": len(campaigns) > 0,
+        "view_mode": f"(режим {dialog_manager.dialog_data['current_mode']})" if user.admin else "",
+    }
 
 
 # === Кнопки ===
@@ -43,9 +60,20 @@ async def on_campaign_selected(
     )
 
 
+async def on_crete_campaign(msg: CallbackQuery, wdg: Button, mng: DialogManager):
+    await mng.start(
+        state=states.CreateCampaign.select_title, data={"verified": mng.dialog_data.get("mode") == Mode.Academy}
+    )
+
+
+async def on_change_mode(msg: CallbackQuery, wdg: Button, mng: DialogManager):
+    mng.dialog_data["current_mode"] = Mode.Academy if mng.dialog_data.get("current_mode") == Mode.Base else Mode.Base
+    await mng.show()
+
+
 # === Окна ===
 campaign_list_window = Window(
-    Const("🏰 Ваши кампании\n\n"),
+    Format("🏰 Ваши кампании {view_mode}\n\n"),
     Const(
         "У вас пока нет доступных кампаний",
         when=lambda data, widget, dialog_manager: not data.get("has_campaigns", False),
@@ -64,16 +92,18 @@ async def on_campaign_selected(
         height=5,
         id="campaigns",
     ),
-    Start(
-        Const("➕ Создать новую"),
-        id="create_campaign",
-        state=states.CreateCampaign.select_title,
-    ),
-    Button(
-        Const("➕ Создать новую (Для академии)"),
-        id="create_verified_campaign",
-        on_click=lambda c, b, d: d.start(states.CreateCampaign.select_title, data={"verified": True}),
-        when="is_admin",
+    Row(
+        Button(
+            Const("➕ Создать новую"),
+            id="create_verified_campaign",
+            on_click=on_crete_campaign,
+        ),
+        Button(
+            Const("🔀 Сменить режим"),
+            id="change_mode",
+            on_click=on_change_mode,
+            when="is_admin",
+        ),
     ),
     state=states.CampaignList.main,
     getter=get_campaigns_data,

handlers/admin/campaign_manage.py

@@ -12,7 +12,7 @@
 
 from db.models.campaign import Campaign
 from db.models.participation import Participation
-from utils.role import Role
+from utils.enums import Role
 
 from . import states
 

handlers/admin/character_management.py

@@ -30,7 +30,7 @@
 from services.settings import settings
 from utils.character import CharacterData as CharData
 from utils.character import parse_character_data
-from utils.role import Role
+from utils.enums import Role
 
 from . import states
 

handlers/admin/create_campaign.py

@@ -16,7 +16,7 @@
 from db.models.campaign import Campaign
 from db.models.participation import Participation
 from services.settings import settings
-from utils.role import Role
+from utils.enums import Role
 
 from . import states
 

handlers/admin/edit_campaign.py

@@ -14,7 +14,7 @@
 from db.models.campaign import Campaign
 from db.models.participation import Participation
 from services.settings import settings
-from utils.role import Role
+from utils.enums import Role
 
 from . import states
 

handlers/admin/invitation.py

@@ -16,8 +16,8 @@
 from db.models.user import User
 from services.invitation import handle_accept_invitation, invitation_getter
 from services.settings import settings
+from utils.enums import Role
 from utils.invitation import generate_link, generate_qr
-from utils.role import Role
 
 from . import states
 

handlers/admin/manage_master.py

@@ -17,7 +17,7 @@
 from db.models.campaign import Campaign
 from db.models.participation import Participation
 from services.settings import settings
-from utils.role import Role
+from utils.enums import Role
 
 from . import states
 

services/campaigns.py

@@ -3,7 +3,7 @@
 from aiogram_dialog.api.entities import MediaAttachment
 
 from db.models import Campaign, Participation
-from utils.role import Role
+from utils.enums import Role
 
 
 async def campaign_getter(dialog_manager: DialogManager, **kwargs):