47 lines
1.0 KiB
Python
47 lines
1.0 KiB
Python
import re
|
|
from difflib import ndiff
|
|
|
|
|
|
def get_diff(original: str, modified: str) -> list[str]:
|
|
"""
|
|
Get the difference between two strings using difflib.
|
|
|
|
Parameters:
|
|
- original: The original string.
|
|
- modified: The modified string.
|
|
|
|
Returns:
|
|
A list of differences.
|
|
"""
|
|
return list(ndiff(original.split(), modified.split()))
|
|
|
|
|
|
def apply_diff(original: str, diff: list[str]) -> str:
|
|
"""
|
|
Apply the difference to the original string.
|
|
|
|
Parameters:
|
|
- original: The original string.
|
|
- diff: The difference obtained from get_diff function.
|
|
|
|
Returns:
|
|
The modified string.
|
|
"""
|
|
result = []
|
|
pattern = re.compile(r"^(\+|-) ")
|
|
|
|
for line in diff:
|
|
match = pattern.match(line)
|
|
if match:
|
|
op = match.group(1)
|
|
content = line[2:]
|
|
if op == "+":
|
|
result.append(content)
|
|
elif op == "-":
|
|
# Ignore deleted lines
|
|
pass
|
|
else:
|
|
result.append(line)
|
|
|
|
return " ".join(result)
|