Make use of yield in __iter__
This commit is contained in:
@@ -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'>" % (
|
||||
|
||||
Reference in New Issue
Block a user