Project

General

Profile

1
# String manipulation
2

    
3
import codecs
4
import re
5

    
6
import util
7

    
8
##### Parsing
9

    
10
def split(sep, str_):
11
    '''Returns [] if str_ == ""'''
12
    if str_ == '': return []
13
    else: return str_.split(sep)
14

    
15
def remove_prefix(prefix, str_):
16
    if str_.startswith(prefix): return str_[len(prefix):]
17
    else: return str_
18

    
19
def remove_prefixes(prefixes, str_):
20
    for prefix in prefixes: str_ = remove_prefix(prefix, str_)
21
    return str_
22

    
23
def with_prefixes(prefixes, str_): return (p+str_ for p in prefixes)
24

    
25
def remove_suffix(suffix, str_):
26
    if str_.endswith(suffix): return str_[:-len(suffix)]
27
    else: return str_
28

    
29
def contains_any(haystack, needles):
30
    for needle in needles:
31
        if haystack.find(needle) >= 0: return True
32
    return False
33

    
34
def overlaps(str0, str1): return str0.find(str1) >= 0 or str1.find(str0) >= 0
35

    
36
##### Unicode
37

    
38
unicode_reader = codecs.getreader('utf_8')
39

    
40
def to_unicode(str_):
41
    if isinstance(str_, unicode): return str_
42
    encodings = ['utf_8', 'latin_1']
43
    for encoding in encodings:
44
        try: return unicode(str_, encoding)
45
        except UnicodeDecodeError, e: pass
46
    raise AssertionError(encoding+' is not a catch-all encoding')
47

    
48
def ustr(val):
49
    '''Like built-in str() but converts to unicode object'''
50
    if not util.is_str(val): val = str(val)
51
    return to_unicode(val)
52

    
53
##### Line endings
54

    
55
def extract_line_ending(line):
56
    '''@return tuple (contents, ending)'''
57
    contents = remove_suffix('\r', remove_suffix('\n', line))
58
    return (contents, line[len(contents):])
59

    
60
def remove_line_ending(line): return extract_line_ending(line)[0]
61

    
62
def ensure_newl(str_): return remove_line_ending(str_)+'\n'
63

    
64
def is_multiline(str_):
65
    newl_idx = str_.find('\n')
66
    return newl_idx >= 0 and newl_idx != len(str_)-1 # has newline before end
67

    
68
def remove_extra_newl(str_):
69
    if is_multiline(str_): return str_
70
    else: return str_.rstrip()
71

    
72
def std_newl(str_): return str_.replace('\r\n', '\n').replace('\r', '\n')
73

    
74
##### Whitespace
75

    
76
def cleanup(str_): return std_newl(str_.strip())
77

    
78
def single_space(str_): return re.sub(r' {2,}', r' ', str_.strip())
79

    
80
def one_line(str_): return re.sub(r'\n *', r' ', cleanup(str_))
81

    
82
##### Control characters
83

    
84
def is_ctrl(char):
85
    '''Whether char is a (non-printable) control character'''
86
    return ord(char) < 32 and not char.isspace()
87

    
88
def strip_ctrl(str_):
89
    '''Strips (non-printable) control characters'''
90
    return ''.join(filter(lambda c: not is_ctrl(c), str_))
(25-25/34)