- Небольшая реорганизация кода в bd

- добавление работы меню ~ в процессе
- Переделал работу tarif_setting

bot.py

@@ -2,14 +2,15 @@
 from telegram import InlineKeyboardButton, InlineKeyboardMarkup, Update 
 from telegram.ext import Application, CallbackQueryHandler, CommandHandler, ContextTypes 
 from db import User, VPNServer, Transaction, Subscription
-from db import last_subscription, create_user , get_sub_list
-import db
-from sqlalchemy import desc
-import json
-import string
-from datetime import datetime
 from db import get_db_session
 from db import init_db, SessionLocal
+from db_operations import last_subscription, create_user , get_sub_list,buy_sub
+from sqlalchemy import desc
+
+import json
+
+from datetime import datetime
+
 from logger_config import setup_logger
 with open('config.json', 'r') as file:
     config = json.load(file)
@@ -42,19 +43,29 @@ async def personal_account(update: Update, context: ContextTypes.DEFAULT_TYPE) -
 
     session = next(get_db_session())
 
-    if session.query(User).filter(User.telegram_id == str(update.effective_user.id)).first() == None:
-        create_user(str(update.effective_user.id),update.effective_user.username) 
-    user = session.query(User).filter(User.telegram_id == str(update.effective_user.id)).first()
+    if session.query(User).filter(User.telegram_id == str(update.callback_query.from_user.id)).first() == None:
+        create_user(str(update.callback_query.from_user.id),update.effective_user.username) 
+    user = session.query(User).filter(User.telegram_id == str(update.callback_query.from_user.id)).first()
 
     subscription = last_subscription(session=session,user=user)
 
+    loading_message = await context.bot.send_message(chat_id=update.callback_query.message.chat.id, text="Загрузка...")
+    loading_message_id = loading_message.message_id  # Сохраняем ID сообщения
+
     if not subscription:
-        await update.message.reply_text(f'Профиль {user.username}\nВы не приобретали ещё у нас подписку, но это явно стоит сделать:)',reply_markup=reply_markup)
-    # проверяем, истекла ли подписка
+        await loading_message_id.edit_text(
+            f'Профиль {user.username}, {user.telegram_id}\nВы не приобретали ещё у нас подписку, но это явно стоит сделать:)\nВаш счёт составляет: {user.balance}'
+        )
+    # Проверяем, истекла ли подписка
     elif subscription.expiry_date < datetime.now():
-        await update.message.reply_text(f'Ваш профиль {user.username},Ваша подписка действует до - {subscription.expiry_date}',reply_markup=reply_markup)
+        await loading_message_id.edit_text(
+            f'Ваш профиль {user.username}, {user.telegram_id}, Ваша подписка действует до - {subscription.expiry_date}'
+        )
     else:
-        await update.message.reply_text(f'Ваш профиль {user.username},\nВаша подписка истекла - {subscription.expiry_date}',reply_markup=reply_markup)
+        await loading_message_id.edit_text(
+            f'Ваш профиль {user.username}, {user.telegram_id},\nВаша подписка истекла - {subscription.expiry_date}'
+        )
+
 
 async def about(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
     keyboard = [
@@ -62,7 +73,7 @@ async def about(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
         InlineKeyboardButton("Главное меню", callback_data="account")
     ]]
     reply_markup = InlineKeyboardMarkup(keyboard)
-    await update.message.reply_text(f'Игорь чё нить напишет, я продублирую',reply_markup=reply_markup)
+    await update.callback_query.message.reply_text(f'Игорь чё нить напишет, я продублирую',reply_markup=reply_markup)
 
 async def support(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
     keyboard = [
@@ -71,7 +82,7 @@ async def support(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
         InlineKeyboardButton("Написать", callback_data="sup") # Нужно через каллбек доделать 
     ]]
     reply_markup = InlineKeyboardMarkup(keyboard)
-    await update.message.reply_text(f'Что бы отправить сообщение поддержке выберите в меню кнопку "Написать", а далее изложите в одном сообщении свою ошибку.',reply_markup=reply_markup)
+    await update.callback_query.message.reply_text(f'Что бы отправить сообщение поддержке выберите в меню кнопку "Написать", а далее изложите в одном сообщении свою ошибку.',reply_markup=reply_markup)
 
 async def pop_up(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
     keyboard = [
@@ -79,7 +90,7 @@ async def pop_up(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
         InlineKeyboardButton("Главное меню", callback_data="account"),
     ]]
     reply_markup = InlineKeyboardMarkup(keyboard)
-    await update.message.reply_text(f'Когда нибудь эта штука заработает',reply_markup=reply_markup)
+    await update.callback_query.message.reply_text(f'Когда нибудь эта штука заработает',reply_markup=reply_markup)
 
 async def buy_subscription(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
     keyboard = [
@@ -88,22 +99,25 @@ async def buy_subscription(update: Update, context: ContextTypes.DEFAULT_TYPE) -
     ]]
     reply_markup = InlineKeyboardMarkup(keyboard)
     session = next(get_db_session())
-    user = session.query(User).filter(User.telegram_id == str(update.effective_user.id)).first()
+    user = session.query(User).filter(User.telegram_id == str(update.callback_query.from_user.id)).first()
     check = last_subscription(session=session,user=user)
 
     if not check:
         keyboard = [
         [
-            InlineKeyboardButton("Тариф 1 \"Бимжик\"", callback_data="bimzhik"),
+            InlineKeyboardButton("Тариф 1 \"Бимжик\"", callback_data="Бимжик"),
         ],
         [
-            InlineKeyboardButton("Тариф 2 \"Бизнес хомячёк\"", callback_data="homyachok"),
+            InlineKeyboardButton("Тариф 2 \"Бизнес хомячёк\"", callback_data="Бизнес_хомячёк"),
         ],
         [
-            InlineKeyboardButton("Тариф 2 \"Продвинутый Акулёнок\"", callback_data="akulenok"),
+            InlineKeyboardButton("Тариф 3 \"Продвинутый Акулёнок\"", callback_data="Продвинутый_Акулёнок"),
+        ],
+        [
+            InlineKeyboardButton("Главное меню", callback_data="account"),
         ]]
         reply_markup = InlineKeyboardMarkup(keyboard)
-        await update.message.reply_text(f'Какую подписку вы хотели бы приобрести\nТариф 1 "Бимжик" - 200 рубликов - 1 месяцок\nТариф 2 "Бизнес хомячёк" - 500 рубликов - 3 месяцка\nТариф 3 "Продвинутый Акулёнок" - 888 рубликов - 888 рубликов\n',reply_markup=reply_markup)
+        await update.callback_query.message.reply_text(f'Какую подписку вы хотели бы приобрести\nТариф 1 "Бимжик" - 200 рубликов - 1 месяцок\nТариф 2 "Бизнес хомячёк" - 500 рубликов - 3 месяцка\nТариф 3 "Продвинутый Акулёнок" - 888 рубликов - 6 месяцков\n',reply_markup=reply_markup)
     # проверяем, истекла ли подписка
     else:
         keyboard = [
@@ -111,7 +125,7 @@ async def buy_subscription(update: Update, context: ContextTypes.DEFAULT_TYPE) -
             InlineKeyboardButton("Главное меню", callback_data="account"),
         ]]
         reply_markup = InlineKeyboardMarkup(keyboard)
-        await update.message.reply_text(f'У вас уже приобретена подписка',reply_markup=reply_markup)
+        await update.callback_query.message.reply_text(f'У вас уже приобретена подписка',reply_markup=reply_markup)
 
 
 async def faq(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
@@ -120,7 +134,50 @@ async def faq(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
         InlineKeyboardButton("Главное меню", callback_data="account"),
     ]]
     reply_markup = InlineKeyboardMarkup(keyboard)
-    await update.message.reply_text(f'Когда нибудь что нибудь здесь будет написано!!;)',reply_markup=reply_markup)
+    await update.callback_query.message.reply_text(f'Когда нибудь что нибудь здесь будет написано!!;)',reply_markup=reply_markup)
+
+async def button_handler(update: Update, context):
+    query = update.callback_query
+
+    loading_message = await query.message.reply_text("Загрузка...")
+    
+    await query.answer()
+
+    session = next(get_db_session())
+    try:
+        if query.data == 'account':
+            await personal_account(update, context=context)
+        elif query.data == 'about':
+            await about(update, context)
+        elif query.data == 'support':
+            await support(update, context)
+        elif query.data == 'pop_up':
+            await pop_up(update, context)
+        elif query.data == 'buy_tarif':
+            await buy_subscription(update, context)
+        elif query.data == 'faq':
+            await faq(update, context)
+        elif query.data == 'payment_history':
+            await active_sub(update, context)
+        
+        if query.data in ['Бимжик', 'Бизнес_хомячёк', 'Продвинутый_Акулёнок']:
+            plan = query.data.replace('_', ' ') if '_' in query.data else query.data
+            check = buy_sub(session, query.from_user.id, plan, logger)
+
+            
+            if check != "OK":
+                await loading_message.edit_text("Неизвестный тариф.")
+            else:
+                await loading_message.edit_text("Тариф успешно установлен!")
+
+    except Exception as e:
+        logger.error(f"Ошибка при обработке запроса пользователя {query.from_user.id}: {e}")
+        await loading_message.edit_text("Произошла ошибка. Пожалуйста, попробуйте снова.")
+    finally:
+        session.close()
+
+
+
 async def active_sub(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
     keyboard = [
     [
@@ -128,7 +185,7 @@ async def active_sub(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None
     ]]
     reply_markup = InlineKeyboardMarkup(keyboard)
     session = next(get_db_session())
-    list_sub = get_sub_list(session, 10, update.effective_user.id)
+    list_sub = get_sub_list(session, 10, update.callback_query.from_user.id)
 
     if list_sub:  
         message = "Ваши подписки:\n"
@@ -139,10 +196,9 @@ async def active_sub(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None
                 message += f"   Устаревшая: {cur_sub.plan}, Дата покупки: {cur_sub.created_at}\n"
     else: 
         message = "Ты пидор, не приобрел у нас подписку?!"
-    await update.message.reply_text(message,reply_markup=reply_markup)
+    await update.callback_query.message.reply_text(message,reply_markup=reply_markup)
 
 def main() -> None:
-    
     init_db()
     db = SessionLocal()
     application = Application.builder().token(config['token']).build()
@@ -155,6 +211,7 @@ def main() -> None:
     application.add_handler(CommandHandler("buy_subscription", buy_subscription))
     application.add_handler(CommandHandler("faq", faq))
     application.add_handler(CommandHandler("active_sub", active_sub))
+    application.add_handler(CallbackQueryHandler(button_handler))
     
 
     application.run_polling(allowed_updates=Update.ALL_TYPES)