Skip to content

Commit

Permalink
fix: correct the OO mess to fix PDFContentParser
Browse files Browse the repository at this point in the history
  • Loading branch information
dhdaines committed Sep 19, 2024
1 parent 37f3b2e commit ba55790
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions playa/psparser.py
Original file line number Diff line number Diff line change
Expand Up @@ -870,7 +870,7 @@ def nextobject(self) -> PSStackEntry[ExtraT]:
:return: keywords, literals, strings, numbers, arrays and dictionaries.
"""
while not self.results:
(pos, token) = self._parser.nexttoken()
(pos, token) = self.nexttoken()
if isinstance(token, (int, float, bool, str, bytes, PSLiteral)):
# normal token
self.push((pos, token))
Expand Down Expand Up @@ -952,7 +952,10 @@ def read(self, pos: int, objlen: int) -> bytes:
return self._parser.read(pos, objlen)

def nexttoken(self) -> Tuple[int, PSBaseParserToken]:
return self._parser.nexttoken()
try:
return self.__next__()
except StopIteration:
raise PSEOF

def get_inline_data(self, target: bytes = b"EI") -> Tuple[int, bytes]:
return self._parser.get_inline_data(target)
Expand All @@ -961,4 +964,4 @@ def __iter__(self):
return self

def __next__(self):
return next(self._parser)
return self._parser.__next__()

0 comments on commit ba55790

Please sign in to comment.