Fix reading from closed stdin

tylerwince · Sep 24, 2019 · f0fe747 · f0fe747
1 parent cd579bd
commit f0fe747
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/flake8_bandit.py b/flake8_bandit.py
@@ -15,8 +15,13 @@
 except ImportError:
     import ConfigParser as configparser
 
+try:
+    from flake8.engine import pep8 as stdin_utils
+except ImportError:
+    from flake8 import utils as stdin_utils
+
 
-__version__ = "2.1.1"
+__version__ = "2.1.2"
 
 
 class BanditTester(object):
@@ -90,7 +95,7 @@ def _load_source(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)
         if not self.tree: