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
|
""" Indentation utilities for Cog.
"""
import re
def whitePrefix(strings):
""" Determine the whitespace prefix common to all non-blank lines
in the argument list.
"""
# 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 reindentBlock(lines, newIndent=''):
""" 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)
oldIndent = whitePrefix(lines)
outLines = []
for l in lines:
if oldIndent:
l = l.replace(oldIndent, nothing, 1)
if l and newIndent:
l = newIndent + l
outLines.append(l)
return sep.join(outLines)
def commonPrefix(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
|
