🐎 FileWrapper.peek returns None when out of elements

miyuchina · Jan 9, 2018 · c458c79 · c458c79
1 parent 49c9b87
commit c458c79
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/mistletoe/block_token.py b/mistletoe/block_token.py
@@ -431,11 +431,10 @@ def start(cls, line):
     @classmethod
     def read(cls, lines):
         line_buffer = []
-        try:
-            while cls._is_legal(lines.peek()) and lines.peek() != '\n':
-                line_buffer.append(next(lines))
-        except StopIteration:
-            pass
+        while (lines.peek() is not None
+                and cls._is_legal(lines.peek())
+                and lines.peek() != '\n'):
+            line_buffer.append(next(lines))
         return line_buffer
 
 

diff --git a/mistletoe/block_tokenizer.py b/mistletoe/block_tokenizer.py
@@ -36,7 +36,7 @@ def reset(self):
     def peek(self):
         if self._index + 1 < len(self.lines):
             return self.lines[self._index+1]
-        raise StopIteration
+        return None
 
 
 def tokenize(iterable, token_types, root=None):