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

Improve Path.__contains__ check.

Browse source
This commit is contained in:
Ethan Dalool 2019-08-01 08:46:37 -07:00
parent 0cfaf72347
commit e478aca497
1 changed files with 11 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
voussoirkit/pathclass.py
View file

@ -33,9 +33,17 @@ class Path:
self.absolute_path = path self.absolute_path = path
def __contains__(self, other): def __contains__(self, other):
if isinstance(other, Path): if isinstance(other, str):
other = other.normcase other_norm = os.path.normcase(other)
return other.startswith(self.normcase) elif isinstance(other, Path):
other_norm = other.normcase
else:
raise TypeError(other)
self_norm = self.normcase
if not self_norm.endswith(os.sep):
self_norm += os.sep
return other_norm.startswith(self_norm)
def __eq__(self, other): def __eq__(self, other):
if not hasattr(other, 'absolute_path'): if not hasattr(other, 'absolute_path'):