fix landscape

diff --git a/python/label.py b/python/label.py
index 41618bb..c154275 100644 (file)
--- a/python/label.py
+++ b/python/label.py
@@ -3,12 +3,12 @@ from PIL import Image, ImageFont, ImageDraw
 class label:
        def __init__(self, length, width=336, landscape=False):
                assert(width == 336 or width == 720)
-               self.landscape = False
+               self.landscape = landscape
                self.fontfile = '/usr/share/fonts/truetype/ttf-dejavu/DejaVuSans-Bold.ttf'
 
                self.im = Image.new("1", (width, length), "white")
                if self.landscape:
-                       self.im.rotate(90)
+                       self.im = self.im.rotate(90)
 
        def set_font(self, fontfile):
                self.fontfile = fontfile
@@ -21,6 +21,6 @@ class label:
                draw.text(pos, text, font=self.font)
 
        def save(self, outfile):
-               if self.landscape:
-                       self.im.rotate(90)
+               if not self.landscape:
+                       self.im = self.im.rotate(90)
                self.im.save(outfile, "PNG")