Fix reading from a closed stdin buffer

JBKahn · Mar 12, 2019 · fe18e84 · fe18e84
1 parent e5d3812
commit fe18e84
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/README.rst b/README.rst
@@ -28,6 +28,11 @@ available in ``flake8``::
 Changes
 -------
 
+3.1.1 - 2019-03-12
+``````````````````
+
+* Fix reading from stdin when it is closed.
+
 3.1.0 - 2018-02-11
 ``````````````````
 * Add a framework classifier for use in pypi.org

diff --git a/flake8_print.py b/flake8_print.py
@@ -3,7 +3,12 @@
 import ast
 from six import PY2, PY3
 
-__version__ = '3.1.0'
+try:
+    from flake8.engine import pep8 as stdin_utils
+except ImportError:
+    from flake8 import utils as stdin_utils
+
+__version__ = '3.1.1'
 
 PRINT_FUNCTION_NAME = "print"
 PPRINT_FUNCTION_NAME = "pprint"
@@ -78,7 +83,7 @@ def __init__(self, tree, filename):
     def load_file(self):
         if self.filename in ("stdin", "-", None):
             self.filename = "stdin"
-            self.lines = pycodestyle.stdin_get_value().splitlines(True)
+            self.lines = stdin_utils.stdin_get_value().splitlines(True)
         else:
             self.lines = pycodestyle.readlines(self.filename)