lifebottle/PythonLib
0
0
Fork 0
mirror of https://github.com/lifebottle/PythonLib.git synced 2026-02-13 15:25:50 -08:00
Code Issues Packages Projects Releases Wiki Activity

Update ToolsTOR.py

Browse Source
This commit is contained in:
Mc-muffin
2023-07-27 22:00:32 -05:00
parent d13d10f259
commit c71d7d7663
1 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
pythonlib/games/ToolsTOR.py
View File

@@ -536,10 +536,7 @@ class ToolsTOR(ToolsTales):
def extract_main_archive(self) -> None:
dat_bin_path = self.paths["extracted_files"] / "DAT"
if dat_bin_path.exists():
print("Cleaning extract folder...")
shutil.rmtree(dat_bin_path)
dat_bin_path.mkdir(exist_ok=True)
self.clean_folder(dat_bin_path)
print("Extracting DAT.BIN files...")
with open( self.dat_bin_original, "rb") as f:
@@ -779,7 +776,7 @@ class ToolsTOR(ToolsTales):
iso.open(str(umd_iso))
extract_to = self.paths["original_files"]
shutil.rmtree(extract_to)
self.clean_folder(extract_to)
files = []
for dirname, _, filelist in iso.walk(iso_path="/"):
@@ -795,4 +792,15 @@ class ToolsTOR(ToolsTales):
output.write(data)
pbar.update(len(data))
iso.close()
iso.close()
def clean_folder(self, path: Path) -> None:
target_files = list(path.iterdir())
if len(target_files) != 0:
print("Cleaning folder...")
for file in target_files:
if file.is_dir():
shutil.rmtree(file)
elif file.name.lower() != ".gitignore":
file.unlink(missing_ok=False)