return bytes, not BytesIO object for preview

sumpfork · Aug 10, 2018 · b2c4851 · b2c4851
1 parent 88690e0
commit b2c4851
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/domdiv/draw.py b/domdiv/draw.py
@@ -28,6 +28,7 @@ class DividerDrawer(object):
     def __init__(self):
         self.odd = True
         self.canvas = None
+        self.options = None
 
     @staticmethod
     def get_image_filepath(fname):

diff --git a/domdiv/main.py b/domdiv/main.py
@@ -512,7 +512,7 @@ def generate_sample(options):
     with Image(blob=buf.getvalue(), resolution=options.preview_resolution) as sample:
         sample.format = 'png'
         sample.save(sample_out)
-        return sample_out
+        return sample_out.getvalue()
 
 
 def parse_papersize(spec):