added suport for chunk read of clusters

This commit is contained in:
Erik Frojdh
2023-06-05 10:20:14 +02:00
parent 8aa1b6bcbd
commit 221f7e0c0d
4 changed files with 61 additions and 12 deletions

View File

@ -2,5 +2,22 @@
from . import ClusterFileReader
class ClusterFile(ClusterFileReader):
def __init__(self, fname):
super().__init__(fname)
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
def __iter__(self):
return self
def __next__(self):
clusters = self.read()
if clusters.size == 0:
raise StopIteration
else:
return clusters
def __enter__(self):
return self
def __exit__(self, exc_type, exc_value, traceback):
pass