129 lines
3.7 KiB
Python
129 lines
3.7 KiB
Python
import os
|
|
import pwd
|
|
import re
|
|
import regex
|
|
|
|
cbuf_re = re.compile(r'[\[\]\\]')
|
|
def cbuf_escape(s):
|
|
return cbuf_re.sub(lambda m: '\\' + m.group(0), s)
|
|
|
|
def flatzip(a, b):
|
|
l = []
|
|
for x, y in zip(a, b):
|
|
l.append(x)
|
|
l.append(y)
|
|
return l
|
|
|
|
def should_ignore_path(path, suffixes):
|
|
for name in path.split('/'):
|
|
for suffix in suffixes:
|
|
if name.endswith(suffix):
|
|
return True
|
|
return False
|
|
|
|
def normal_path(path):
|
|
#path = os.path.realpath(path)
|
|
path = os.path.normpath(path)
|
|
home = os.getenv('HOME')
|
|
isdir = os.path.isdir(path)
|
|
if path.startswith(home):
|
|
path = path.replace(home, '~', 1)
|
|
if isdir and not path.endswith('/'):
|
|
return path + '/'
|
|
else:
|
|
return path
|
|
|
|
def expand_tilde(path):
|
|
isd = path.endswith('/')
|
|
if not path.startswith('~'):
|
|
return path
|
|
parts = path.split('/', 1)
|
|
if parts[0] == '~':
|
|
parts[0] = os.getenv('HOME')
|
|
elif parts[0].startswith('~'):
|
|
users = [x[0] for x in pwd.getpwall()]
|
|
if parts[0][1:] in users:
|
|
home = pwd.getpwnam(parts[0][1:])[5]
|
|
parts[0] = home
|
|
if len(parts) > 1:
|
|
s = os.path.join(parts[0], parts[1])
|
|
else:
|
|
s = parts[0]
|
|
s = os.path.realpath(s)
|
|
if os.path.isdir(s) and isd:
|
|
s += '/'
|
|
return s
|
|
|
|
def cleanse(s):
|
|
s2 = s.replace("\n", "")
|
|
return s2
|
|
|
|
def padtrunc(s, i, c=' '):
|
|
return s.ljust(i, c)[:i]
|
|
def pad(s, i, c=' '):
|
|
return s.ljust(i, c)
|
|
|
|
def count_leading_whitespace(s):
|
|
m = regex.leading_whitespace.match(s)
|
|
assert m, "count leading whitespace failed somehow"
|
|
return m.end() - m.start()
|
|
|
|
def dump(x):
|
|
d = {}
|
|
for name in dir(x):
|
|
d[name] = getattr(x, name)
|
|
return '%s: %r' % (x, d)
|
|
|
|
def get_margin_limit(w, def_limit=80):
|
|
lname = '%s.margin' % w.mode.name.lower()
|
|
if lname in w.application.config:
|
|
return w.application.config[lname]
|
|
else:
|
|
return w.application.config.get('margin', def_limit)
|
|
|
|
def get_margin_color(w, def_color='blue'):
|
|
lname = '%s.margin_color' % w.mode.name.lower()
|
|
if lname in w.application.config:
|
|
return w.application.config[lname]
|
|
else:
|
|
return w.application.config.get('margin_color', def_color)
|
|
|
|
# emulate defaultdict for 2.4
|
|
try:
|
|
from collections import defaultdict
|
|
except:
|
|
class defaultdict(dict):
|
|
def __init__(self, default_factory=None, *a, **kw):
|
|
if (default_factory is not None and
|
|
not hasattr(default_factory, '__call__')):
|
|
raise TypeError('first argument must be callable')
|
|
dict.__init__(self, *a, **kw)
|
|
self.default_factory = default_factory
|
|
def __getitem__(self, key):
|
|
try:
|
|
return dict.__getitem__(self, key)
|
|
except KeyError:
|
|
return self.__missing__(key)
|
|
def __missing__(self, key):
|
|
if self.default_factory is None:
|
|
raise KeyError(key)
|
|
self[key] = value = self.default_factory()
|
|
return value
|
|
def __reduce__(self):
|
|
if self.default_factory is None:
|
|
args = tuple()
|
|
else:
|
|
args = self.default_factory,
|
|
return type(self), args, None, None, self.items()
|
|
def copy(self):
|
|
return self.__copy__()
|
|
def __copy__(self):
|
|
return type(self)(self.default_factory, self)
|
|
def __deepcopy__(self, memo):
|
|
import copy
|
|
return type(self)(self.default_factory,
|
|
copy.deepcopy(self.items()))
|
|
def __repr__(self):
|
|
return 'defaultdict(%s, %s)' % (self.default_factory,
|
|
dict.__repr__(self))
|