Skip to content
This repository was archived by the owner on Nov 3, 2023. It is now read-only.

Commit 8566623

Browse files
committed
Fix indent issues in parsing and PEP8.
1 parent 59081e7 commit 8566623

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/pydocstyle/parser.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -224,6 +224,8 @@ def __init__(self, message):
224224

225225

226226
class TokenStream(object):
227+
NEWLINES = {tk.NEWLINE, tk.INDENT, tk.DEDENT}
228+
227229
def __init__(self, filelike):
228230
self._generator = tk.generate_tokens(filelike.readline)
229231
self.current = Token(*next(self._generator, None))
@@ -236,7 +238,7 @@ def move(self):
236238
current = self._next_from_generator()
237239
self.current = None if current is None else Token(*current)
238240
self.line = self.current.start[0] if self.current else self.line
239-
self.got_newline = (previous.kind == tk.NEWLINE)
241+
self.got_newline = (previous.kind in self.NEWLINES)
240242
return previous
241243

242244
def _next_from_generator(self):
@@ -400,7 +402,6 @@ def parse_definitions(self, class_, all=False):
400402
else:
401403
self.stream.move()
402404

403-
404405
def parse_all(self):
405406
"""Parse the __all__ definition in a module."""
406407
assert self.current.value == '__all__'

0 commit comments

Comments
 (0)