FilesExpand file tree

 develop

/

patch.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

41 lines (33 loc) · 1.48 KB

 develop

/

patch.py

Copy path

Top

File metadata and controls

• Code
• Blame

41 lines (33 loc) · 1.48 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

import rich

from pygit2 import Diff

from pygit2.repository import Repository

from rich.status import Status

def apply_patch(git_repo: Repository, patch: str):

"""Apply a git patch to the repository.

Args:

git_repo: The repository to apply the patch to

patch: A string containing the git patch/diff

"""

# Parse the entire patch into a Diff object

diff_patch = Diff.parse_diff(patch)

total_files = len(list(diff_patch))

error_count = 0

status = Status(f"[cyan]Applying patch to {total_files} files...[/cyan]", spinner="dots")

status.start()

# Apply each file's changes individually

for idx, patch_file in enumerate(diff_patch, 1):

try:

status.update(f"[cyan]Applying patch [{idx}/{total_files}]: {patch_file.delta.new_file.path}[/cyan]")

# Create a new diff containing just this file's changes

file_diff = Diff.parse_diff(patch_file.data)

# Apply the individual file changes

git_repo.apply(file_diff)

except Exception as e:

error_count += 1

rich.print(f"[red]✗[/red] Error applying patch to {patch_file.delta.new_file.path}: {e!s}")

status.stop()

# Display summary

if error_count == 0:

rich.print(f"[green]✓ Successfully applied patch to all {total_files} files[/green]")

else:

rich.print(f"[yellow]⚠ Applied patch with {error_count} error{'s' if error_count > 1 else ''} out of {total_files} files[/yellow]")