diff --git a/src/dir/directory.py b/src/dir/directory.py index fbb9380..8e0f37e 100644 --- a/src/dir/directory.py +++ b/src/dir/directory.py @@ -14,6 +14,22 @@ class Directory(ABC): def __iter__(self): return self.contents.__iter__() + def __getitem__(self, name): + for e in self: + if e.name == name: + return e + raise KeyError + + def prune(self, name): + for e in self: + if e.name == name: + self.contents.remove(e) + return + raise KeyError + + def __str__(self): + return self.name + @property def name(self): return self.path.name