fix

[librarian.git] / src / librarian / cover.py

diff --git a/src/librarian/cover.py b/src/librarian/cover.py
index 0e9b079..3581f59 100644 (file)
--- a/src/librarian/cover.py
+++ b/src/librarian/cover.py
@@ -1,17 +1,13 @@
-# -*- coding: utf-8 -*-
-#
 # This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
-# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+# Copyright © Fundacja Wolne Lektury. See NOTICE for more information.
 #
-from __future__ import unicode_literals
-
 import re
 from PIL import Image, ImageFont, ImageDraw, ImageFilter
-from six import BytesIO
+from io import BytesIO
 from librarian import get_resource, OutputFile, URLOpener
 
 
-class Metric(object):
+class Metric:
     """Gets metrics from an object, scaling it by a factor."""
     def __init__(self, obj, scale):
         self._obj = obj
@@ -25,7 +21,7 @@ class Metric(object):
             return src
 
 
-class TextBox(object):
+class TextBox:
     """Creates an Image with a series of centered strings."""
 
     SHADOW_X = 3
@@ -108,7 +104,7 @@ class TextBox(object):
         return image
 
 
-class Cover(object):
+class Cover:
     """Abstract base class for cover images generator."""
     width = 600
     height = 800
@@ -153,11 +149,21 @@ class Cover(object):
 
     def __init__(self, book_info, format=None, width=None, height=None, cover_logo=None):
         self.book_info = book_info
+
+        self.predesigned = False
+        if book_info.cover_class == 'image':
+            self.predesigned = True
+
+        # TODO: deprecated
+        if book_info.cover_box_position == 'none':
+            self.predesigned = True
+
         self.authors = [auth.readable() for auth in book_info.authors]
         self.title = book_info.title
         if format is not None:
             self.format = format
         self.set_size(width, height)
+        self.cover_logo = cover_logo
 
     def set_size(self, width, height):
         if width and height:
@@ -207,7 +213,7 @@ class Cover(object):
 
         author_font = ImageFont.truetype(
             self.author_font_ttf, metr.author_font_size,
-            layout_engine=ImageFont.LAYOUT_BASIC)
+            layout_engine=ImageFont.Layout.BASIC)
         for pa in self.pretty_authors():
             tbox.text(pa, self.author_color, author_font, metr.author_lineskip,
                       self.author_shadow)
@@ -221,7 +227,7 @@ class Cover(object):
             )
         title_font = ImageFont.truetype(
             self.title_font_ttf, metr.title_font_size,
-            layout_engine=ImageFont.LAYOUT_BASIC)
+            layout_engine=ImageFont.Layout.BASIC)
         tbox.text(self.pretty_title(), self.title_color, title_font,
                   metr.title_lineskip, self.title_shadow)
         text_img = tbox.image()
@@ -235,7 +241,7 @@ class Cover(object):
             img = img.resize((
                     int(round(img.size[0] * self.scale_after)),
                     int(round(img.size[1] * self.scale_after))),
-                Image.ANTIALIAS)
+                Image.Resampling.LANCZOS)
         return img
 
     def mime_type(self):
@@ -297,22 +303,22 @@ class WLCover(Cover):
     format = 'JPEG'
 
     epoch_colors = {
-        u'Starożytność': '#9e3610',
-        u'Średniowiecze': '#564c09',
-        u'Renesans': '#8ca629',
-        u'Barok': '#a6820a',
-        u'Oświecenie': '#f2802e',
-        u'Romantyzm': '#db4b16',
-        u'Pozytywizm': '#961060',
-        u'Modernizm': '#7784e0',
-        u'Dwudziestolecie międzywojenne': '#3044cf',
-        u'Współczesność': '#06393d',
+        'Starożytność': '#9e3610',
+        'Średniowiecze': '#564c09',
+        'Renesans': '#8ca629',
+        'Barok': '#a6820a',
+        'Oświecenie': '#f2802e',
+        'Romantyzm': '#db4b16',
+        'Pozytywizm': '#961060',
+        'Modernizm': '#7784e0',
+        'Dwudziestolecie międzywojenne': '#3044cf',
+        'Współczesność': '#06393d',
     }
     set_title_color = True
 
     kind_box_position = {
-        u'Liryka': 'top',
-        u'Epika': 'bottom',
+        'Liryka': 'top',
+        'Epika': 'bottom',
     }
 
     def __init__(self, book_info, format=None, width=None, height=None,
@@ -378,7 +384,7 @@ class WLCover(Cover):
                       )
         author_font = ImageFont.truetype(
             self.author_font_ttf, metr.author_font_size,
-            layout_engine=ImageFont.LAYOUT_BASIC)
+            layout_engine=ImageFont.Layout.BASIC)
         for pa in self.pretty_authors():
             box.text(pa, font=author_font, line_height=metr.author_lineskip,
                      color=self.author_color, shadow_color=self.author_shadow,
@@ -395,7 +401,7 @@ class WLCover(Cover):
         # Write title.
         title_font = ImageFont.truetype(
             self.title_font_ttf, metr.title_font_size,
-            layout_engine=ImageFont.LAYOUT_BASIC)
+            layout_engine=ImageFont.Layout.BASIC)
         box.text(self.pretty_title(),
                  line_height=metr.title_lineskip,
                  font=title_font,
@@ -499,7 +505,7 @@ class WLCover(Cover):
                     cut = 0
                 else:
                     cut = (resized[1] - trg_size[1]) // 2
-                src = src.resize(resized, Image.ANTIALIAS)
+                src = src.resize(resized, Image.Resampling.LANCZOS)
                 src = src.crop((0, cut, src.size[0], src.size[1] - cut))
             else:
                 resized = (
@@ -507,7 +513,7 @@ class WLCover(Cover):
                     trg_size[1],
                 )
                 cut = (resized[0] - trg_size[0]) // 2
-                src = src.resize(resized, Image.ANTIALIAS)
+                src = src.resize(resized, Image.Resampling.LANCZOS)
                 src = src.crop((cut, 0, src.size[0] - cut, src.size[1]))
 
             img.paste(src, (metr.bar_width, 0))
@@ -662,7 +668,7 @@ class LogoWLCover(WLCover):
                         logo.size[1] * widths[i] / logo.size[0] * L_scale
                     ))
                 ),
-                Image.ANTIALIAS)
+                Image.Resampling.LANCZOS)
             if self.logos_right:
                 cursor -= logo.size[0]
 
@@ -685,7 +691,7 @@ class LogoWLCover(WLCover):
             draw = ImageDraw.Draw(img2)
             author_font = ImageFont.truetype(
                 self.author_font_ttf, metr.annotation_height,
-                layout_engine=ImageFont.LAYOUT_BASIC)
+                layout_engine=ImageFont.Layout.BASIC)
             draw.text((self.annotation_height, self.annotation_height), self.annotation, font=author_font, fill='#FFFFFF')
             img2.show()
             img2 = img2.rotate(90)
@@ -875,7 +881,7 @@ class PrestigioCover(Cover):
     title_font_size = 50
 
     def pretty_title(self):
-        return u"„%s”" % self.title
+        return "„%s”" % self.title
 
 
 class BookotekaCover(Cover):
@@ -1005,9 +1011,10 @@ COVER_CLASSES = {
 
 
 def make_cover(book_info, *args, **kwargs):
+    cover_class_name = None
     if 'cover_class' in kwargs:
         cover_class_name = kwargs.pop('cover_class')
-    else:
-        cover_class_name = book_info.cover_class
+    if not cover_class_name:
+        cover_class_name = 'default'
     cover_class = COVER_CLASSES[cover_class_name]
     return cover_class(book_info, *args, **kwargs)