forked from Imagelibrary/littlefs
scripts: Removed item/iter magic methods from fs object classes
So no more __getitem__, __contains__, or __iter__ for Rbyd, Btree, Mdir, Mtree, Lfs.File, etc. These were way too error-prone, especially when accidental unpacking triggered unintended disk traversal and weird error states. We didn't even use the implicit behavior because we preferred the full name for heavy disk operations. The motivation for this was Python not catching this bug, which is a bit silly: rid, rattr, *path_ = rbyd
This commit is contained in:
@@ -857,18 +857,6 @@ class Rbyd:
|
||||
else:
|
||||
return rattr_
|
||||
|
||||
def __getitem__(self, key):
|
||||
if not isinstance(key, tuple):
|
||||
key = (key,)
|
||||
|
||||
return self.lookup(*key)
|
||||
|
||||
def __contains__(self, key):
|
||||
if not isinstance(key, tuple):
|
||||
key = (key,)
|
||||
|
||||
return self.lookup(*key) is not None
|
||||
|
||||
def rids(self, *,
|
||||
path=False):
|
||||
rid = -1
|
||||
@@ -935,9 +923,6 @@ class Rbyd:
|
||||
yield rattr_
|
||||
tag_ = rattr_.tag
|
||||
|
||||
def __iter__(self):
|
||||
return self.rattrs()
|
||||
|
||||
# lookup by name
|
||||
def namelookup(self, did, name):
|
||||
# binary search
|
||||
@@ -1142,18 +1127,6 @@ class Btree:
|
||||
else:
|
||||
return rattr_
|
||||
|
||||
def __getitem__(self, key):
|
||||
if not isinstance(key, tuple):
|
||||
key = (key,)
|
||||
|
||||
return self.lookup(*key)
|
||||
|
||||
def __contains__(self, key):
|
||||
if not isinstance(key, tuple):
|
||||
key = (key,)
|
||||
|
||||
return self.lookup(*key) is not None
|
||||
|
||||
# note leaves only iterates over leaf rbyds, whereas traverse
|
||||
# traverses all rbyds
|
||||
def leaves(self, *,
|
||||
@@ -1284,9 +1257,6 @@ class Btree:
|
||||
else:
|
||||
yield rattr
|
||||
|
||||
def __iter__(self):
|
||||
return self.rattrs()
|
||||
|
||||
# lookup by name
|
||||
def namelookupleaf(self, did, name, *,
|
||||
path=False,
|
||||
@@ -1538,18 +1508,6 @@ class Mdir:
|
||||
return self.rbyd.lookup(mid.mrid, tag, mask,
|
||||
path=path)
|
||||
|
||||
def __getitem__(self, key):
|
||||
if not isinstance(key, tuple):
|
||||
key = (key,)
|
||||
|
||||
return self.lookup(*key)
|
||||
|
||||
def __contains__(self, key):
|
||||
if not isinstance(key, tuple):
|
||||
key = (key,)
|
||||
|
||||
return self.lookup(*key) is not None
|
||||
|
||||
def mids(self, *,
|
||||
path=False):
|
||||
for r in self.rbyd.rids(
|
||||
@@ -1586,9 +1544,6 @@ class Mdir:
|
||||
yield from self.rbyd.rattrs(mid.mrid, tag, mask,
|
||||
path=path)
|
||||
|
||||
def __iter__(self):
|
||||
return self.rattrs()
|
||||
|
||||
# lookup by name
|
||||
def namelookup(self, did, name):
|
||||
# unlike rbyd namelookup, we need an exact match here
|
||||
@@ -1886,18 +1841,6 @@ class Mtree:
|
||||
else:
|
||||
return mdir, name
|
||||
|
||||
def __getitem__(self, key):
|
||||
if not isinstance(key, tuple):
|
||||
key = (key,)
|
||||
|
||||
return self.lookup(*key)
|
||||
|
||||
def __contains__(self, key):
|
||||
if not isinstance(key, tuple):
|
||||
key = (key,)
|
||||
|
||||
return self.lookup(*key) is not None
|
||||
|
||||
# iterate over all mdirs, this includes the mrootchain
|
||||
def _leaves(self, *,
|
||||
path=False,
|
||||
@@ -2129,9 +2072,6 @@ class Mtree:
|
||||
else:
|
||||
yield rattr
|
||||
|
||||
def __iter__(self):
|
||||
return self.mids()
|
||||
|
||||
# lookup by name
|
||||
def _namelookupleaf(self, did, name, *,
|
||||
path=False,
|
||||
@@ -3113,9 +3053,6 @@ class Lfs:
|
||||
break
|
||||
mid = self.mid(mbid, mrid)
|
||||
|
||||
def __iter__(self):
|
||||
return self.files()
|
||||
|
||||
def orphans(self,
|
||||
all=False):
|
||||
all_ = all; del all
|
||||
@@ -3493,9 +3430,6 @@ class Lfs:
|
||||
path=path,
|
||||
depth=depth)
|
||||
|
||||
def __iter__(self):
|
||||
return self.datas()
|
||||
|
||||
# some convience operations for reading data
|
||||
def bytes(self, *,
|
||||
depth=None):
|
||||
|
||||
Reference in New Issue
Block a user