From 55cbe579f77c2b128cd6276e8b4d3bd99deb058f Mon Sep 17 00:00:00 2001
From: Capybara-z <stas89091000@yandex.ru>
Date: Fri, 20 Jun 2025 00:48:06 +0300
Subject: [PATCH] fix: trial menu logic / feat: admin cluster pagination

---
 handlers/admin/users/users_handler.py |  1 +
 handlers/start.py                     | 16 ++++++++--------
 2 files changed, 9 insertions(+), 8 deletions(-)

diff --git a/handlers/admin/users/users_handler.py b/handlers/admin/users/users_handler.py
index 4788dec4..a2eec5f5 100644
--- a/handlers/admin/users/users_handler.py
+++ b/handlers/admin/users/users_handler.py
@@ -1458,6 +1458,7 @@ async def handle_create_key_start(
     builder = InlineKeyboardBuilder()
     for cluster in cluster_names:
         builder.button(text=f"🌐 {cluster}", callback_data=cluster)
+    builder.adjust(2)
     builder.row(build_admin_back_btn())
 
     await callback_query.message.edit_text(
diff --git a/handlers/start.py b/handlers/start.py
index 19063ebd..3bfaba1d 100644
--- a/handlers/start.py
+++ b/handlers/start.py
@@ -18,6 +18,7 @@ from config import (
     DONATIONS_ENABLE,
     SHOW_START_MENU_ONCE,
     SUPPORT_CHAT_URL,
+    TRIAL_TIME_DISABLE,
 )
 from database import (
     add_user,
@@ -303,16 +304,15 @@ async def show_start_menu(message: Message, admin: bool, session: AsyncSession):
     if session is not None:
         trial_status = await get_trial(session, message.chat.id)
         logger.info(f"Trial status для {message.chat.id}: {trial_status}")
-        if trial_status == 0:
-            builder.row(
-                InlineKeyboardButton(text=TRIAL_SUB, callback_data="create_key")
-            )
     else:
-        logger.warning(
-            f"Сессия базы данных отсутствует, пропускаем проверку триала для {message.chat.id}"
-        )
+        logger.warning(f"Сессия базы данных отсутствует, пропускаем проверку триала для {message.chat.id}")
 
-    if trial_status != 0 or not SHOW_START_MENU_ONCE:
+    show_trial_button = trial_status == 0 and not TRIAL_TIME_DISABLE
+    show_profile_button = not SHOW_START_MENU_ONCE or trial_status != 0 or TRIAL_TIME_DISABLE
+
+    if show_trial_button:
+        builder.row(InlineKeyboardButton(text=TRIAL_SUB, callback_data="create_key"))
+    if show_profile_button:
         builder.row(InlineKeyboardButton(text=MAIN_MENU, callback_data="profile"))
 
     if CHANNEL_EXISTS: