mirror of
https://github.com/All-Hands-AI/OpenHands.git
synced 2026-01-13 16:58:07 -05:00
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
import difflib
|
|
|
|
import whatthepatch
|
|
|
|
|
|
def get_diff(old_contents: str, new_contents: str, filepath: str = 'file') -> str:
|
|
diff = list(
|
|
difflib.unified_diff(
|
|
old_contents.split('\n'),
|
|
new_contents.split('\n'),
|
|
fromfile=filepath,
|
|
tofile=filepath,
|
|
# do not output unchange lines
|
|
# because they can cause `parse_diff` to fail
|
|
n=0,
|
|
)
|
|
)
|
|
return '\n'.join(map(lambda x: x.rstrip(), diff))
|
|
|
|
|
|
def parse_diff(diff_patch: str) -> list[whatthepatch.patch.Change]:
|
|
# handle empty patch
|
|
if diff_patch.strip() == '':
|
|
return []
|
|
|
|
patch = whatthepatch.parse_patch(diff_patch)
|
|
patch_list = list(patch)
|
|
assert len(patch_list) == 1, (
|
|
'parse_diff only supports single file diff. But got:\nPATCH:\n'
|
|
+ diff_patch
|
|
+ '\nPATCH LIST:\n'
|
|
+ str(patch_list)
|
|
)
|
|
changes = patch_list[0].changes
|
|
|
|
# ignore changes that are the same (i.e., old_lineno == new_lineno)
|
|
output_changes = []
|
|
for change in changes:
|
|
if change.old != change.new:
|
|
output_changes.append(change)
|
|
return output_changes
|