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
42
|
import difflib
import os.path
__all__ = ["unified_diff"]
def unified_diff(
astr: str,
bstr: str,
filename: str,
n: int = 3,
) -> str:
"""
Returns a unified diff string for the two inputs.
Does not currently support creation or deletion where one of the filenames
is `/dev/null` or patchlevels other than `-p1`.
Does handle the "no newline at end of file" properly UNLIKE DIFFLIB.
"""
if os.path.isabs(filename):
a_filename = filename
b_filename = filename
else:
a_filename = f"a/{filename}"
b_filename = f"b/{filename}"
buf = []
gen = difflib.unified_diff(
astr.splitlines(True),
bstr.splitlines(True),
a_filename,
b_filename,
n=n,
)
for line in gen:
buf.append(line)
if not line.endswith("\n"):
# Assume this is the only case where it can happen
buf.append("\n\\ No newline at end of file\n")
return "".join(buf)
|
