patch-ng

import os import tempfile import shutil from patch_ng import from_string, PatchSet # Create a temporary directory to work in temp_dir = tempfile.mkdtemp() original_file_path = os.path.join(temp_dir, "my_file.txt") try: # 1. Create an original file with open(original_file_path, "w") as f: f.write("Line 1: Hello world\n") f.write("Line 2: This is a test.\n") f.write("Line 3: End of file.\n") print(f"Original file '{os.path.basename(original_file_path)}' created in {temp_dir}:\n" \ f"{open(original_file_path, 'r').read()}") # 2. Define a unified diff patch string # This patch changes 'Line 2' and adds 'New Line 2.5' patch_content = """--- a/my_file.txt +++ b/my_file.txt @@ -1,3 +1,4 @@ Line 1: Hello world -Line 2: This is a test. +Line 2: This line was changed. +New Line 2.5: Inserted line. Line 3: End of file. """ # 3. Parse the patch string # The `strip` argument removes leading path components (e.g., 'a/' or 'b/') patchset = from_string(patch_content) if not patchset: print("Failed to parse patch content.") else: # 4. Apply the patch to the specified root directory # The `root` argument is crucial to apply the patch outside the current working directory. success = patchset.apply(strip=1, root=temp_dir) if success: print(f"Patch applied successfully to files in {temp_dir}.") print(f"\nPatched file content of '{os.path.basename(original_file_path)}':\n" \ f"{open(original_file_path, 'r').read()}") else: print("Failed to apply patch.") finally: # Clean up the temporary directory shutil.rmtree(temp_dir) print(f"\nCleaned up temporary directory: {temp_dir}")