voussoir/voussoirkit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add force_sep attribute.

Browse source
This commit is contained in:
Ethan Dalool 2019-08-03 01:01:00 -07:00
parent 4f7a6c1d38
commit 452ad7f937
1 changed files with 32 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

58
voussoirkit/pathclass.py
View file

@ -19,31 +19,31 @@ class Path:
''' '''
I started to use pathlib.Path, but it was too much of a pain. I started to use pathlib.Path, but it was too much of a pain.
''' '''
def __init__(self, path): def __init__(self, path, force_sep=None):
self.force_sep = force_sep
self.sep = force_sep or os.sep
if isinstance(path, Path): if isinstance(path, Path):
self.absolute_path = path.absolute_path self.absolute_path = path.absolute_path
else: else:
path = path.strip() path = path.strip()
if re.search('[A-Za-z]:$', path): if re.search('[A-Za-z]:$', path):
# Bare Windows drive letter. # Bare Windows drive letter.
path += os.sep path += self.sep
path = normalize_sep(path) path = normalize_sep(path)
path = os.path.normpath(path) path = os.path.normpath(path)
path = os.path.abspath(path) path = os.path.abspath(path)
self.absolute_path = path self.absolute_path = path
self.absolute_path = self.absolute_path.replace('/', self.sep).replace('\\', self.sep)
def __contains__(self, other): def __contains__(self, other):
if isinstance(other, str): other = Path(other, force_sep=self.force_sep)
other_norm = os.path.normcase(other)
elif isinstance(other, Path):
other_norm = other.normcase
else:
raise TypeError(other)
self_norm = self.normcase self_norm = self.normcase
if not self_norm.endswith(os.sep): if not self_norm.endswith(self.sep):
self_norm += os.sep self_norm += self.sep
return other_norm.startswith(self_norm) return other.normcase.startswith(self_norm)
def __eq__(self, other): def __eq__(self, other):
if not hasattr(other, 'absolute_path'): if not hasattr(other, 'absolute_path'):
@ -68,7 +68,7 @@ class Path:
extension = extension.strip('.') extension = extension.strip('.')
if extension == '': if extension == '':
return self return self
return Path(self.absolute_path + '.' + extension) return self.parent.with_child(self.basename + '.' + extension)
@property @property
def basename(self): def basename(self):
@ -76,11 +76,12 @@ class Path:
def correct_case(self): def correct_case(self):
self.absolute_path = get_path_casing(self.absolute_path) self.absolute_path = get_path_casing(self.absolute_path)
self.absolute_path = self.absolute_path.replace('/', self.sep).replace('\\', self.sep)
return self.absolute_path return self.absolute_path
@property @property
def depth(self): def depth(self):
return len(self.absolute_path.rstrip(os.sep).split(os.sep)) return len(self.absolute_path.rstrip(self.sep).split(self.sep))
@property @property
def exists(self): def exists(self):
@ -105,7 +106,7 @@ class Path:
def join(self, subpath): def join(self, subpath):
if not isinstance(subpath, str): if not isinstance(subpath, str):
raise TypeError('subpath must be a string') raise TypeError('subpath must be a string')
return Path(os.path.join(self.absolute_path, subpath)) return Path(os.path.join(self.absolute_path, subpath), force_sep=self.force_sep)
def listdir(self): def listdir(self):
children = os.listdir(self.absolute_path) children = os.listdir(self.absolute_path)
@ -114,12 +115,14 @@ class Path:
@property @property
def normcase(self): def normcase(self):
return os.path.normcase(self.absolute_path) norm = os.path.normcase(self.absolute_path)
norm = norm.replace('/', self.sep).replace('\\', self.sep)
return norm
@property @property
def parent(self): def parent(self):
parent = os.path.dirname(self.absolute_path) parent = os.path.dirname(self.absolute_path)
parent = self.__class__(parent) parent = self.__class__(parent, force_sep=self.force_sep)
return parent return parent
@property @property
@ -127,7 +130,7 @@ class Path:
return self.relative_to(os.getcwd()) return self.relative_to(os.getcwd())
def relative_to(self, other, simple=False): def relative_to(self, other, simple=False):
other = Path(other) other = Path(other, force_sep=self.force_sep)
if self == other: if self == other:
return '.' return '.'
@ -138,9 +141,9 @@ class Path:
if self in other: if self in other:
relative = self.absolute_path relative = self.absolute_path
relative = relative.replace(other.absolute_path, '', 1) relative = relative.replace(other.absolute_path, '', 1)
relative = relative.lstrip(os.sep) relative = relative.lstrip(self.sep)
if not simple: if not simple:
relative = '.' + os.sep + relative relative = '.' + self.sep + relative
return relative return relative
common = common_path([other.absolute_path, self.absolute_path], fallback=None) common = common_path([other.absolute_path, self.absolute_path], fallback=None)
@ -148,22 +151,25 @@ class Path:
if common is None: if common is None:
return self.absolute_path return self.absolute_path
common = Path(common, force_sep=self.force_sep)
backsteps = other.depth - common.depth backsteps = other.depth - common.depth
backsteps = os.sep.join('..' for x in range(backsteps)) backsteps = self.sep.join('..' for x in range(backsteps))
common = common.absolute_path common = common.absolute_path
if not common.endswith(os.sep): if not common.endswith(self.sep):
common += os.sep common += self.sep
unique = self.absolute_path.replace(common, '', 1) unique = self.absolute_path.replace(common, '', 1)
return os.path.join(backsteps, unique) relative_path = os.path.join(backsteps, unique)
relative_path = relative_path.replace('/', self.sep).replace('\\', self.sep)
return relative_path
def replace_extension(self, extension): def replace_extension(self, extension):
extension = extension.rsplit('.', 1)[-1] extension = extension.rsplit('.', 1)[-1]
base = os.path.splitext(self.absolute_path)[0] base = os.path.splitext(self.basename)[0]
if extension == '': if extension == '':
return Path(base) return self.parent.with_child(base)
return Path(base + '.' + extension) return self.parent.with_child(base + '.' + extension)
@property @property
def size(self): def size(self):