From cbecbcecda0cf3a5a0d31c012cd28416bade7b4e Mon Sep 17 00:00:00 2001 From: xpamych Date: Wed, 10 May 2023 12:23:31 +0300 Subject: [PATCH] Revert "change if else and fix function servers" This reverts commit d4eece79e3a7f904a5424a98c68b626c8075ec1a. --- LICENSE.md | 0 README.md | 0 hlna.py | 276 ++++++++++++++++++++++++++------------------ hlna_discord_bot.py | 0 logo.png | Bin pytest.py | 0 6 files changed, 161 insertions(+), 115 deletions(-) mode change 100755 => 100644 LICENSE.md mode change 100755 => 100644 README.md mode change 100755 => 100644 hlna_discord_bot.py mode change 100755 => 100644 logo.png mode change 100755 => 100644 pytest.py diff --git a/LICENSE.md b/LICENSE.md old mode 100755 new mode 100644 diff --git a/README.md b/README.md old mode 100755 new mode 100644 diff --git a/hlna.py b/hlna.py index c1d3c38..b4b32cd 100755 --- a/hlna.py +++ b/hlna.py @@ -1,23 +1,28 @@ #!/usr/bin/env python3 import os -import time -import threading - -from pathlib import Path -from pprint import pprint +import datetime import yaml import click import colorama -import click_completion +from pathlib import Path from rcon.source import Client home_dir = Path.home() -dir_server = f"{home_dir}/ARK_Servers/" -dir_config = f"{home_dir}/.config/hlna/maps/" +dir_server = f"{home_dir}/Servers/" +dir_server_ark = f"{dir_server}/ARK/" +server_dir = f"{dir_server_ark}ShooterGame/Binaries/Linux/" +dir_unit = f"{home_dir}/.config/systemd/user/" +dir_config = f"{home_dir}/.config/hlna/" +dir_logs = f"{dir_config}logs/" +dir_maps = f"{dir_config}maps/" +dir_deactivated = f"{dir_maps}deactivated/" + mods_id = "" -listen_server = True +now = datetime.datetime.now() +date = now.strftime("%Y-%m-%d") +time = now.strftime("%H:%M:%S") @click.group() @@ -25,6 +30,20 @@ def hlna(): pass +def create_dir(directory): + """Проверка и создание директории""" + if not os.path.exists(directory): + os.mkdir(directory) + + +create_dir(dir_server) +create_dir(dir_server_ark) +create_dir(dir_unit) +create_dir(dir_config) +create_dir(dir_logs) +create_dir(dir_maps) + + def find_file(path): """Находим все конфиги в зависимости от пути""" arr = next(os.walk(path), (None, None, []))[2] # [] if no file @@ -37,28 +56,17 @@ def find_file(path): return arr -list_config = find_file(dir_config) -delist_config = find_file(dir_config+"deactivated") +list_config = find_file(dir_maps) +delist_config = find_file(dir_deactivated) def print_line(text): """Добавление тире вокруг текста, покраска""" - print(colorama.Fore.YELLOW + "-"*30) + print(colorama.Fore.YELLOW + "-" * 30) print(f"{colorama.Fore.GREEN} + {text}") - print(colorama.Fore.YELLOW + "-"*30 + colorama.Style.RESET_ALL) + print(colorama.Fore.YELLOW + "-" * 30 + colorama.Style.RESET_ALL) -def create_dir(directory): - """Проверка и создание директории""" - if not os.path.exists(directory): - os.mkdir(directory) - if not os.path.exists(f"{home_dir}/.config/hlna/"): - os.chdir(f"{home_dir}/.config/") - os.mkdir("hlna") - if not os.path.exists(f"{home_dir}/.config/hlna/maps"): - os.chdir(f"{home_dir}/.config/hlna/") - os.mkdir("maps") - def check_int(number=""): """Проверка на ввод числа""" while True: @@ -69,28 +77,36 @@ def check_int(number=""): x = int(x) return x except ValueError: - print("Введите число") + print_line("Введите число") -create_dir(dir_server) -create_dir(dir_config) +@hlna.command(help='Сбор настроек для сервера или кластера') +def config(): + while True: + count_server = check_int("""Выберите игру для конфигурирования + 1. ARK Survival Evolved + : """) + if count_server == 1: + config_ark() + break + else: + print_line("Пока есть только ARK xD") -@hlna.command(help='Для конфигурирования параметров запускаемого сервера или кластера серверов') -def config(list_config=list_config): +def config_ark(list_config=list_config): + """Сбор данных для конфига""" data = {} port_s = [] rcon_p = [] query_p = [] - + cluster_id = "" cluster_dir_override = "" - + count_cluster = check_int("""Укажите требуется ли кластер? 1. Да 2. Нет : """) - if count_cluster == 1: cluster_server = True cluster_id = input("Укажите id для кластера, любое сочетание символов: \n") @@ -104,17 +120,14 @@ def config(list_config=list_config): for i in list_config: data = read_yaml(i) print(f"{i} : {data['map']}") - # id_srv[data['SessionName']] = data['id_server'] count_maps = check_int("Укажите количество карт: \n") if count_maps == 0: # 0 возвращает check_int когда, ничего не было введено count_maps = 1 - + for i in range(count_maps): while True: "Проверка на выбор карты из списка" - - # os.system("clear") amount_map = check_int("""Выберите карту из списка указав номер 1. The Island 2. The Center @@ -180,11 +193,10 @@ def config(list_config=list_config): if port in ports_arr: print("Порт уже занят") else: - return port - + return port + if list_config: data = read_yaml(list_config[-1]) - while True: name_server = input("Укажите название Сервера: \n") if name_server == "": @@ -194,7 +206,6 @@ def config(list_config=list_config): while new_name in list_config: new_name = f"{map_s}{str(count)}" count += 1 - list_config.append(new_name) print(list_config) break @@ -203,20 +214,19 @@ def config(list_config=list_config): break else: if name_server in list_config: - print("Имя занято") + print_line("Имя занято") else: list_config.append(name_server) # если enter, то ставим последним элементом карту break + print("Укажите порт сервера:\n") port_server = ports(port_s) print("Укажите query порт сервера:\n") query_port = ports(query_p) - if port_server == 0: port_server = 7777 if query_port == 0: query_port = 27015 - print("Порт Сервера=", port_server) print("Query Port=", query_port) @@ -231,41 +241,87 @@ def config(list_config=list_config): max_players = check_int("Укажите максимальное количество игроков: \n") if max_players == 0: max_players = 70 - - yaml_create(cluster_server, map_s, list_config[-1], port_server, query_port, rcon_enabled, rcon_port, adminpassword_server, password_server, max_players, cluster_id, cluster_dir_override) + + print("Передавать сервер в глобальный список серверов steam?") + listen_server_amount = check_int("""\n + 1. Да + 2. Нет + :""") + if listen_server_amount == 1: + listen_server = True + elif listen_server_amount == 2: + listen_server = False + else: + listen_server = True + + yaml_create(cluster_server, map_s, list_config[-1], port_server, query_port, rcon_enabled, rcon_port, + adminpassword_server, password_server, max_players, cluster_id, cluster_dir_override, listen_server) -def yaml_create(cluster_server, map_s, name_server, port_server, query_port, rcon_enabled, rcon_port, adminpassword_server, password_server, max_players, cluster_id, cluster_dir_override): +def yaml_create(cluster_server, map_s, name_server, port_server, query_port, rcon_enabled, rcon_port, + adminpassword_server, password_server, max_players, cluster_id, cluster_dir_override, listen_server): settings_hlna = [ { - 'map' : map_s, - 'Cluster' : cluster_server, - 'SessionName' : name_server, - 'Port' : port_server, - 'QueryPort' : query_port, - 'RCONEnabled' : rcon_enabled, - 'RCONPort' : rcon_port, + 'map': map_s, + 'Cluster': cluster_server, + 'SessionName': name_server, + 'Port': port_server, + 'QueryPort': query_port, + 'RCONEnabled': rcon_enabled, + 'RCONPort': rcon_port, 'ServerAdminPassword': adminpassword_server, - 'ServerPassword' : password_server, - 'MaxPlayers' : max_players, - 'ModsId' : mods_id, - 'Listen' : listen_server, - 'ServerPath' : dir_server, - 'clusterid' : cluster_id, - 'clusterdir' : cluster_dir_override + 'ServerPassword': password_server, + 'MaxPlayers': max_players, + 'ModsId': mods_id, + 'Listen': listen_server, + 'ServerPath': dir_server, + 'clusterid': cluster_id, + 'clusterdir': cluster_dir_override } ] - with open(dir_config + f"{name_server}", 'w') as yamlfile: + with open(dir_maps + f"{name_server}", 'w') as yamlfile: yaml.dump(settings_hlna, yamlfile) print(colorama.Fore.GREEN + "Конфиг создан" + colorama.Style.RESET_ALL) + systemd_unit_create() + + +def systemd_unit_create(name_server=list_config): + for i in name_server: + data = read_yaml(i) + if not data['Cluster']: + ntff = "" + else: + ntff = "-NoTransferFromFiltering" + unit_file = f"{dir_unit}ARK_{data['SessionName']}.service" + unit_text = f'''[Unit] +Description=ARK: Survival Evolved dedicated server - {data['map']} +Wants=network-online.target +After=syslog.target network.target nss-lookup.target network-online.target + +[Service] +ExecStartPre=/usr/bin/steamcmd +force_install_dir {dir_server} +login anonymous +app_update 376030 +quit +TimeoutStartSec=1200 +ExecStart={dir_server_ark}ShooterGameServer {data['map']}?listen={data['Listen']}?SessionName={data['SessionName']}?Port={data['Port']}?QueryPort={data['QueryPort']}?RCONEnabled={data['RCONEnabled']}?RCONPort={data['RCONPort']}?ServerAdminPassword={data['ServerAdminPassword']}?MaxPlayers={data['MaxPlayers']} -clusterid={data['clusterid']} -ClusterDirOverride={data['clusterdir']} {ntff} +LimitNOFILE=100000 +ExecReload=/bin/kill -s HUP $MAINPID +ExecStop=/bin/kill -s INT $MAINPID + +[Install] +WantedBy=default.target +''' + with open(unit_file, 'w') as systemd_unit: + systemd_unit.write(unit_text) + os.system(f"systemctl --user unmask ARK_{data['SessionName']}.service") + os.system(f"systemctl --user enable ARK_{data['SessionName']}.service") + os.system('systemctl --user daemon-reload') def test_mod_install(): pathTest = f"{dir_server}ShooterGame/Saved/Config/LinuxServer/" if os.path.exists(pathTest): os.chdir(pathTest) -# "Добавить файл в Game.ini, если модов несколько добавляем еще строку ModIDS= [ModInstaller] ModIDS=" -# c этим не разобрался + # "Добавить файл в Game.ini, если модов несколько добавляем еще строку ModIDS= [ModInstaller] ModIDS=" + # c этим не разобрался os.system("echo ActiveMods=2943454417 >> GameUserSettings.ini") else: print_line("Сервер не установлен") @@ -277,11 +333,12 @@ test_mod_install() @hlna.command() @click.option("-m", required=True, help="Название Сервера") @click.option("-e/-d", default=True, help="-e активировать карты, -d деактивировать") -def enablemap(m,e): - """Тут переписать надо""" - m = m.split(",") - # m = check_name_map(m, False) - create_dir(dir_config + "deactivated") +def enablemap(m, e): + """Включение/выключение карт""" + + m = m.split(",") + if not os.path.isdir(dir_deactivated): + create_dir(dir_deactivated) if e == True: port_s = [] query_p = [] @@ -297,33 +354,41 @@ def enablemap(m,e): data = read_yaml(i) if data['Port'] in port_s: print("Предлагаем заменить") + continue if data['QueryPort'] in query_p: print("Заменить query port?") - x = os.system(f"mv {dir_config}deactivated/{i} {dir_config + i} 2>> {dir_config}logs") #Добавить текущее время + continue + x = os.system(f"mv {dir_deactivated}{i} {dir_maps} >> {dir_logs}{date} 2>&1") # Добавить текущее время + with open(f"{dir_logs}{date}.log", "a") as f: + f.write(f"[{time}] File {i} has been moved to {dir_maps}\n") if x == 0: print(f"Карта активирована - {i}") else: print(f"{i} либо уже активирована, либо такой карты нет") - except: - print("ошибка") + finally: pass + # except: + # print("ошибка при активации карты") + # pass else: - for i in m: try: if i in delist_config: print(f"Карта {i} уже есть в деактивированных") continue - x = os.system(f"mv {dir_config + i} {dir_config}deactivated/{i} 2>> {dir_config}logs") #Добавить текущее время + x = os.system( + f"mv {dir_maps}{i} {dir_deactivated} >> {dir_logs}{date} 2>&1") # Добавить текущее время + with open(f"{dir_logs}{date}.log", "a") as f: + f.write(f"[{time}] File {i} has been moved to {dir_deactivated}\n") if x == 0: print(f"Карта деактивирована - {i}") else: print(f"{i} либо деактивирована, либо такой карты нет") except: pass - -@hlna.command() + +@hlna.command() def servers(map_server=list_config): # Добавить сортивку по кластерам и вывод несколько столбиков if map_server == [] and delist_config == []: @@ -349,10 +414,10 @@ def servers(map_server=list_config): Rcon порт : {data['RCONPort']} Максимальное кол-во игроков: {data['MaxPlayers']}""") print("-" * 40) - - if delist_config!=[]: + + if delist_config != []: x = input("Есть неактивные сервера, показать Y/n: ") - if x!="n": + if x != "n": for i in delist_config: data = read_yaml(i, False) print(f""" @@ -368,8 +433,8 @@ def servers(map_server=list_config): Rcon порт : {data['RCONPort']} Максимальное кол-во игроков: {data['MaxPlayers']}""") print("-" * 40) - - + + @hlna.command(help='Для запуска, сконфигурированного сервера или кластера') @click.option('-m', default='all', help="Название карты для запуска или all для запуска все карт") @click.option('-b', default='', help="") @@ -396,36 +461,19 @@ def start(m, b, name_server=list_config): else: names_serverstart = choose_map(names_serverstart) - server_dir = dir_server + "ShooterGame/Binaries/Linux/" - print_line("Валидация файлов сервера") - x = os.system(f"steamcmd +force_install_dir {dir_server} +login anonymous +app_update 376030 +quit") - os.chdir(server_dir) for i in names_serverstart: data = read_yaml(i) - starting_map = dict_mapname[i] - print_line(data['Cluster']) - ntff = '' if not data['Cluster'] else "-NoTransferFromFiltering" - def starting(i): - os.system(f"{server_dir}ShooterGameServer {i}?SessionName={data['SessionName']}?Port={data['Port']}\ - ?QueryPort={data['QueryPort']}?RCONEnabled={data['RCONEnabled']}?RCONPort={data['RCONPort']}\ - ?ServerAdminPassword={data['ServerAdminPassword']}?MaxPlayers={data['MaxPlayers']}\ - ?GameModIds={data['ModsId']}?listen={data['Listen']} -clusterid={data['clusterid']}\ - -ClusterDirOverride={data['clusterdir']} {ntff}") - if x == 0: - print_line("Запускаем карту" + i) - threads = threading.Thread(target=starting, args=(starting_map,)) - threads.start() - else: - print(f"Карта не запущена, сервер не установлен") - + os.system(f"systemctl --user start ARK_{data['SessionName']}.service") else: print("Ни одной карты не установлено") - -def read_yaml(name_server, flag=True): + +def read_yaml(list_config, flag=True): # Читаем конфиги активных или неактивных карт в зависимости от флага - dirs = f"{dir_config}{name_server}" if flag else f"{dir_config}deactivated/{name_server}" - + if flag: + dirs = f"{dir_maps}{list_config}" + else: + dirs = f"{dir_deactivated}{list_config}" with open(dirs, "r") as yamlfile: data = yaml.load(yamlfile, Loader=yaml.FullLoader) return data[0] # возвращаем словарь со всеми значениями @@ -435,27 +483,27 @@ def choose_map(arr): new_arr = [] arr = sorted(arr) print('Найдены сервера с этой картой') - for i,map in enumerate(arr): - print(f"{i+1}) {map}") + for i, map in enumerate(arr): + print(f"{i + 1}) {map}") while True: try: x = input("Выберите сервер из списка, либо несколько через запятую: ").split(',') - x = [int (i) for i in x] + x = [int(i) for i in x] break except: print("Неправильный ввод") - + for i in x: - new_arr.append(arr[i-1]) + new_arr.append(arr[i - 1]) print('Выбранные сервера:', new_arr) - + return new_arr @hlna.command() -@click.argument('c', nargs=1) +@click.argument('c', nargs=1) @click.option('-m', required=True, help="Название карты для применения rcon команды") -def rcon(m,c): +def rcon(m, c): print_line("Команда: ", c) print_line("Карты: ", m) dict_mapname = {} @@ -466,7 +514,7 @@ def rcon(m,c): dict_adminpwd[data['RCONPort']] = data['ServerAdminPassword'] rcon_ports = [] for ns, v in dict_mapname.items(): - print_line(f"переменные v и m {v} & {m}") # обьединить с таким же блоком в start() + print_line(f"переменные v и m {v} & {m}") # обьединить с таким же блоком в start() if v in m: rcon_ports.append(ns) print_line(f"Карта которая запускается {ns}") @@ -482,7 +530,5 @@ def zero(x=""): return "" -# if __name__ == 'hlna': -# input = zero if __name__ == '__main__': hlna() diff --git a/hlna_discord_bot.py b/hlna_discord_bot.py old mode 100755 new mode 100644 diff --git a/logo.png b/logo.png old mode 100755 new mode 100644 diff --git a/pytest.py b/pytest.py old mode 100755 new mode 100644