Изменено получение списка установленных карт, создание и чтение конфиг файлов карт, добавлена возможность кластера (работает).
This commit is contained in:
parent
f6a82c466d
commit
0f3f9903f4
75
dodo.py
75
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()
|
||||
|
||||
Loading…
Reference in New Issue
Block a user