diff --git a/dodo.py b/dodo.py index 7772b1f..4981db9 100755 --- a/dodo.py +++ b/dodo.py @@ -2,22 +2,26 @@ import os import yaml import threading -#import click -import fire from pathlib import Path +import click +import colorama + home_dir = Path.home() dir_server = f"{home_dir}/ARK_Servers/" config_dodo = f"{home_dir}/.config/dodo/" mods_id = "" listen_server = True + +@click.group() +def dodo(): + pass + "------Узнаем список карт--------------" # Это все же надо сделать перед запуском карт -try: #пробуем открыть конфиг с названиями карт - with open(config_dodo + "map_server") as x: #открытие конфига с названиями карт - map_server = x.read().splitlines() #Собираем по строчно все названия карт в массив -except: - map_server = [] #если ошибка или файла нет создаем пустой массив + +list_config = next(os.walk(config_dodo), (None, None, []))[2] # [] if no file +list_config.remove('.directory') "--------------------------------------" def print_line(text): "Добавление тире вокруг текста, потом сюда можно поместить и расскраску текста" @@ -42,9 +46,10 @@ def check_int(number=""): create_dir(dir_server) create_dir(config_dodo) - -def yaml_edit(map_server=map_server): - claster_id = None +@dodo.command() +def config(map_server=list_config): + cluster_dir_override = "" + cluster_id = "" count_maps = 1 count_cluster = check_int("""Укажите требуется ли кластер? 1. Да @@ -56,7 +61,9 @@ def yaml_edit(map_server=map_server): else: cluster_server = False if cluster_server == True: - claster_id = input("Укажите id для кластера, любое сочитание символов: \n") + cluster_id = input("Укажите id для кластера, любое сочетание символов: \n") + create_dir(dir_server + cluster_id) + cluster_dir_override = (dir_server + cluster_id) if map_server != []: print("Уже установленные карты: ", map_server) count_maps = check_int("Укажите количество карт: \n") @@ -107,14 +114,10 @@ def yaml_edit(map_server=map_server): password_server = input("Укажите пароль Сервера: \n") max_players = check_int("Укажите максимальное количество игроков: \n") - yaml_create(cluster_server, map_server, name_server, port_server, query_port, password_server, max_players, claster_id) - "--------------Запись в файл map_server------------------------" - with open(f"{config_dodo}/map_server", "w") as file: #создаем или открываем файл с картами - for line in map_server: # пробегаемся по массиву с картами - file.write(line + '\n') # и записываем по строчно - "--------------------------------------------------------------" + yaml_create(cluster_server, map_server, name_server, port_server, query_port, password_server, max_players, cluster_id, cluster_dir_override) -def yaml_create(cluster_server, map_server, name_server, port_server, query_port, password_server, max_players, claster_id): + +def yaml_create(cluster_server, map_server, name_server, port_server, query_port, password_server, max_players, cluster_id, cluster_dir_override): settings_dodo = [ { 'Cluster' : cluster_server, @@ -129,32 +132,39 @@ def yaml_create(cluster_server, map_server, name_server, port_server, query_port } ] - print(claster_id) - if claster_id != None: - settings_dodo[0]['clasterid'] = claster_id - with open(config_dodo + f"{map_server[-1]}.yaml", 'w') as yamlfile: + print(cluster_id) + if cluster_id != None: + settings_dodo[0]['clusterid'] = cluster_id + settings_dodo[0]['clusterdir'] = cluster_dir_override + with open(config_dodo + f"{map_server[-1]}", 'w') as yamlfile: data = yaml.dump(settings_dodo, yamlfile) print("Конфиг создан") +@dodo.command() +def start(map_server=list_config): + map_choise = [] + map_list = check_int(map_server) + + for j in str(map_list): + map_choise.append(map_server[int(j) - 1]) + map_server = map_choise -def install(map_server=map_server): for i in map_server: - cluster_server, name_server, port_server, query_port, password_server, max_players, dir_server, mods_id, listen_server = read_yaml(i) + cluster_server, name_server, port_server, query_port, password_server, max_players, dir_server, mods_id, listen_server, cluster_id, cluster_dir_ovverride = read_yaml(i) create_dir(dir_server + i) print_line("Качаем карту: " + i) os.system(f"steamcmd +force_install_dir {dir_server + i} +login anonymous +app_update 376030 +quit") print_line("Карта скачана: " + i) def start(i): os.chdir(dir_server + i + "/ShooterGame/Binaries/Linux/") - os.system(f"./ShooterGameServer {i}?SessionName={name_server}?Port={port_server}?QueryPort={query_port}?MaxPlayers={max_players}?GameModIds={mods_id}?listen={listen_server}") - print_line("server " +i + " запустился") + os.system(f"./ShooterGameServer {i}?SessionName={name_server}?Port={port_server}?QueryPort={query_port}?MaxPlayers={max_players}?GameModIds={mods_id}?listen={listen_server} -clusterid={cluster_id} -ClusterDirOverride={cluster_dir_ovverride}") threads = threading.Thread(target=start, args=(i,)) threads.start() def read_yaml(map_server): print(map_server) - with open(f"{home_dir}/.config/dodo/{map_server}.yaml", "r") as yamlfile: + with open(f"{home_dir}/.config/dodo/{map_server}", "r") as yamlfile: data = yaml.load(yamlfile, Loader=yaml.FullLoader) data = data[0] dir_server = data['ServerPath'] @@ -166,6 +176,13 @@ def read_yaml(map_server): mods_id = data['ModsId'] listen_server = data['Listen'] cluster_server = data['Cluster'] - return cluster_server, name_server, port_server, query_port, password_server, max_players, dir_server, mods_id, listen_server + cluster_id = data['clusterid'] + cluster_dir_ovverride = data['clusterdir'] + print(cluster_id) + print(cluster_dir_ovverride) + return cluster_server, name_server, port_server, query_port, password_server, max_players, dir_server, mods_id, listen_server, cluster_id, cluster_dir_ovverride #print("Запуск программы в формате: dodo.py <название функции> параметры") -fire.Fire() + + +if __name__ == '__main__': + dodo()