diff options
| -rw-r--r-- | python/label.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/python/label.py b/python/label.py index 41618bb..c154275 100644 --- 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") |