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

This commit is contained in:
Sitislllll@gmail.com 2023-02-10 08:58:42 +03:00
parent 18fa2d6e28
commit bfa992a8d5

126
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. Нет""")
count_cluster = input("")
if count_cluster == "1":
2. Нет
: """)
if count_cluster == 1:
cluster_server = True
else:
cluster_server = False
while cluster_server == True:
count_maps = input("Укажите количество карт:\n")
if count_maps.isdigit:
break
print("Введите цифры")
count_maps = check_int("Укажите количество карт: \n")
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""")
amount_map = input("")
if amount_map == "1":
12. Fjordur
: """)
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>")
port_server = input("")
print("Укажите Query-порт Сервера <27015>")
query_port = input("")
print("Укажите пароль Сервера")
password_server = input("")
print("Укажите максимальное количество игроков")
max_players = input("")
name_server = input("Укажите название Сервера: \n")
port_server = check_int("Укажите порт Сервера <7777>: \n")
query_port = check_int("Укажите Query-порт Сервера <27015>: \n")
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)
"--------------Запись в файл 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):
os.mkdir(dir_server + i)
print("-"*20)
print("Качаем карту " + i)
print("-"*20)
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("-"*20)
print("Карта скачана " + i)
print("-"*20)
print_line("Карта скачана: " + i)
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("server " +i + " Запустился")
print("-"*20)
print_line("server " +i + " запустился")
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
c = input("""Введите
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)
#print("Запуск программы в формате: dodo.py <название функции> параметры")
fire.Fire()