mirror of
https://github.com/SantaSpeen/Rcon-VK-Bot.git
synced 2026-05-19 08:00:26 +00:00
[~] ping in server check
[~] Bot to modules/bot.py [~] 1.3.0 -> 1.3.1
This commit is contained in:
+11
-118
@@ -1,144 +1,37 @@
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
import requests
|
||||
import vk
|
||||
from loguru import logger
|
||||
|
||||
from modules import config, rcon, get_server_status, enter_to_exit, Permissions, __version__, new_version
|
||||
|
||||
|
||||
class Bot:
|
||||
|
||||
def __init__(self):
|
||||
self.vk = vk.API(access_token=config.vk.token, v=5.199)
|
||||
self.group_id = self.vk.groups.getById()['groups'][0]['id']
|
||||
with open('help_message.txt', encoding="utf-8") as f:
|
||||
self.help_message = f.read()
|
||||
logger.info(f"[BOT] ID группы: {self.group_id}")
|
||||
|
||||
def get_lp_server(self):
|
||||
lp = self.vk.groups.getLongPollServer(group_id=self.group_id)
|
||||
return lp.get('server'), lp.get('key'), lp.get('ts')
|
||||
|
||||
def write(self, peer_id, message):
|
||||
if len(message) > 4095:
|
||||
messages = (len(message) // 4095)
|
||||
for i in range(1, messages + 1):
|
||||
if i > 30:
|
||||
logger.error("[BOT] Сообщение слишком длинное...")
|
||||
break
|
||||
self.write(peer_id, message[:4095 * i])
|
||||
else:
|
||||
self.vk.messages.send(message=message, peer_id=peer_id, random_id=0)
|
||||
|
||||
def rcon_cmd_handle(self, cmd, from_id, peer_id, _write=True, _allow=False):
|
||||
a, r = perms.is_allowed(from_id, cmd.split()[0])
|
||||
if _allow:
|
||||
r = cmd
|
||||
if a or _allow:
|
||||
answer = rcon(cmd)
|
||||
logger.info(f"[BOT] User: {from_id}({r}) in Chat: {peer_id} use RCON cmd: \"{cmd}\", "
|
||||
f"with answer: \"{answer}\"")
|
||||
if _write:
|
||||
self.write(peer_id, f"RCON:\n{answer}")
|
||||
else:
|
||||
return answer
|
||||
else:
|
||||
logger.info(f"[BOT] User: {from_id}({r}) in Chat: {peer_id} no have rights RCON cmd: \"{cmd}\".")
|
||||
|
||||
def _bot_handle(self, message):
|
||||
from_id = message['from_id']
|
||||
global perms
|
||||
if perms.is_allowed(from_id, "bot"):
|
||||
peer_id = message['peer_id']
|
||||
text = message['text']
|
||||
logger.info(f"[BOT] {peer_id}:{from_id}:{text}")
|
||||
tsplit = text.split(" ")
|
||||
cmds = ("Доступные команды:\n"
|
||||
" .bot help - Вывести это сообщение.\n"
|
||||
" .bot info - Выводит краткую информацию о боте."
|
||||
" .bot perm user <> - Добавить пользователя \n"
|
||||
" .bot perm reload - Перезагружает пермишины")
|
||||
if len(tsplit) == 1:
|
||||
self.write(peer_id, cmds)
|
||||
return
|
||||
match tsplit[1]:
|
||||
case "perm":
|
||||
match tsplit[2] if len(tsplit) > 2 else None:
|
||||
case "reload":
|
||||
perms = Permissions.load()
|
||||
self.write(peer_id, "Права перезагружены")
|
||||
case _:
|
||||
self.write(peer_id, ".bot perm ?")
|
||||
case "info":
|
||||
st = get_server_status()
|
||||
self.write(peer_id, f"RconVkBot\n"
|
||||
f"Версия бота: {__version__}, последняя: {not new_version}\n"
|
||||
f"Пинг до сервера: {st.latency:.3f}ms\n"
|
||||
f"Бедрок: {st.motd.bedrock} ({st.version.protocol})\n")
|
||||
case _:
|
||||
self.write(peer_id, cmds)
|
||||
|
||||
def message_handle(self, message):
|
||||
from_id = message['from_id']
|
||||
peer_id = message['peer_id']
|
||||
text = message['text']
|
||||
match text:
|
||||
case i if i.startswith(".rcon "):
|
||||
self.rcon_cmd_handle(i[6:], from_id, peer_id)
|
||||
case i if i.startswith(".bot"):
|
||||
self._bot_handle(message)
|
||||
case "!help":
|
||||
logger.info(f"[BOT] {peer_id}:{from_id}:{text}")
|
||||
self.write(peer_id, self.help_message)
|
||||
case "!online":
|
||||
logger.info(f"[BOT] {peer_id}:{from_id}:{text}")
|
||||
players = get_server_status().players
|
||||
self.write(peer_id, f"На сервере сейчас {players.online}/{players.max}")
|
||||
case "!id":
|
||||
logger.info(f"[BOT] {peer_id}:{from_id}:{text}")
|
||||
self.write(peer_id,
|
||||
f"Твой ID: {from_id}\nРоль: {perms.get_role(from_id)}\nНик: {perms.get_nick(from_id)}")
|
||||
|
||||
def listen(self):
|
||||
server, key, ts = self.get_lp_server()
|
||||
logger.info("[BOT] Начинаю получать сообщения..")
|
||||
while True:
|
||||
lp = requests.get(f'{server}?act=a_check&key={key}&ts={ts}&wait=25').json()
|
||||
try:
|
||||
if lp.get('failed') is not None:
|
||||
key = self.get_lp_server()[1]
|
||||
if ts != lp.get('ts') and lp.get('updates'):
|
||||
updates = lp['updates'][0]
|
||||
if updates['type'] == "message_new":
|
||||
self.message_handle(updates['object']['message'])
|
||||
ts = lp.get('ts')
|
||||
except Exception as i:
|
||||
ts = lp.get('ts')
|
||||
logger.exception(i)
|
||||
|
||||
from modules import config, rcon, get_server_status, enter_to_exit
|
||||
from modules.bot import Bot
|
||||
from modules.perms import Permissions
|
||||
|
||||
if __name__ == '__main__':
|
||||
Permissions.perm_file = Path(config.permissions_file)
|
||||
perms = Permissions.load()
|
||||
# Check token
|
||||
if not config.vk.token:
|
||||
logger.error("Токен ВК не найден.")
|
||||
enter_to_exit()
|
||||
bot = Bot()
|
||||
# Test RCON
|
||||
print("Проверка RCON..", end="")
|
||||
if rcon("list").startswith("Rcon error"):
|
||||
logger.error("RCON не отвечает. Проверьте блок \"rcon\" в config.json")
|
||||
enter_to_exit()
|
||||
logger.info("RCON доступен.")
|
||||
# Test Minecraft Server
|
||||
print("Проверка сервера..", end="")
|
||||
try:
|
||||
_players = get_server_status().players
|
||||
logger.info(f"Проверка сервера. Онлайн: {_players.online}/{_players.max}")
|
||||
st = get_server_status()
|
||||
players = st.players
|
||||
logger.info(f"Сервер доступен. Пинг: {st.latency:.3f}ms, Онлайн: {players.online}/{players.max}")
|
||||
except Exception as e:
|
||||
logger.exception(e)
|
||||
logger.info("Сервер не отвечает. Проверьте блок \"minecraft\" в config.json")
|
||||
enter_to_exit()
|
||||
try:
|
||||
bot = Bot()
|
||||
bot.listen()
|
||||
except KeyboardInterrupt:
|
||||
sys.exit(0)
|
||||
|
||||
@@ -11,10 +11,12 @@ from pathlib import Path
|
||||
import requests
|
||||
from loguru import logger
|
||||
from mcrcon import MCRcon
|
||||
from ruamel.yaml import YAML
|
||||
|
||||
from .perms import Permissions, yaml
|
||||
yaml = YAML()
|
||||
yaml.default_flow_style = False
|
||||
|
||||
__version__ = '1.3.0'
|
||||
__version__ = '1.3.1'
|
||||
|
||||
raw_config = """\
|
||||
{
|
||||
@@ -64,7 +66,7 @@ def init_logger():
|
||||
os.remove(file)
|
||||
logger.remove(0)
|
||||
logger.add(log_file)
|
||||
logger.add(sys.stdout, format="<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> | "
|
||||
logger.add(sys.stdout, format="\r<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> | "
|
||||
"<level>{level: <8}</level> | {message}")
|
||||
|
||||
|
||||
@@ -91,7 +93,6 @@ else:
|
||||
host = config.rcon.host
|
||||
port = config.rcon.port
|
||||
password = config.rcon.password
|
||||
Permissions.perm_file = Path(config.permissions_file)
|
||||
|
||||
|
||||
def rcon(cmd):
|
||||
@@ -131,4 +132,4 @@ def new_version():
|
||||
return False
|
||||
|
||||
|
||||
new_version = new_version()
|
||||
is_new_version = new_version()
|
||||
|
||||
@@ -0,0 +1,119 @@
|
||||
import requests
|
||||
import vk
|
||||
from loguru import logger
|
||||
|
||||
import modules
|
||||
from modules import config, rcon, get_server_status, is_new_version
|
||||
from modules.perms import Permissions
|
||||
|
||||
|
||||
class Bot:
|
||||
|
||||
def __init__(self):
|
||||
self.vk = vk.API(access_token=config.vk.token, v=5.199)
|
||||
print("Получение GroupID..", end="")
|
||||
self.group_id = self.vk.groups.getById()['groups'][0]['id']
|
||||
with open('help_message.txt', encoding="utf-8") as f:
|
||||
self.help_message = f.read()
|
||||
logger.info(f"[BOT] ID группы: {self.group_id}")
|
||||
|
||||
def get_lp_server(self):
|
||||
lp = self.vk.groups.getLongPollServer(group_id=self.group_id)
|
||||
return lp.get('server'), lp.get('key'), lp.get('ts')
|
||||
|
||||
def write(self, peer_id, message):
|
||||
if len(message) > 4095:
|
||||
messages = (len(message) // 4095)
|
||||
for i in range(1, messages + 1):
|
||||
if i > 30:
|
||||
logger.error("[BOT] Сообщение слишком длинное...")
|
||||
break
|
||||
self.write(peer_id, message[:4095 * i])
|
||||
else:
|
||||
self.vk.messages.send(message=message, peer_id=peer_id, random_id=0)
|
||||
|
||||
def rcon_cmd_handle(self, cmd, from_id, peer_id, _write=True, _allow=False):
|
||||
a, r = perms.is_allowed(from_id, cmd.split()[0])
|
||||
if _allow:
|
||||
r = cmd
|
||||
if a or _allow:
|
||||
answer = rcon(cmd)
|
||||
logger.info(f"[BOT] User: {from_id}({r}) in Chat: {peer_id} use RCON cmd: \"{cmd}\", "
|
||||
f"with answer: \"{answer}\"")
|
||||
if _write:
|
||||
self.write(peer_id, f"RCON:\n{answer}")
|
||||
else:
|
||||
return answer
|
||||
else:
|
||||
logger.info(f"[BOT] User: {from_id}({r}) in Chat: {peer_id} no have rights RCON cmd: \"{cmd}\".")
|
||||
|
||||
def _bot_handle(self, message):
|
||||
from_id = message['from_id']
|
||||
global perms
|
||||
if perms.is_allowed(from_id, "bot"):
|
||||
peer_id = message['peer_id']
|
||||
text = message['text']
|
||||
logger.info(f"[BOT] {peer_id}:{from_id}:{text}")
|
||||
tsplit = text.split(" ")
|
||||
cmds = ("Доступные команды:\n"
|
||||
" .bot help - Вывести это сообщение.\n"
|
||||
" .bot info - Выводит краткую информацию о боте.\n"
|
||||
" .bot perm user <> - Добавить пользователя \n"
|
||||
" .bot perm reload - Перезагружает пермишины")
|
||||
if len(tsplit) == 1:
|
||||
self.write(peer_id, cmds)
|
||||
return
|
||||
match tsplit[1]:
|
||||
case "perm":
|
||||
match tsplit[2] if len(tsplit) > 2 else None:
|
||||
case "reload":
|
||||
perms = Permissions.load()
|
||||
self.write(peer_id, "Права перезагружены")
|
||||
case _:
|
||||
self.write(peer_id, ".bot perm ?")
|
||||
case "info":
|
||||
st = get_server_status()
|
||||
self.write(peer_id, f"RconVkBot\n"
|
||||
f"Версия бота: {modules.__version__}, последняя: {not is_new_version}\n"
|
||||
f"Пинг до сервера: {st.latency:.3f}ms\n"
|
||||
f"Бедрок: {st.motd.bedrock} ({st.version.protocol})\n")
|
||||
case _:
|
||||
self.write(peer_id, cmds)
|
||||
|
||||
def message_handle(self, message):
|
||||
from_id = message['from_id']
|
||||
peer_id = message['peer_id']
|
||||
text = message['text']
|
||||
match text:
|
||||
case i if i.startswith(".rcon "):
|
||||
self.rcon_cmd_handle(i[6:], from_id, peer_id)
|
||||
case i if i.startswith(".bot"):
|
||||
self._bot_handle(message)
|
||||
case "!help":
|
||||
logger.info(f"[BOT] {peer_id}:{from_id}:{text}")
|
||||
self.write(peer_id, self.help_message)
|
||||
case "!online":
|
||||
logger.info(f"[BOT] {peer_id}:{from_id}:{text}")
|
||||
players = get_server_status().players
|
||||
self.write(peer_id, f"На сервере сейчас {players.online}/{players.max}")
|
||||
case "!id":
|
||||
logger.info(f"[BOT] {peer_id}:{from_id}:{text}")
|
||||
self.write(peer_id,
|
||||
f"Твой ID: {from_id}\nРоль: {perms.get_role(from_id)}\nНик: {perms.get_nick(from_id)}")
|
||||
|
||||
def listen(self):
|
||||
server, key, ts = self.get_lp_server()
|
||||
logger.info("[BOT] Начинаю получать сообщения..")
|
||||
while True:
|
||||
lp = requests.get(f'{server}?act=a_check&key={key}&ts={ts}&wait=25').json()
|
||||
try:
|
||||
if lp.get('failed') is not None:
|
||||
key = self.get_lp_server()[1]
|
||||
if ts != lp.get('ts') and lp.get('updates'):
|
||||
updates = lp['updates'][0]
|
||||
if updates['type'] == "message_new":
|
||||
self.message_handle(updates['object']['message'])
|
||||
ts = lp.get('ts')
|
||||
except Exception as i:
|
||||
ts = lp.get('ts')
|
||||
logger.exception(i)
|
||||
@@ -3,10 +3,8 @@ import sys
|
||||
from pathlib import Path
|
||||
|
||||
from loguru import logger
|
||||
from ruamel.yaml import YAML
|
||||
|
||||
yaml = YAML()
|
||||
yaml.default_flow_style = False
|
||||
from modules import yaml
|
||||
|
||||
|
||||
class Permissions:
|
||||
|
||||
Reference in New Issue
Block a user