- Добавил две базы данных для сап системы
- Сап система готова на 40%~
This commit is contained in:
Disledg
85
bot.py
85
bot.py
@@ -1,6 +1,6 @@
|
||||
|
||||
from telegram import InlineKeyboardButton, InlineKeyboardMarkup, Update
|
||||
from telegram.ext import Application, CallbackQueryHandler, CommandHandler, ContextTypes
|
||||
from telegram.ext import Application, CallbackQueryHandler, CommandHandler, ContextTypes,MessageHandler, filters, CommandHandler
|
||||
from db import User, VPNServer, Transaction, Subscription
|
||||
from db import get_db_session
|
||||
from db import init_db, SessionLocal
|
||||
@@ -15,7 +15,6 @@ from logger_config import setup_logger
|
||||
with open('config.json', 'r') as file:
|
||||
config = json.load(file)
|
||||
|
||||
|
||||
logger = setup_logger()
|
||||
|
||||
|
||||
@@ -28,7 +27,8 @@ async def start(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
[InlineKeyboardButton("Поддержка", callback_data='support')],
|
||||
]
|
||||
reply_markup = InlineKeyboardMarkup(keyboard)
|
||||
await update.message.reply_text(f'Добро пожаловать в ...! Здесь вы можете приобрести VPN. И нечего более',reply_markup=reply_markup)
|
||||
loading_message = await context.bot.send_message(chat_id=update.callback_query.message.chat.id, text="Загрузка...")
|
||||
await loading_message.edit_text(f'Добро пожаловать в ...! Здесь вы можете приобрести VPN. И нечего более',reply_markup=reply_markup)
|
||||
|
||||
async def personal_account(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
keyboard = [
|
||||
@@ -36,7 +36,7 @@ async def personal_account(update: Update, context: ContextTypes.DEFAULT_TYPE) -
|
||||
InlineKeyboardButton("Пополнить баланс", callback_data="pop_up"),
|
||||
InlineKeyboardButton("Приобрести подписку", callback_data='buy_tarif'),
|
||||
],
|
||||
[InlineKeyboardButton("FAQ", callback_data='faq')],
|
||||
[InlineKeyboardButton("❔FAQ❔", callback_data='faq')],
|
||||
[InlineKeyboardButton("История платежей", callback_data='payment_history')]
|
||||
]
|
||||
reply_markup = InlineKeyboardMarkup(keyboard)
|
||||
@@ -50,52 +50,54 @@ async def personal_account(update: Update, context: ContextTypes.DEFAULT_TYPE) -
|
||||
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 loading_message_id.edit_text(
|
||||
f'Профиль {user.username}, {user.telegram_id}\nВы не приобретали ещё у нас подписку, но это явно стоит сделать:)\nВаш счёт составляет: {user.balance}'
|
||||
await loading_message.edit_text(
|
||||
f'Профиль {user.username}, {user.telegram_id}\nВы не приобретали ещё у нас подписку, но это явно стоит сделать:)\nВаш счёт составляет: {user.balance}',reply_markup=reply_markup
|
||||
)
|
||||
# Проверяем, истекла ли подписка
|
||||
elif subscription.expiry_date < datetime.now():
|
||||
await loading_message_id.edit_text(
|
||||
f'Ваш профиль {user.username}, {user.telegram_id}, Ваша подписка действует до - {subscription.expiry_date}'
|
||||
await loading_message.edit_text(
|
||||
f'Ваш профиль {user.username}, {user.telegram_id}, Ваша подписка действует до - {subscription.expiry_date}',reply_markup=reply_markup
|
||||
)
|
||||
else:
|
||||
await loading_message_id.edit_text(
|
||||
f'Ваш профиль {user.username}, {user.telegram_id},\nВаша подписка истекла - {subscription.expiry_date}'
|
||||
await loading_message.edit_text(
|
||||
f'Ваш профиль {user.username}, {user.telegram_id},\nВаша подписка истекла - {subscription.expiry_date}',reply_markup=reply_markup
|
||||
)
|
||||
|
||||
|
||||
async def about(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
keyboard = [
|
||||
[
|
||||
InlineKeyboardButton("Главное меню", callback_data="account")
|
||||
InlineKeyboardButton("Главное меню", callback_data="start")
|
||||
]]
|
||||
reply_markup = InlineKeyboardMarkup(keyboard)
|
||||
await update.callback_query.message.reply_text(f'Игорь чё нить напишет, я продублирую',reply_markup=reply_markup)
|
||||
loading_message = await context.bot.send_message(chat_id=update.callback_query.message.chat.id, text="Загрузка...")
|
||||
await loading_message.edit_text(f'Игорь чё нить напишет, я продублирую',reply_markup=reply_markup)
|
||||
|
||||
async def support(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
keyboard = [
|
||||
[
|
||||
InlineKeyboardButton("Главное меню", callback_data="account"),
|
||||
InlineKeyboardButton("Главное меню", callback_data="start"),
|
||||
InlineKeyboardButton("Написать", callback_data="sup") # Нужно через каллбек доделать
|
||||
]]
|
||||
reply_markup = InlineKeyboardMarkup(keyboard)
|
||||
await update.callback_query.message.reply_text(f'Что бы отправить сообщение поддержке выберите в меню кнопку "Написать", а далее изложите в одном сообщении свою ошибку.',reply_markup=reply_markup)
|
||||
loading_message = await context.bot.send_message(chat_id=update.callback_query.message.chat.id, text="Загрузка...")
|
||||
await loading_message.edit_text(f'Что бы отправить сообщение поддержке выберите в меню кнопку "Написать", а далее изложите в одном сообщении свою ошибку.',reply_markup=reply_markup)
|
||||
|
||||
async def pop_up(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
keyboard = [
|
||||
[
|
||||
InlineKeyboardButton("Главное меню", callback_data="account"),
|
||||
InlineKeyboardButton("Главное меню", callback_data="start"),
|
||||
]]
|
||||
reply_markup = InlineKeyboardMarkup(keyboard)
|
||||
await update.callback_query.message.reply_text(f'Когда нибудь эта штука заработает',reply_markup=reply_markup)
|
||||
loading_message = await context.bot.send_message(chat_id=update.callback_query.message.chat.id, text="Загрузка...")
|
||||
await loading_message.edit_text(f'Когда нибудь эта штука заработает',reply_markup=reply_markup)
|
||||
|
||||
async def buy_subscription(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
keyboard = [
|
||||
[
|
||||
InlineKeyboardButton("Главное меню", callback_data="account"),
|
||||
InlineKeyboardButton("Главное меню", callback_data="start"),
|
||||
]]
|
||||
reply_markup = InlineKeyboardMarkup(keyboard)
|
||||
session = next(get_db_session())
|
||||
@@ -114,43 +116,60 @@ async def buy_subscription(update: Update, context: ContextTypes.DEFAULT_TYPE) -
|
||||
InlineKeyboardButton("Тариф 3 \"Продвинутый Акулёнок\"", callback_data="Продвинутый_Акулёнок"),
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton("Главное меню", callback_data="account"),
|
||||
InlineKeyboardButton("Главное меню", callback_data="start"),
|
||||
]]
|
||||
reply_markup = InlineKeyboardMarkup(keyboard)
|
||||
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)
|
||||
loading_message = await context.bot.send_message(chat_id=update.callback_query.message.chat.id, text="Загрузка...")
|
||||
await loading_message.edit_text(f'Какую подписку вы хотели бы приобрести\nТариф 1 "Бимжик" - 200 рубликов - 1 месяцок\nТариф 2 "Бизнес хомячёк" - 500 рубликов - 3 месяцка\nТариф 3 "Продвинутый Акулёнок" - 888 рубликов - 6 месяцков\n',reply_markup=reply_markup)
|
||||
# проверяем, истекла ли подписка
|
||||
else:
|
||||
keyboard = [
|
||||
[
|
||||
InlineKeyboardButton("Главное меню", callback_data="account"),
|
||||
InlineKeyboardButton("Главное меню", callback_data="start"),
|
||||
]]
|
||||
reply_markup = InlineKeyboardMarkup(keyboard)
|
||||
await update.callback_query.message.reply_text(f'У вас уже приобретена подписка',reply_markup=reply_markup)
|
||||
loading_message = await context.bot.send_message(chat_id=update.callback_query.message.chat.id, text="Загрузка...")
|
||||
await loading_message.edit_text(f'У вас уже приобретена подписка',reply_markup=reply_markup)
|
||||
|
||||
|
||||
async def faq(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
keyboard = [
|
||||
[
|
||||
InlineKeyboardButton("Главное меню", callback_data="account"),
|
||||
InlineKeyboardButton("Главное меню", callback_data="start"),
|
||||
]]
|
||||
reply_markup = InlineKeyboardMarkup(keyboard)
|
||||
await update.callback_query.message.reply_text(f'Когда нибудь что нибудь здесь будет написано!!;)',reply_markup=reply_markup)
|
||||
loading_message = await context.bot.send_message(chat_id=update.callback_query.message.chat.id, text="Загрузка...")
|
||||
await loading_message.edit_text(f'Когда нибудь что нибудь здесь будет написано!!;)',reply_markup=reply_markup)
|
||||
|
||||
|
||||
async def sup(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
if context.user_data.get('awaiting_input'):
|
||||
user_input = update.message.text
|
||||
await update.message.reply_text(f"Вы ввели: {user_input}")
|
||||
|
||||
# После получения текста сбрасываем ожидание
|
||||
context.user_data['awaiting_input'] = False
|
||||
else:
|
||||
await update.message.reply_text("Выберите команду или нажмите кнопку для продолжения.")
|
||||
|
||||
|
||||
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)
|
||||
await personal_account(update,context)
|
||||
elif query.data == 'start':
|
||||
await start(update,context)
|
||||
elif query.data == 'about':
|
||||
await about(update, context)
|
||||
elif query.data == 'support':
|
||||
await support(update, context)
|
||||
elif query.data == 'sup':
|
||||
context.user_data['awaiting_input'] = True
|
||||
elif query.data == 'pop_up':
|
||||
await pop_up(update, context)
|
||||
elif query.data == 'buy_tarif':
|
||||
@@ -161,6 +180,7 @@ async def button_handler(update: Update, context):
|
||||
await active_sub(update, context)
|
||||
|
||||
if query.data in ['Бимжик', 'Бизнес_хомячёк', 'Продвинутый_Акулёнок']:
|
||||
loading_message = await query.message.reply_text("Загрузка...")
|
||||
plan = query.data.replace('_', ' ') if '_' in query.data else query.data
|
||||
check = buy_sub(session, query.from_user.id, plan, logger)
|
||||
|
||||
@@ -172,7 +192,7 @@ async def button_handler(update: Update, context):
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка при обработке запроса пользователя {query.from_user.id}: {e}")
|
||||
await loading_message.edit_text("Произошла ошибка. Пожалуйста, попробуйте снова.")
|
||||
await query.message.reply_text("Произошла ошибка. Пожалуйста, попробуйте снова.")
|
||||
finally:
|
||||
session.close()
|
||||
|
||||
@@ -181,12 +201,12 @@ async def button_handler(update: Update, context):
|
||||
async def active_sub(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||
keyboard = [
|
||||
[
|
||||
InlineKeyboardButton("Главное меню", callback_data="account"),
|
||||
InlineKeyboardButton("Главное меню", callback_data="start"),
|
||||
]]
|
||||
reply_markup = InlineKeyboardMarkup(keyboard)
|
||||
session = next(get_db_session())
|
||||
list_sub = get_sub_list(session, 10, update.callback_query.from_user.id)
|
||||
|
||||
loading_message = await context.bot.send_message(chat_id=update.callback_query.message.chat.id, text="Загрузка...")
|
||||
if list_sub:
|
||||
message = "Ваши подписки:\n"
|
||||
for cur_sub in list_sub:
|
||||
@@ -196,7 +216,8 @@ 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.callback_query.message.reply_text(message,reply_markup=reply_markup)
|
||||
|
||||
await loading_message.edit_text(message,reply_markup=reply_markup)
|
||||
|
||||
def main() -> None:
|
||||
init_db()
|
||||
@@ -212,7 +233,7 @@ def main() -> None:
|
||||
application.add_handler(CommandHandler("faq", faq))
|
||||
application.add_handler(CommandHandler("active_sub", active_sub))
|
||||
application.add_handler(CallbackQueryHandler(button_handler))
|
||||
|
||||
application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, sup))
|
||||
|
||||
application.run_polling(allowed_updates=Update.ALL_TYPES)
|
||||
db.close()
|
||||
|
||||
18
db.py
18
db.py
@@ -54,6 +54,24 @@ class Transaction(Base):
|
||||
|
||||
user = relationship("User", back_populates="transactions")
|
||||
|
||||
class Requests(Base):
|
||||
__tablename__ = 'requests'
|
||||
|
||||
id = Column(String,primary_key=True,default=generate_uuid)
|
||||
user_id = Column(String,ForeignKey('users.id'))
|
||||
username = Column(String)
|
||||
created_at = Column(DateTime,default=datetime.now)
|
||||
content = Column(String)
|
||||
status = Column(String,default='open')
|
||||
user = relationship("User",back_populates="requests")
|
||||
|
||||
class Administrators(Base):
|
||||
__tablename__ = 'admins'
|
||||
|
||||
id = Column(String,primary_key=True,default=generate_uuid)
|
||||
user_id = Column(String,ForeignKey('users.id'))
|
||||
admin = Column(bool,default=False)
|
||||
user = relationship("User",back_populates="admins")
|
||||
# VPN-серверы
|
||||
class VPNServer(Base):
|
||||
__tablename__ = 'vpn_servers'
|
||||
|
||||
Reference in New Issue
Block a user