1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
from telebot.types import InlineKeyboardMarkup, InlineKeyboardButton
from config_data.config import ALLOWED_USERS
from database.models import User, Server
def admin_markup():
""" Inline buttons для выбора меню админа """
actions = InlineKeyboardMarkup(row_width=2)
actions.add(InlineKeyboardButton(text=f"🖥 Управление серверами", callback_data="servers"))
actions.add(InlineKeyboardButton(text=f"👥 Управление пользователями", callback_data="users"))
actions.add(InlineKeyboardButton(text=f"🚪 Выйти", callback_data="Exit"))
return actions
def users_markup():
""" Inline buttons для выбора юзера """
users_obj = User.select()
actions = InlineKeyboardMarkup(row_width=2)
for user in users_obj:
if int(user.user_id) not in ALLOWED_USERS:
actions.add(InlineKeyboardButton(text=f"👤 {user.full_name}", callback_data=user.id))
actions.add(InlineKeyboardButton(text=f"🔙 Назад", callback_data="Exit"))
return actions
def get_servers_markup():
""" Inline buttons для выдачи всех локаций серверов """
actions = InlineKeyboardMarkup(row_width=1)
servers_obj = Server.select()
for server in servers_obj:
actions.add(InlineKeyboardButton(text=f"🌍 {server.location}", callback_data=str(server.id)))
actions.add(InlineKeyboardButton(text=f"➕ Добавить сервер", callback_data="Add"))
return actions
def get_vpn_markup(server_id):
""" Inline buttons для выдачи vpn ключей сервера """
cur_server: Server = Server.get_by_id(server_id)
actions = InlineKeyboardMarkup(row_width=2)
for vpn_key_obj in cur_server.keys:
actions.add(InlineKeyboardButton(text=f"🔑 {vpn_key_obj.name}", callback_data=f"VPN - {str(vpn_key_obj.id)}"))
actions.add(InlineKeyboardButton(text=f"🔄 Сгенерировать новый ключ", callback_data=f"Generate {cur_server.id}"))
actions.add(InlineKeyboardButton(text=f"🗑 Удалить сервер", callback_data=f"Delete {cur_server.id}"),
InlineKeyboardButton(text=f"🔙 Назад", callback_data="Cancel")) # Возврат в меню серверов
return actions
def delete_vpn_markup(vpn_obj_id: int):
""" Inline buttons для удаления vpn ключа """
actions = InlineKeyboardMarkup(row_width=2)
actions.add(InlineKeyboardButton(text=f"🗑 Удалить", callback_data=f"Del - {vpn_obj_id}"),
InlineKeyboardButton(text=f"🚪 Выйти", callback_data="Cancel"))
return actions
def key_actions_markup(key_id: int) -> InlineKeyboardMarkup:
"""
Клавиатура для управления ключом:
- Приостановить/возобновить
- Отозвать
- Назад
"""
markup = InlineKeyboardMarkup(row_width=2)
markup.add(
InlineKeyboardButton("⏸ Приостановить", callback_data=f"action_suspend_{key_id}"),
InlineKeyboardButton("▶️ Возобновить", callback_data=f"action_resume_{key_id}"),
InlineKeyboardButton("🗑 Отозвать", callback_data=f"action_revoke_{key_id}"),
InlineKeyboardButton("🔙 Назад", callback_data="Cancel")
)
return markup
|
