118 lines
2.8 KiB
Python
118 lines
2.8 KiB
Python
|
#!/bin/python3
|
||
|
|
|
||
|
|
import os
|
||
|
|
import pathlib
|
||
|
|
import sys
|
||
|
|
import shutil
|
||
|
|
|
||
|
|
config_file_name = "openmw.cfg"
|
||
|
|
config_file_path = os.path.join(pathlib.Path.home(), ".config", "openmw", config_file_name)
|
||
|
|
|
||
|
|
if (not os.path.exists(config_file_path)):
|
||
|
|
print("Could not find", config_file_name, "at", config_file_path)
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
def getarg(args: list, index: int, arg_type: type):
|
||
|
|
if (index >= len(args)):
|
||
|
|
print("Missing argument", index)
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
arg = args[index]
|
||
|
|
|
||
|
|
if (type(arg) != arg_type):
|
||
|
|
print("Argument", index, "must be a", str(arg_type))
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
return arg
|
||
|
|
|
||
|
|
def install_mod(args: list) -> None:
|
||
|
|
mod_data_path = getarg(args, 0, str)
|
||
|
|
|
||
|
|
if (not os.path.exists(mod_data_path)):
|
||
|
|
print("Path \"%s\" does not exist" % mod_data_path)
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
mod_data_path = os.path.abspath(mod_data_path)
|
||
|
|
|
||
|
|
with open(config_file_path, "r") as config_file:
|
||
|
|
for line in config_file:
|
||
|
|
if (mod_data_path in line):
|
||
|
|
print(mod_data_path, "is already installed!")
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
with open(config_file_path, "a") as config_file:
|
||
|
|
config_file.write("data=\"%s\"\n" % mod_data_path)
|
||
|
|
|
||
|
|
for file_name in os.listdir(mod_data_path):
|
||
|
|
if (file_name.endswith(".bsa")):
|
||
|
|
config_file.write("fallback-archive=%s\n" % file_name)
|
||
|
|
|
||
|
|
print(mod_data_path, "installed!")
|
||
|
|
|
||
|
|
def uninstall_mod(args: list) -> None:
|
||
|
|
mod_data_path = os.path.abspath(getarg(args, 0, str))
|
||
|
|
|
||
|
|
with open(config_file_path, "r") as config_file:
|
||
|
|
lines = config_file.readlines()
|
||
|
|
|
||
|
|
with open(config_file_path, "w") as config_file:
|
||
|
|
line_index = 0
|
||
|
|
|
||
|
|
while (line_index < len(lines)):
|
||
|
|
if mod_data_path in lines[line_index]:
|
||
|
|
line_index += 1
|
||
|
|
|
||
|
|
while (
|
||
|
|
(line_index < len(lines)) and
|
||
|
|
lines[line_index].startswith("fallback-archive")
|
||
|
|
):
|
||
|
|
line_index += 1
|
||
|
|
else:
|
||
|
|
config_file.write(lines[line_index])
|
||
|
|
|
||
|
|
line_index += 1
|
||
|
|
|
||
|
|
def list_mods(args: list) -> None:
|
||
|
|
with open(config_file_path, "r") as config_file:
|
||
|
|
mod_iterations = 0
|
||
|
|
|
||
|
|
for line in config_file:
|
||
|
|
if (line.startswith("data")):
|
||
|
|
split_line = line.split("=")
|
||
|
|
|
||
|
|
if (len(split_line) == 2):
|
||
|
|
mod_data_path = split_line[1].strip("\t\n ")
|
||
|
|
mod_name = os.path.split(mod_data_path)[1].strip("\"")
|
||
|
|
|
||
|
|
print("[%d]: %s (%s)" % (mod_iterations, mod_name, mod_data_path))
|
||
|
|
|
||
|
|
mod_iterations += 1
|
||
|
|
|
||
|
|
def backup_config(args: list) -> None:
|
||
|
|
backup_path = getarg(args, 0, str)
|
||
|
|
|
||
|
|
if (not os.path.exists(backup_path) and not os.path.exists(os.path.split(backup_path)[0])):
|
||
|
|
print("Config file backup path is not valid")
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
shutil.copy2(config_file_path, backup_path)
|
||
|
|
|
||
|
|
actions = {
|
||
|
|
"install": install_mod,
|
||
|
|
"uninstall": uninstall_mod,
|
||
|
|
"list": list_mods,
|
||
|
|
"backup": backup_config
|
||
|
|
}
|
||
|
|
|
||
|
|
if (len(sys.argv) < 2):
|
||
|
|
print("Expected an action")
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
action_name = sys.argv[1]
|
||
|
|
|
||
|
|
if (not action_name in actions):
|
||
|
|
print("Unknown action \"%s\". Supported actions:" % action_name, list(actions.keys()))
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
actions[action_name](sys.argv[2:])
|