Atlas/Atlas-Install
1
0
Fork 0
mirror of https://github.com/Faraphel/Atlas-Install.git synced 2025-07-05 20:28:27 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

added overwrite mode for PatchFile (write only if already existing) and shortened the writing code to the write_patch function

Browse source
This commit is contained in:
Faraphel 2022-07-16 00:01:15 +02:00
parent 4852c209bb
commit de52540735
1 changed files with 19 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
source/mkw/Patch/PatchFile.py
View file

@ -1,5 +1,5 @@
from io import BytesIO from io import BytesIO
from typing import Generator from typing import Generator, IO
from source.mkw.Patch import * from source.mkw.Patch import *
from source.mkw.Patch.PatchOperation import PatchOperation from source.mkw.Patch.PatchOperation import PatchOperation
@ -58,15 +58,26 @@ class PatchFile(PatchObject):
self.patch, patch_name, patch_content self.patch, patch_name, patch_content
) )
def write_patch(destination: Path, patch_content: IO) -> None:
"""
Write a patch content to the destination. Automatically create the directory and seek to the start.
:param destination: file where the content will be written
:param patch_content: content of the file to write
"""
destination.parent.mkdir(parents=True, exist_ok=True)
with open(destination, "wb") as file:
patch_content.seek(0)
file.write(patch_content.read())
match self.configuration["mode"]: match self.configuration["mode"]:
# if the mode is copy, replace the subfile in the game by the PatchFile # if the mode is copy, replace the subfile in the game by the PatchFile
case "copy" | "edit": case "copy" | "edit":
# print(f"[{self.configuration['mode']}] copying {self} to {game_subpath}") write_patch(game_subpath.parent / patch_name, patch_content)
game_subpath.parent.mkdir(parents=True, exist_ok=True) # if the mode is overwrite, only write if the file existed before
with open(game_subpath.parent / patch_name, "wb") as file: case "overwrite":
patch_content.seek(0) if (game_subpath.parent / patch_name).exists():
file.write(patch_content.read()) write_patch(game_subpath.parent / patch_name, patch_content)
# if the mode is match, replace all the subfiles that match match_regex by the PatchFile # if the mode is match, replace all the subfiles that match match_regex by the PatchFile
case "match": case "match":
@ -74,13 +85,9 @@ class PatchFile(PatchObject):
# disallow patching files outside of the game # disallow patching files outside of the game
if not game_subfile.relative_to(extracted_game.path): if not game_subfile.relative_to(extracted_game.path):
raise PathOutsidePatch(game_subfile, extracted_game.path) raise PathOutsidePatch(game_subfile, extracted_game.path)
# patch the game with the subpatch
# print(f"[match] copying {self} to {game_subfile}")
game_subfile.parent.mkdir(parents=True, exist_ok=True) # patch the game with the subpatch
with open(game_subfile, "wb") as file: write_patch(game_subfile, patch_content)
patch_content.seek(0)
file.write(patch_content.read())
# ignore if mode is "ignore", useful if the file is used as a resource for an operation # ignore if mode is "ignore", useful if the file is used as a resource for an operation
case "ignore": pass case "ignore": pass