.

hlna.py

@@ -1,98 +1,47 @@
-#!/usr/bin/env python3
-"Модуль для создания и управления серверами Арк"
-import os
-import traceback
 
+#!/usr/bin/env python3
+import os
 import yaml
 import threading
 from pathlib import Path
-
 import click
 import colorama
-import datetime
-
 home_dir = Path.home()
 dir_server = f"{home_dir}/ARK_Servers/"
 config_hlna = f"{home_dir}/.config/hlna/"
 mods_id = ""
 listen_server = True
-
-
 @click.group()
 def hlna():
     pass
-
-
-
 list_config = next(os.walk(config_hlna), (None, None, []))[2]  # [] if no file
 x = list_config.count('.directory')
-if x > 0:
+if x>0:
     list_config.remove('.directory')
-
-
-def logging(l):
-    "Записываем логи в файл"
-    with open(dir_server + "logs", "a") as filelog:
-        now1 = datetime.datetime.today()
-        now1 = now1.strftime("%d-%m %H:%M | ")
-        filelog.write(now1 + l)
-
-
 def print_line(text):
     "Добавление тире вокруг текста, покраска"
     print(colorama.Fore.YELLOW + "-"*30)
     print(colorama.Fore.GREEN + text)
     print(colorama.Fore.YELLOW + "-"*30 + colorama.Style.RESET_ALL)
-
-
 def create_dir(directory):
     "Проверка и создание директории"
-    try:
+    if not os.path.exists(directory):
-        if not os.path.exists(directory):
+        os.mkdir(directory)
-            os.mkdir(directory)
-    except:
-        logging(traceback.format_exc())
-        logging(f"Не удалось создать директорию - {directory}")
-
-
 def check_int(number=""):
     "Проверка на ввод числа"
-    x=""
     while True:
         try:
             x = input(number)
-            if bool(x) == True:
+            if x == "":
-                x = x.replace(' ', '')
-                x = x.replace(',', '')
-                x = int(x)
-                return x
-            else:
                 return 0
+            x = x.replace(' ', '')
+            x = x.replace(',', '')
+            x = int(x)
+            return x
         except ValueError:
-            logging(traceback.format_exc())
-            logging(f"Не прошла проверка на ввод числа, func(check_int) ввод пользователя {x}")
             print("Введите число")
-
-
 create_dir(dir_server)
 create_dir(config_hlna)
-
-
-<<<<<<< HEAD:dodo.py
-@dodo.command(help='Установка модов, укажите через запятую id модов')
-@click.option('--id', default="", help='id модификаций из мастерской steam')
-def modinstall(id, map_server=list_config):
-    print(id)
-    print(type(id))
-    print("Выберите карты, которым добавить моды")
-    map_server=choose_map(map_server)
-    print("map_server=",map_server)
-    print("Перезагрузите сервера для установки модов")
-
-
-@dodo.command(help='Для конфигурирования параметров запускаемого сервера или кластера серверов')
-def config(map_server=list_config):
-=======
 @hlna.command(help='Для конфигурирования параметров запускаемого сервера или кластера серверов')
 def config(map_server=list_config):
     query_p = []
@@ -103,8 +52,6 @@ def config(map_server=list_config):
             data = data[0]
             port_s.append(data['Port'])
             query_p.append(data['QueryPort'])
-
->>>>>>> master:hlna.py
     cluster_dir_override = ""
     cluster_id = ""
     count_cluster = check_int("""Укажите требуется ли кластер?
@@ -120,12 +67,10 @@ def config(map_server=list_config):
         cluster_dir_override = (dir_server + cluster_id)
     else:
         cluster_server = False
-
     if map_server != []:
         print("Уже установленные карты: ")
         for i in map_server:
             print(i)
-
     count_maps = check_int("Укажите количество карт: \n")
     if count_maps == 0: # 0 возрвращает check_int когда, ничего не было введено
         count_maps = 1
@@ -176,28 +121,15 @@ def config(map_server=list_config):
                 map_server.append("LostIsland")
             case 12:
                 map_server.append("Fjordur")
-
         name_server = input("Укажите название Сервера: \n")
         if name_server == "":
             name_server = map_server[-1]
-        query_p = []
-        port_s = []
-        for i in map_server:
-            try:
-                with open(f"{home_dir}/.config/dodo/{i}", "r") as yamlfile:
-                    data = yaml.load(yamlfile, Loader=yaml.FullLoader)
-                    data = data[0]
-                    port_s.append(data['Port'])
-                    query_p.append(data['QueryPort'])
-            except:
-                pass
         while True:
             port_server = check_int("Укажите порт Сервера <7777>: \n")
             if port_server == 0:
                 if port_s == []:
                     port_server = 7777
                 else:
-                    print()
                     port_server = max(port_s) + 2
             if port_server is port_s: # если веденный порт есть в списке портов
                 print("Этот порт уже занят")
@@ -218,12 +150,9 @@ def config(map_server=list_config):
                 break
         password_server = input("Укажите пароль Сервера: \n")
         max_players     = check_int("Укажите максимальное количество игроков: \n")
-        if max_players == 0:
+        if max_players == "":
             max_players = 70
-
         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, cluster_id, cluster_dir_override):
     settings_hlna = [
         {
@@ -238,24 +167,16 @@ def yaml_create(cluster_server, map_server, name_server, port_server, query_port
             'ServerPath'     : dir_server
         }
     ]
-
+    print(cluster_id)
     if cluster_id != None:
         settings_hlna[0]['clusterid'] = cluster_id
         settings_hlna[0]['clusterdir'] = cluster_dir_override
     with open(config_hlna + f"{map_server[-1]}", 'w') as yamlfile:
         data = yaml.dump(settings_hlna, yamlfile)
         print(colorama.Fore.GREEN + "Конфиг создан" + colorama.Style.RESET_ALL)
-
-
-<<<<<<< HEAD:dodo.py
-@dodo.command()
-def servers(map_servers=list_config):
-    if map_servers==[]:
-=======
 @hlna.command()
 def servers(map_server=list_config):
     if map_server==[]:
->>>>>>> master:hlna.py
         print("Сервера не установлены")
     else:
         for i in map_server:
@@ -271,79 +192,41 @@ def servers(map_server=list_config):
             Моды: {mods_id}
             Кластер id: {cluster_id}""")
         print("\n")
-
-
 @hlna.command(help='Для запуска, сконфигурированного сервера или кластера')
 def start(map_server=list_config):
-    if map_server == []:
-        return print("Нет карт для запуска")
-    print("Выберите карты для запуска")
-    map_server = choose_map(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, 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} -clusterid={cluster_id} -ClusterDirOverride={cluster_dir_ovverride}")
-        if os.path.exists(dir_server + i + "/ShooterGame/Binaries/Linux/"):
-            threads = threading.Thread(target=start, args=(i,))
-            threads.start()
-        else:
-            print("Сервер не установлен, проверьте, достаточно ли места на диске для установки карты")
-
-
-def choose_map(map_server):
     map_choise = []
     while True:
         "Проверка на выбор имеющихся карт"
         check = True
         for i in range(len(map_server)):
-            print(f"{i + 1}) {map_server[i]}")
+            print(f"{i+1}) {map_server[i]}")
-
+        map_list = input("Выбери карты для запуска в формате 1 2 3 или 1,2,3")
-        map_list = input("Выбери карты в формате 1 2 3 или 1,2,3\n")
         map_list = map_list.replace(',', ' ')
         map_list = list(set(map_list.split()))
         try:
             for s in map_list:
-                if int(s) > len(map_server):
+                    if int(s) > len(map_server):
-                    check = False
+                        check = False
             if len(map_list) <= len(map_server) and check == True:
                 break
         except:
-            logging(traceback.format_exc())
-            logging(f"Ошибка при выборе карт для запуска, пользователь ввел - {map_list}")
             print("Неправильный ввод")
-<<<<<<< HEAD:dodo.py
-    for j in str(map_list):
-        map_choise.append(map_server[int(j) - 1])
-    map_server = map_choise
-    return map_server
-=======
         print_line(map_list)
     for j in map_list:
         print_line(j)
         map_choise.append(map_server[int(j) - 1])
     map_server = map_choise
-
     for i in map_server:
         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} -clusterid={cluster_id} -ClusterDirOverride={cluster_dir_ovverride}")
         threads = threading.Thread(target=start, args=(i,))
         threads.start()
->>>>>>> master:hlna.py
-
-
 def read_yaml(map_server):
     with open(f"{home_dir}/.config/hlna/{map_server}", "r") as yamlfile:
         data = yaml.load(yamlfile, Loader=yaml.FullLoader)
@@ -359,8 +242,8 @@ def read_yaml(map_server):
         cluster_server = data['Cluster']
         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
-
-
 if __name__ == '__main__':
     hlna()