4cf5bc0c63
- Change the arg parsing to a variant that works on my Windows PC - Remove the /bin/python3 comment at the start as it was leading to errors on my side - Check if the OS is a Windows and in that case define the default location of the config file - Add an argument to define the location of the configfile in case not the default location is used. This was used mainly to test the functions
137 lines
3.9 KiB
Python
Executable File
137 lines
3.9 KiB
Python
Executable File
import argparse
|
|
import os
|
|
import pathlib
|
|
import shutil
|
|
from platform import system
|
|
|
|
|
|
def install_mod(path, config_path) -> None:
|
|
mod_data_path = path
|
|
|
|
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_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(path, config_path) -> None:
|
|
mod_data_path = path
|
|
|
|
with open(config_path, "r") as config_file:
|
|
lines = config_file.readlines()
|
|
|
|
with open(config_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(config_path) -> None:
|
|
with open(config_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(path, config_path) -> None:
|
|
backup_path = path
|
|
|
|
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_path, backup_path)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument("--install", type=str, default=None, action="store", required=False,
|
|
help="Provide this parameter with the path of a single mod folder that should be added to the "
|
|
"config file")
|
|
parser.add_argument("--uninstall", type=str, default=None, action="store", required=False,
|
|
help="Provide this parameter with path of a single mod folder that should be removed from the "
|
|
"config file")
|
|
parser.add_argument("--list", default=None, action="store_true", required=False,
|
|
help="Commands the script to list the current mod folders from the config file.")
|
|
parser.add_argument("--backup", type=str, default=None, action="store", required=False,
|
|
help="Provide this parameter with the path where the backup file should be placed.")
|
|
parser.add_argument("--configfile", type=str, default=None, action="store", required=False,
|
|
help="In case the config file is not in the default location, provide this parameter with the "
|
|
"path including(!) the openmw.cfg file.")
|
|
|
|
p = parser.parse_args()
|
|
|
|
if (p.configfile):
|
|
config_file_path = p.configfile
|
|
|
|
else:
|
|
|
|
config_file_name = "openmw.cfg"
|
|
plt = system()
|
|
|
|
if (plt == "Windows"):
|
|
config_file_path = os.path.join(pathlib.Path.home(), "Documents", "my games", "openmw", config_file_name)
|
|
else:
|
|
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)
|
|
|
|
if (p.install):
|
|
install_path = p.install
|
|
install_mod(install_path, config_file_path)
|
|
|
|
elif (p.uninstall):
|
|
uninstall_path = p.uninstall
|
|
uninstall_mod(uninstall_path, config_file_path)
|
|
|
|
elif (p.list):
|
|
list_mods(config_file_path)
|
|
|
|
elif (p.backup):
|
|
backup_path = p.backup
|
|
backup_config(backup_path, config_file_path)
|
|
|
|
else:
|
|
print("No valid parameter provided. You can use -h in order to see the valid possibilities.")
|