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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
"""Indentation utilities for Cog."""
import re
def white_prefix(strings):
"""Find the whitespace prefix common to non-blank lines in `strings`."""
# Remove all blank lines from the list
strings = [s for s in strings if s.strip() != ""]
if not strings:
return ""
# Find initial whitespace chunk in the first line.
# This is the best prefix we can hope for.
pat = r"\s*"
if isinstance(strings[0], bytes):
pat = pat.encode("utf-8")
prefix = re.match(pat, strings[0]).group(0)
# Loop over the other strings, keeping only as much of
# the prefix as matches each string.
for s in strings:
for i in range(len(prefix)):
if prefix[i] != s[i]:
prefix = prefix[:i]
break
return prefix
def reindent_block(lines, new_indent=""):
"""Re-indent a block of text.
Take a block of text as a string or list of lines.
Remove any common whitespace indentation.
Re-indent using `newIndent`, and return it as a single string.
"""
sep, nothing = "\n", ""
if isinstance(lines, bytes):
sep, nothing = b"\n", b""
if isinstance(lines, (bytes, str)):
lines = lines.split(sep)
old_indent = white_prefix(lines)
out_lines = []
for line in lines:
if old_indent:
line = line.replace(old_indent, nothing, 1)
if line and new_indent:
line = new_indent + line
out_lines.append(line)
return sep.join(out_lines)
def common_prefix(strings):
"""Find the longest string that is a prefix of all the strings."""
if not strings:
return ""
prefix = strings[0]
for s in strings:
if len(s) < len(prefix):
prefix = prefix[: len(s)]
if not prefix:
return ""
for i in range(len(prefix)):
if prefix[i] != s[i]:
prefix = prefix[:i]
break
return prefix
|
