- 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