Fix for PY2

wkentaro · Jun 5, 2018 · 0fcc8de · 0fcc8de
1 parent 58001de
commit 0fcc8de
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/gdown/__init__.py b/gdown/__init__.py
@@ -155,7 +155,10 @@ def main():
     args = parser.parse_args()
 
     if args.output == '-':
-        args.output = sys.stdout.buffer
+        if six.PY3:
+            args.output = sys.stdout.buffer
+        else:
+            args.output = sys.stdout
 
     if args.id:
         url = 'https://drive.google.com/uc?id={id}'.format(id=args.url_or_id)