summaryrefslogtreecommitdiff
path: root/sys/src/cmd/python/Tools/framer/framer/util.py
blob: 73f33096459b9fef5a397eb90e1b9838345f8724 (plain)
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
def cstring(s, width=70):
    """Return C string representation of a Python string.

    width specifies the maximum width of any line of the C string.
    """
    L = []
    for l in s.split("\n"):
        if len(l) < width:
            L.append(r'"%s\n"' % l)

    return "\n".join(L)

def unindent(s, skipfirst=True):
    """Return an unindented version of a docstring.

    Removes indentation on lines following the first one, using the
    leading whitespace of the first indented line that is not blank
    to determine the indentation.
    """

    lines = s.split("\n")
    if skipfirst:
        first = lines.pop(0)
        L = [first]
    else:
        L = []
    indent = None
    for l in lines:
        ls = l.strip()
        if ls:
            indent = len(l) - len(ls)
            break
    L += [l[indent:] for l in lines]

    return "\n".join(L)