Запуск через аргументы, функции для директорий и проверки ввода чисел

dodo.py

@@ -2,7 +2,8 @@
 import os
 import yaml
 import threading
-import click
+#import click
+import fire
 from pathlib import Path
 
 home_dir = Path.home()
@@ -11,17 +12,32 @@ config_dodo = f"{home_dir}/.config/dodo/"
 mods_id = ""
 listen_server = True
 "------Узнаем список карт--------------"
-try:  #пробуем открыть конфиг с названиями карт, если файла нет создаем пустой массив
+# Это все же надо сделать перед запуском карт
+try:  #пробуем открыть конфиг с названиями карт
     with open(config_dodo + "map_server") as x: #открытие конфига с названиями карт
-            map_server = x.readlines() #Собираем по строчно все названия карт в массив
+            map_server = x.read().splitlines() #Собираем по строчно все названия карт в массив
 except:
-    map_server = []
+    map_server = [] #если ошибка или файла нет создаем пустой массив
 "--------------------------------------"
-
+def print_line(text):
+    "Добавление тире вокруг текста, потом сюда можно поместить и расскраску текста"
+    print("-"*20)
+    print(text)
+    print("-"*20)
 def create_dir(directory):
     "Проверка и создание директории"
     if not os.path.exists(directory):
         os.mkdir(directory)
+def check_int(number=""):
+    "Проверка на ввод числа"
+    while True:
+        try:
+            x = int(input(number))
+            return x
+        except ValueError:
+            print("Введите число")
+        
+
 
 create_dir(dir_server)
 create_dir(config_dodo)
@@ -29,21 +45,18 @@ create_dir(config_dodo)
 
 def yaml_edit(map_server):
     count_maps = 1
-    print("""Укажите требуется ли кластер?
+    count_cluster = check_int("""Укажите требуется ли кластер?
     1. Да
-    2. Нет""")
+    2. Нет
-    count_cluster = input("")
+    : """)
-    if count_cluster == "1":
+    if count_cluster == 1:
         cluster_server = True
     else:
         cluster_server = False
-    while cluster_server == True:
+    
-        count_maps = input("Укажите количество карт:\n")
+    count_maps = check_int("Укажите количество карт: \n")
-        if count_maps.isdigit:
-            break
-        print("Введите цифры")
     for i in range(int(count_maps)):
-        print("""Выберите карту из списка указав номер
+        amount_map = check_int("""Выберите карту из списка указав номер
         1. The Island
         2. The Center
         3. Scorched Earth
@@ -55,46 +68,39 @@ def yaml_edit(map_server):
         9. Crystal Isles
         10. Genesis: Part 2
         11. Lost Island
-        12. Fjordur""")
+        12. Fjordur
-        amount_map = input("")
+        : """)
-        if amount_map == "1":
+        match amount_map:
+            case 1:
                 map_server.append("TheIsland")
-        elif amount_map == "2":
+            case 2:
                 map_server.append("TheCenter")
-        elif amount_map == "3":
+            case 3:
                 map_server.append("ScorchedEarth_P")
-        elif amount_map == "4":
+            case 4:
                 map_server.append("Ragnarok")
-        elif amount_map == "5":
+            case 5:
                 map_server.append("Aberration_P")
-        elif amount_map == "6":
+            case 6:
                 map_server.append("Extinction")
-        elif amount_map == "7":
+            case 7:
                 map_server.append("Valguero_P")
-        elif amount_map == "8":
+            case 8:
                 map_server.append("Genesis")
-        elif amount_map == "9":
+            case 9:
                 map_server.append("CrystalIsles")
-        elif amount_map == "10":
+            case 10:
                 map_server.append("Gen2")
-        elif amount_map == "11":
+            case 11:
                 map_server.append("LostIsland")
-        elif amount_map == "12":
+            case 12:
                 map_server.append("Fjordur")
-        print("Укажите название Сервера")
-        name_server = input("")
 
-        print("Укажите порт Сервера <7777>")
+        name_server     = input("Укажите название Сервера: \n")
-        port_server = input("")
+        port_server     = check_int("Укажите порт Сервера <7777>: \n")
-
+        query_port      = check_int("Укажите Query-порт Сервера <27015>: \n")
-        print("Укажите Query-порт Сервера <27015>")
+        password_server = input("Укажите пароль Сервера: \n")
-        query_port = input("")
+        max_players     = check_int("Укажите максимальное количество игроков: \n")
-
-        print("Укажите пароль Сервера")
-        password_server = input("")
-
-        print("Укажите максимальное количество игроков")
-        max_players = input("")
 
         yaml_create(cluster_server, map_server, name_server, port_server, query_port, password_server, max_players)
     "--------------Запись в файл map_server------------------------"
@@ -128,21 +134,14 @@ def yaml_create(cluster_server, map_server, name_server, port_server, query_port
 def install(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)
-        if not os.path.isdir(dir_server + i):
+        create_dir(dir_server + i)
-            os.mkdir(dir_server + i)
+        print_line("Качаем карту: " + i)
-            print("-"*20)
-            print("Качаем карту " + i)
-            print("-"*20)
         os.system(f"steamcmd +force_install_dir {dir_server + i} +login anonymous +app_update 376030 +quit")
-        print("-"*20)
+        print_line("Карта скачана: " + i)
-        print("Карта скачана " + i)
-        print("-"*20)
         def start(i):
-            os.chdir(dir_server + i + "/" + "ShooterGame/Binaries/Linux/")
+            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("-"*20)
+            print_line("server " +i + " запустился")
-            print("server " +i + " Запустился")
-            print("-"*20)
         threads = threading.Thread(target=start, args=(i,))
         threads.start()
 
@@ -152,11 +151,6 @@ def read_yaml(map_server):
     with open(f"{home_dir}/.config/dodo/{map_server}.yaml", "r") as yamlfile:
         data = yaml.load(yamlfile, Loader=yaml.FullLoader)
         data = data[0]
-        print("-"*20)
-        print(data)
-        print(type(data))
-        print("-"*20)
-        #map_server = data['Map']
         dir_server = data['ServerPath']
         name_server = data['SessionName']
         port_server = data['Port']
@@ -167,15 +161,5 @@ def read_yaml(map_server):
         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
-
+#print("Запуск программы в формате: dodo.py <название функции> параметры")
-c = input("""Введите
+fire.Fire()
-1.Настроить сервера
-2.Запустить сервера""")
-if c=="1":
-    cluster_server, map_server, name_server, port_server, query_port, password_server, max_players, count_maps = yaml_edit(map_server)
-    m = input("Запустить сервер? y/N")
-    if m.lower() == "y":
-        install(map_server)
-
-elif c=="2":
-    install(map_server)