Make use of yield in __iter__

This commit is contained in:
2023-03-15 03:33:24 +01:00
parent 73eaa0e89b
commit 404f71dddd
2 changed files with 3 additions and 10 deletions

View File

@@ -167,15 +167,8 @@ class Bitstream:
return self
def __iter__(self) -> Bitstream:
self._iter_index = 0
return self
def __next__(self) -> bool:
if self._iter_index >= self._bit_length:
raise StopIteration
bit = self[self._iter_index]
self._iter_index += 1
return bit
for i in range(len(self)):
yield self[i]
def __repr__(self) -> str:
return "<Bitstream addr:0x%012x offset:%d len:%d data:'%s'>" % (