- bot telegram
This commit is contained in:
167
hlna_bot.py
167
hlna_bot.py
@@ -1,167 +0,0 @@
|
||||
#!/usr/bin/env python3
|
||||
import errno
|
||||
import time
|
||||
import sys
|
||||
import os
|
||||
import telebot
|
||||
from telebot import types
|
||||
import subprocess
|
||||
import yaml
|
||||
import colorama
|
||||
|
||||
import hlna
|
||||
|
||||
list_config = hlna.list_config
|
||||
|
||||
|
||||
def start():
|
||||
if not os.path.exists(f"{hlna.home_dir}/.config/hlna/bot/hlna_bot"):
|
||||
key_api = input("Укажите ключ бота\n")
|
||||
os.chdir(f"{hlna.home_dir}/.config/hlna/")
|
||||
try:
|
||||
os.mkdir("bot")
|
||||
except OSError as e:
|
||||
if e.errno == errno.EEXIST:
|
||||
print('Каталог создан')
|
||||
else:
|
||||
raise
|
||||
create_config(key_api)
|
||||
else:
|
||||
x = input("Уже есть запомненный ключа api. Запустить его? Y/n \n")
|
||||
if x == "" or x.lower == 'y':
|
||||
read_config()
|
||||
|
||||
elif x == "n":
|
||||
os.remove(f"{hlna.home_dir}/.config/hlna/bot/hlna_bot")
|
||||
start()
|
||||
|
||||
|
||||
def check_status(m):
|
||||
for i in list_config:
|
||||
data = hlna.read_yaml(i)
|
||||
hlna.print_line(data)
|
||||
x = os.system(f"lsof -w -i :{data['Port']}")
|
||||
n_srv = data['SessionName']
|
||||
if x != 0:
|
||||
hlna_bot.send_message(m.chat.id, text=f"Сервер {n_srv} не работает 🛑")
|
||||
elif x == 0:
|
||||
hlna_bot.send_message(m.chat.id, text=f"Сервер {n_srv} запустился 🟢")
|
||||
|
||||
|
||||
def read_config():
|
||||
with open(f"{hlna.home_dir}/.config/hlna/bot/hlna_bot", "r") as yamlfile:
|
||||
data = yaml.load(yamlfile, Loader=yaml.FullLoader)
|
||||
return data[0]
|
||||
|
||||
|
||||
def create_config(key_api):
|
||||
config_hlna_bot = [
|
||||
{
|
||||
'key_api': key_api,
|
||||
}
|
||||
]
|
||||
with open(f"{hlna.home_dir}/.config/hlna/bot/hlna_bot", 'w') as yamlfile:
|
||||
yaml.dump(config_hlna_bot, yamlfile)
|
||||
print(colorama.Fore.GREEN + "Конфиг бота создан" + colorama.Style.RESET_ALL)
|
||||
|
||||
|
||||
start()
|
||||
|
||||
|
||||
data = read_config()
|
||||
k_api = data['key_api']
|
||||
hlna_bot = telebot.TeleBot(f"{k_api}", parse_mode=None)
|
||||
|
||||
|
||||
@hlna_bot.message_handler(commands=["start"])
|
||||
def m_mainmenu(m):
|
||||
menu = types.ReplyKeyboardMarkup(resize_keyboard=True)
|
||||
k_start = types.KeyboardButton("🦖 Старт")
|
||||
k_stop = types.KeyboardButton("🦇 Выключить")
|
||||
menu.add(k_start, k_stop)
|
||||
|
||||
k_restart = types.KeyboardButton("🦕 Перезагрузить")
|
||||
k_update = types.KeyboardButton("🔸 Обновить")
|
||||
menu.add(k_restart, k_update)
|
||||
|
||||
k_status = types.KeyboardButton("🧾 Статус")
|
||||
menu.add(k_status)
|
||||
|
||||
k_mods = types.KeyboardButton("⏬ Установка модов")
|
||||
menu.add(k_mods)
|
||||
|
||||
hlna_bot.send_message(m.chat.id, text="Привет, {0.first_name}! Это панель управления сервером".format(m.from_user),
|
||||
reply_markup=menu)
|
||||
|
||||
|
||||
@hlna_bot.message_handler(content_types=["text"])
|
||||
def m_chose(m):
|
||||
if m.text.strip() == '🦖 Старт':
|
||||
if list_config == []:
|
||||
menu_install = types.ReplyKeyboardMarkup(resize_keyboard=True)
|
||||
k_install = types.KeyboardButton("Установить")
|
||||
k_back = types.KeyboardButton("Назад")
|
||||
menu_install.add(k_back, k_install)
|
||||
hlna_bot.send_message(m.chat.id, text="{0.first_name}, ниодна карта не установлена".format(m.from_user),
|
||||
reply_markup=menu_install)
|
||||
hlna_bot.register_next_step_handler(m, m_install)
|
||||
else:
|
||||
hlna.read_yaml(list_config[-1], flag=True)
|
||||
menu_start = types.ReplyKeyboardMarkup(resize_keyboard=True)
|
||||
for i in list_config:
|
||||
k_mapname = types.KeyboardButton(i)
|
||||
menu_start.add(k_mapname)
|
||||
k_back = types.KeyboardButton("Назад")
|
||||
menu_start.add(k_back)
|
||||
hlna_bot.send_message(m.chat.id, text="{0.first_name}, выбери карту".format(m.from_user), reply_markup=menu_start)
|
||||
elif m.text.strip() == '🧾 Статус':
|
||||
check_status(m)
|
||||
elif m.text.strip() == 'Назад':
|
||||
m_mainmenu(m)
|
||||
for i in list_config:
|
||||
if m.text.strip() == f'{i}':
|
||||
hlna.print_line(f"Старт карты {i}")
|
||||
hlna.start()
|
||||
|
||||
# @hlna_bot.message_handler(content_types=["text"])
|
||||
# def m_mapstart(m):
|
||||
# hlna.print_line("Старт карты")
|
||||
# for i in list_config:
|
||||
# if m.text.strip() == f'{i}':
|
||||
# hlna.print_line(f"Старт карты {i}")
|
||||
# hlna.start()
|
||||
|
||||
|
||||
def m_install(m):
|
||||
if m.text.strip() == 'Установить':
|
||||
hlna_bot.send_message(m.chat.id, subprocess.check_output(hlna.config(), shell=True))
|
||||
|
||||
|
||||
print("Бот запущен")
|
||||
hlna_bot.polling(none_stop=True, interval=0, timeout=999)
|
||||
|
||||
# subprocess.run(["arkmanager", "start", "--alwaysrestart", "@all"])
|
||||
# elif m.text.strip() == '🦇 Выключить':
|
||||
# subprocess.run(["arkmanager", "stop", "@all"])
|
||||
# elif m.text.strip() == '🦕 Перезагрузить':
|
||||
# subprocess.run(["arkmanager","restart", "--alwaysrestart", "--warn", "@all"])
|
||||
# elif (m.text.strip() == '🔸 Обновить'):
|
||||
# subprocess.run(["arkmanager", "update", "--update-mods", "@all"])
|
||||
#
|
||||
#
|
||||
# elif (m.text.strip() == '⏬ Установка модов'):
|
||||
# bot.send_message(m.from_user.id, "Введите id модификации")
|
||||
# bot.register_next_step_handler(m, mods)
|
||||
# else:
|
||||
# pass
|
||||
# else:
|
||||
# if m.text.strip() == 'Zxx843Aj':
|
||||
# th = Thread(target=check, args=(m,))
|
||||
# th.start()
|
||||
# id.append(m.chat.id)
|
||||
# bot.send_message(m.chat.id, text="{0.first_name} вы авторизованы, нажмите /start для работы с ботом".format(m.from_user))
|
||||
# return id
|
||||
# def mods(m):
|
||||
# id = m.text.strip()
|
||||
# subprocess.run(["arkmanager", "installmods", id, "@all"])
|
||||
# subprocess.run(["arkmanager", "enablemod", id, "@all"])
|
||||
Reference in New Issue
Block a user