Small refactor in messaging.

[wolnelektury.git] / src / catalogue / management / commands / gencover.py

diff --git a/src/catalogue/management/commands/gencover.py b/src/catalogue/management/commands/gencover.py
index 420ea63..c0d099b 100644 (file)
--- a/src/catalogue/management/commands/gencover.py
+++ b/src/catalogue/management/commands/gencover.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-

 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #


@@ -14,12 +13,17 @@ class Command(BaseCommand):
         parser.add_argument('--width', type=int)
         parser.add_argument('--height', type=int)
         parser.add_argument('--bleed', action='store_true')
         parser.add_argument('--width', type=int)
         parser.add_argument('--height', type=int)
         parser.add_argument('--bleed', action='store_true')

+        parser.add_argument('--cover-class', type=str)

 
     def handle(self, *args, **options):
         slug = options['slug']
         width = options['width']
         height = options.get('height')
 
     def handle(self, *args, **options):
         slug = options['slug']
         width = options['width']
         height = options.get('height')

+        cover_class = options.get('cover_class')

         bleed = 20 if options['bleed'] else 0
         wldoc = Book.objects.get(slug=slug).wldocument()
         bleed = 20 if options['bleed'] else 0
         wldoc = Book.objects.get(slug=slug).wldocument()

-        cover = make_cover(wldoc.book_info, width=width, height=height, bleed=bleed)
+        kwargs = {}
+        if cover_class:
+            kwargs['cover_class'] = cover_class
+        cover = make_cover(wldoc.book_info, width=width, height=height, bleed=bleed, **kwargs)

         cover.save('%s.jpg' % slug)
         cover.save('%s.jpg' % slug)