Run pilbox.image in py2 or py3

agschwender · Oct 16, 2017 · ca02d79 · ca02d79
1 parent 7e4824a
commit ca02d79
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pilbox/image.py b/pilbox/image.py
@@ -518,7 +518,10 @@ def main():
                         quality=options.quality,
                         progressive=options.progressive,
                         preserve_exif=options.preserve_exif)
-    sys.stdout.write(stream.read())
+    try:
+        sys.stdout.buffer.write(stream.read())
+    except AttributeError:
+        sys.stdout.write(stream.read())
     stream.close()