Toast/Leek
1
0
Fork 0
Code Issues 5 Pull requests 2 Projects 3 Releases 3 Activity

ModInstaller: add install method

Browse source
This commit is contained in:
Toast 2025-06-03 00:33:25 +02:00 • committed by Toast
parent 6290a18c52
commit 56b63fe673
1 changed files with 25 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

26
src/leek/mod_installer.py
View file

@ -1,12 +1,18 @@
from pathlib import Path from pathlib import Path
from zipfile import Path as ZipPath from zipfile import Path as ZipPath
from zipfile import ZipFile, is_zipfile from zipfile import ZipFile, is_zipfile
from leek.utils import GameFinder
class ModInstaller: class ModInstaller:
__archive_path: Path
__archive_format: str __archive_format: str
__is_flat: bool
__is_installed: bool
def __init__(self, mod_path: Path) -> None: def __init__(self, mod_path: Path) -> None:
self.__archive_path = mod_path
self.__is_installed = False
if is_zipfile(mod_path): if is_zipfile(mod_path):
self.__archive_format = "zip" self.__archive_format = "zip"
archive: ZipFile = ZipFile(mod_path) archive: ZipFile = ZipFile(mod_path)
@ -22,11 +28,13 @@ class ModInstaller:
config_toml = things_in_root[0] / "config.toml" config_toml = things_in_root[0] / "config.toml"
if config_toml.exists() and config_toml.is_file(): if config_toml.exists() and config_toml.is_file():
# Mod can be extracted as is # Mod can be extracted as is
self.__is_flat = False
return return
case _: case _:
config_toml = path / "config.toml" config_toml = path / "config.toml"
if config_toml.exists() and config_toml.is_file(): if config_toml.exists() and config_toml.is_file():
# Mod needs to be extracted in a folder # Mod needs to be extracted in a folder
self.__is_flat = True
return return
# If we ever get here there's either no mod on the archive, or we failed to find it # If we ever get here there's either no mod on the archive, or we failed to find it
@ -35,11 +43,27 @@ class ModInstaller:
else: else:
raise UnsupportedArchiveTypeError(str(mod_path)) raise UnsupportedArchiveTypeError(str(mod_path))
@property
def is_installed(self) -> bool:
return self.__is_installed
def install(self) -> None: def install(self) -> None:
""" """
Install the mod to Project Diva's mod folder Install the mod to Project Diva's mod folder
""" """
raise NotImplementedError if self.__is_installed:
return
game_path: Path = GameFinder.find()
if self.__is_flat:
mod_folder_name: str = self.__archive_path.stem
ZipFile(self.__archive_path).extractall(
path=str(game_path / "mods" / mod_folder_name)
)
else:
ZipFile(self.__archive_path).extractall(path=str(game_path / "mods"))
self.__is_installed = True
class UnsupportedArchiveTypeError(Exception): class UnsupportedArchiveTypeError(Exception):