From 33801f58c4e31ee1827f2fa4b50c3041dfefcc3d Mon Sep 17 00:00:00 2001
From: izzzzzi <izmailov9@gmail.com>
Date: Sat, 22 Feb 2025 14:27:56 +0500
Subject: [PATCH] Refactor inline referral handler and admin stats time display

- Simplified inline referral handler by removing unnecessary try-except block
- Increased cache time for inline query results
- Added Moscow timezone for admin stats update time display
---
 handlers/admin/admin_stats.py |  5 ++--
 handlers/profile.py           | 44 +++++++++++++----------------------
 2 files changed, 19 insertions(+), 30 deletions(-)

diff --git a/handlers/admin/admin_stats.py b/handlers/admin/admin_stats.py
index 26528067..bb5697a8 100644
--- a/handlers/admin/admin_stats.py
+++ b/handlers/admin/admin_stats.py
@@ -4,6 +4,7 @@ from typing import Any
 from aiogram import F, Router
 from aiogram.exceptions import TelegramBadRequest
 from aiogram.types import CallbackQuery
+import pytz
 
 from filters.admin import IsAdminFilter
 from keyboards.admin.panel_kb import AdminPanelCallback, build_admin_back_kb
@@ -54,8 +55,8 @@ async def handle_stats(callback_query: CallbackQuery, session: Any):
             int(datetime.utcnow().timestamp() * 1000),
         )
         expired_keys = total_keys - active_keys
-
-        update_time = datetime.now().strftime("%d.%m.%y %H:%M:%S")
+        moscow_tz = pytz.timezone("Europe/Moscow")
+        update_time = datetime.now(moscow_tz).strftime("%d.%m.%y %H:%M:%S")
 
         stats_message = (
             f"📊 <b>Подробная статистика проекта:</b>\n\n"
diff --git a/handlers/profile.py b/handlers/profile.py
index 38c68ec5..4d09345d 100644
--- a/handlers/profile.py
+++ b/handlers/profile.py
@@ -220,37 +220,25 @@ async def invite_handler(callback_query: CallbackQuery):
 
 @router.inline_query(F.query.in_(["referral", "ref", "invite"]))
 async def inline_referral_handler(inline_query: InlineQuery):
-    try:
-        if not inline_query.from_user:
-            return await inline_query.answer([])
+    referral_link = f"https://t.me/{USERNAME_BOT}?start=referral_{inline_query.from_user.id}"
 
-        if not USERNAME_BOT:
-            return await inline_query.answer([])
+    results: list[InlineQueryResultArticle] = []
 
-        referral_link = f"https://t.me/{USERNAME_BOT}?start=referral_{inline_query.from_user.id}"
+    for index, offer in enumerate(REFERRAL_OFFERS):
+        description = offer["description"][:64]
+        message_text = offer["message"].format(trial_time=TRIAL_TIME)[:4096]
 
-        if not inline_query.query:
-            return await inline_query.answer([])
+        builder = InlineKeyboardBuilder()
+        builder.row(InlineKeyboardButton(text=offer["title"], url=referral_link))
 
-        results: list[InlineQueryResultArticle] = []
-
-        for index, offer in enumerate(REFERRAL_OFFERS):
-            description = offer["description"][:64]
-            message_text = offer["message"].format(trial_time=TRIAL_TIME)[:4096]
-
-            builder = InlineKeyboardBuilder()
-            builder.row(InlineKeyboardButton(text=offer["title"], url=referral_link))
-
-            results.append(
-                InlineQueryResultArticle(
-                    id=str(index),
-                    title=offer["title"],
-                    description=description,
-                    input_message_content=InputTextMessageContent(message_text=message_text, parse_mode=ParseMode.HTML),
-                    reply_markup=builder.as_markup(),
-                )
+        results.append(
+            InlineQueryResultArticle(
+                id=str(index),
+                title=offer["title"],
+                description=description,
+                input_message_content=InputTextMessageContent(message_text=message_text, parse_mode=ParseMode.HTML),
+                reply_markup=builder.as_markup(),
             )
+        )
 
-        await inline_query.answer(results=results, cache_time=1)
-    except Exception:
-        await inline_query.answer([])
+    await inline_query.answer(results=results, cache_time=86400, is_personal=True)