Изменено получение списка установленных карт, создание и чтение конфиг файлов карт, добавлена возможность кластера (работает).

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: #открытие конфига с названиями карт
+list_config = next(os.walk(config_dodo), (None, None, []))[2]  # [] if no file
-            map_server = x.read().splitlines() #Собираем по строчно все названия карт в массив
+list_config.remove('.directory')
-except:
-    map_server = [] #если ошибка или файла нет создаем пустой массив
 "--------------------------------------"
 def print_line(text):
     "Добавление тире вокруг текста, потом сюда можно поместить и расскраску текста"
@@ -42,9 +46,10 @@ def check_int(number=""):
 create_dir(dir_server)
 create_dir(config_dodo)
 
-
+@dodo.command()
-def yaml_edit(map_server=map_server):
+def config(map_server=list_config):
-    claster_id = None
+    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)
+        yaml_create(cluster_server, map_server, name_server, port_server, query_port, password_server, max_players, cluster_id, cluster_dir_override)
-    "--------------Запись в файл map_server------------------------"
-    with open(f"{config_dodo}/map_server", "w") as file: #создаем или открываем файл с картами
-        for  line in map_server: # пробегаемся по массиву с картами
-            file.write(line + '\n') # и записываем по строчно
-    "--------------------------------------------------------------"
 
-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)
+    print(cluster_id)
-    if claster_id != None:
+    if cluster_id != None:
-        settings_dodo[0]['clasterid'] = claster_id
+        settings_dodo[0]['clusterid'] = cluster_id
-    with open(config_dodo + f"{map_server[-1]}.yaml", 'w') as yamlfile:
+        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}")
+            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}")
-            print_line("server " +i + " запустился")
         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()