NinjaCheetah/libWiiPy
1
0
Fork 0
mirror of https://github.com/NinjaCheetah/libWiiPy.git synced 2025-04-25 21:01:01 -04:00
Code Issues Packages Projects Releases Wiki Activity

Allow for extracting a U8 archive to an existing (empty) directory

Browse Source
This commit is contained in:
Campbell 2024-07-05 08:34:23 +10:00
parent 1f731bbc81
commit 45638769a9
Signed by: NinjaCheetah
GPG Key ID: 670C282B3291D63D
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/libWiiPy/archive/u8.py
View File

@ -172,9 +172,14 @@ def extract_u8(u8_data, output_folder) -> None:
The path to a new folder to extract the archive to.
"""
output_folder = pathlib.Path(output_folder)
if pathlib.Path.is_dir(output_folder):
raise ValueError("Output folder already exists!")
os.mkdir(output_folder)
# Check if the path already exists, and if it does, ensure that it is both a directory and empty.
if output_folder.exists():
if output_folder.is_dir() and next(os.scandir(output_folder), None):
raise ValueError("Output folder is not empty!")
elif output_folder.is_file():
raise ValueError("A file already exists with the provided name!")
else:
os.mkdir(output_folder)
# Create a new U8Archive object and load the provided U8 file data into it.
u8_archive = U8Archive()
u8_archive.load(u8_data)