diff options
Diffstat (limited to 'isort/main.py')
| -rw-r--r-- | isort/main.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/isort/main.py b/isort/main.py index 5da3fb3a..1bafdf95 100644 --- a/isort/main.py +++ b/isort/main.py @@ -330,7 +330,13 @@ def main(argv=None): file_names = arguments.pop('files', []) if file_names == ['-']: - SortImports(file_contents=sys.stdin.read(), write_to_stdout=True, **arguments) + try: + # python 3 + file_ = sys.stdin.buffer + except AttributeError: + # python 2 + file_ = sys.stdin + SortImports(file_=file_, write_to_stdout=True, **arguments) else: if not file_names: file_names = ['.'] |