src/librarian/cover.py

   1 # -*- coding: utf-8 -*-
   2 #
   3 # This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
   4 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   5 #
   6 from __future__ import unicode_literals
   7
   8 import re
   9 from PIL import Image, ImageFont, ImageDraw, ImageFilter
  10 from six import BytesIO
  11 from librarian import get_resource, OutputFile, URLOpener
  12
  13
  14 class Metric(object):
  15     """Gets metrics from an object, scaling it by a factor."""
  16     def __init__(self, obj, scale):
  17         self._obj = obj
  18         self._scale = float(scale)
  19
  20     def __getattr__(self, name):
  21         src = getattr(self._obj, name)
  22         if src and self._scale:
  23             return type(src)(self._scale * src)
  24         else:
  25             return src
  26
  27
  28 class TextBox(object):
  29     """Creates an Image with a series of centered strings."""
  30
  31     SHADOW_X = 3
  32     SHADOW_Y = 3
  33     SHADOW_BLUR = 3
  34
  35     def __init__(self, max_width, max_height, padding_x=None, padding_y=None, bar_width=0, bar_color=None):
  36         if padding_x is None:
  37             padding_x = self.SHADOW_X + self.SHADOW_BLUR
  38         if padding_y is None:
  39             padding_y = self.SHADOW_Y + self.SHADOW_BLUR
  40
  41         self.max_width = max_width
  42         self.bar_width = bar_width
  43         self.bar_color = bar_color
  44         self.max_text_width = max_width - 2 * padding_x - bar_width
  45         self.padding_x = padding_x
  46         self.padding_y = padding_y
  47         self.height = padding_y
  48         self.img = Image.new('RGBA', (max_width, max_height))
  49         self.draw = ImageDraw.Draw(self.img)
  50         self.shadow_img = None
  51         self.shadow_draw = None
  52
  53     def skip(self, height):
  54         """Skips some vertical space."""
  55         self.height += height
  56
  57     def text(self, text, color='#000', font=None, line_height=20,
  58              shadow_color=None, centering=True):
  59         """Writes some centered text."""
  60         text = re.sub(r'\s+', ' ', text)
  61         if shadow_color:
  62             if not self.shadow_img:
  63                 self.shadow_img = Image.new('RGBA', self.img.size)
  64                 self.shadow_draw = ImageDraw.Draw(self.shadow_img)
  65         while text:
  66             line = text
  67             line_width = self.draw.textsize(line, font=font)[0]
  68             while line_width > self.max_text_width:
  69                 parts = line.rsplit(' ', 1)
  70                 if len(parts) == 1:
  71                     line_width = self.max_text_width
  72                     break
  73                 line = parts[0]
  74
  75                 if line[-2] == ' ':
  76                     line = line[:-2]
  77
  78                 line_width = self.draw.textsize(line, font=font)[0]
  79             line = line.strip() + ' '
  80
  81             if centering:
  82                 pos_x = (self.max_width - line_width) // 2
  83             else:
  84                 pos_x = self.bar_width + self.padding_x
  85
  86             if shadow_color:
  87                 self.shadow_draw.text(
  88                         (pos_x + self.SHADOW_X, self.height + self.SHADOW_Y),
  89                         line, font=font, fill=shadow_color
  90                 )
  91
  92             self.draw.text((pos_x, self.height), line, font=font, fill=color)
  93             self.height += line_height
  94             # go to next line
  95             text = text[len(line):]
  96
  97     def image(self):
  98         """Creates the actual Image object."""
  99         image = Image.new('RGBA', (self.max_width,
 100                                    int(round(self.height + self.padding_y))))
 101
 102         if self.shadow_img:
 103             shadow = self.shadow_img.filter(ImageFilter.BLUR)
 104             image.paste(shadow, (0, 0), shadow)
 105             image.paste(self.img, (0, 0), self.img)
 106         else:
 107             image.paste(self.img, (0, 0))
 108         return image
 109
 110
 111 class Cover(object):
 112     """Abstract base class for cover images generator."""
 113     width = 600
 114     height = 800
 115     background_color = '#fff'
 116     background_img = None
 117
 118     author_top = 100
 119     author_margin_left = 20
 120     author_margin_right = 20
 121     author_lineskip = 40
 122     author_color = '#000'
 123     author_shadow = None
 124     author_font_ttf = get_resource('fonts/DejaVuSerif.ttf')
 125     author_font_size = 30
 126
 127     title_top = 100
 128     title_margin_left = 20
 129     title_margin_right = 20
 130     title_lineskip = 54
 131     title_color = '#000'
 132     title_shadow = None
 133     title_font_ttf = get_resource('fonts/DejaVuSerif.ttf')
 134     title_font_size = 40
 135
 136     logo_bottom = None
 137     logo_width = None
 138     uses_dc_cover = False
 139
 140     format = 'JPEG'
 141     scale = 1
 142     scale_after = 1
 143
 144     exts = {
 145         'JPEG': 'jpg',
 146         'PNG': 'png',
 147         }
 148
 149     mime_types = {
 150         'JPEG': 'image/jpeg',
 151         'PNG': 'image/png',
 152         }
 153
 154     def __init__(self, book_info, format=None, width=None, height=None, cover_logo=None):
 155         self.authors = [auth.readable() for auth in book_info.authors]
 156         self.title = book_info.title
 157         if format is not None:
 158             self.format = format
 159         if width and height:
 160             self.height = int(round(height * self.width / width))
 161         scale = max(float(width or 0) / self.width,
 162                     float(height or 0) / self.height)
 163         if scale >= 1:
 164             self.scale = scale
 165         elif scale:
 166             self.scale_after = scale
 167
 168     def pretty_authors(self):
 169         """Allows for decorating authors' names."""
 170         return [self.authors]
 171
 172     def pretty_title(self):
 173         """Allows for decorating title."""
 174         return self.title
 175
 176     def image(self):
 177         metr = Metric(self, self.scale)
 178         img = Image.new('RGB', (metr.width, metr.height),
 179                         self.background_color)
 180
 181         if self.background_img:
 182             background = Image.open(self.background_img)
 183             img.paste(background, None, background)
 184             del background
 185
 186         # WL logo
 187         if metr.logo_width:
 188             logo = Image.open(get_resource('res/wl-logo.png'))
 189             logo = logo.resize((
 190                 metr.logo_width,
 191                 int(round(logo.size[1] * metr.logo_width / logo.size[0]))
 192             ))
 193             img.paste(logo, (
 194                 (metr.width - metr.logo_width) // 2,
 195                 img.size[1] - logo.size[1] - metr.logo_bottom
 196             ))
 197
 198         top = metr.author_top
 199         tbox = TextBox(
 200             metr.width - metr.author_margin_left - metr.author_margin_right,
 201             metr.height - top,
 202             )
 203
 204         author_font = ImageFont.truetype(
 205             self.author_font_ttf, metr.author_font_size,
 206             layout_engine=ImageFont.LAYOUT_BASIC)
 207         for pa in self.pretty_authors():
 208             tbox.text(pa, self.author_color, author_font, metr.author_lineskip,
 209                       self.author_shadow)
 210         text_img = tbox.image()
 211         img.paste(text_img, (metr.author_margin_left, top), text_img)
 212
 213         top += text_img.size[1] + metr.title_top
 214         tbox = TextBox(
 215             metr.width - metr.title_margin_left - metr.title_margin_right,
 216             metr.height - top,
 217             )
 218         title_font = ImageFont.truetype(
 219             self.title_font_ttf, metr.title_font_size,
 220             layout_engine=ImageFont.LAYOUT_BASIC)
 221         tbox.text(self.pretty_title(), self.title_color, title_font,
 222                   metr.title_lineskip, self.title_shadow)
 223         text_img = tbox.image()
 224         img.paste(text_img, (metr.title_margin_left, top), text_img)
 225
 226         return img
 227
 228     def final_image(self):
 229         img = self.image()
 230         if self.scale_after != 1:
 231             img = img.resize((
 232                     int(round(img.size[0] * self.scale_after)),
 233                     int(round(img.size[1] * self.scale_after))),
 234                 Image.ANTIALIAS)
 235         return img
 236
 237     def mime_type(self):
 238         return self.mime_types[self.format]
 239
 240     def ext(self):
 241         return self.exts[self.format]
 242
 243     def save(self, *args, **kwargs):
 244         default_kwargs = {
 245                 'format': self.format,
 246                 'quality': 95,
 247         }
 248         default_kwargs.update(kwargs)
 249         return self.final_image().save(*args, **default_kwargs)
 250
 251     def output_file(self, *args, **kwargs):
 252         imgstr = BytesIO()
 253         self.save(imgstr, *args, **kwargs)
 254         return OutputFile.from_bytes(imgstr.getvalue())
 255
 256
 257 class WLCover(Cover):
 258     """Wolne Lektury cover without logos."""
 259     width = 600
 260     height = 833
 261     uses_dc_cover = True
 262     author_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
 263     author_font_size = 20
 264     author_lineskip = 30
 265     author_centering = True
 266     title_font_ttf = get_resource('fonts/DejaVuSerif-Bold.ttf')
 267     title_font_size = 30
 268     title_lineskip = 40
 269     title_box_width = 350
 270     title_centering = True
 271
 272     box_top_margin = 100
 273     box_bottom_margin = 100
 274     box_padding_y = 20
 275     box_above_line = 10
 276     box_below_line = 15
 277     box_line_left = 75
 278     box_line_right = 275
 279     box_line_width = 2
 280     box_padding_x = 0
 281     box_bar_width = 0
 282
 283     logo_top = 15
 284     logo_width = 140
 285
 286     bar_width = 35
 287     bar_color = '#000'
 288     box_position = 'middle'
 289     background_color = '#444'
 290     author_color = '#444'
 291     background_img = get_resource('res/cover.png')
 292     format = 'JPEG'
 293
 294     epoch_colors = {
 295         u'Starożytność': '#9e3610',
 296         u'Średniowiecze': '#564c09',
 297         u'Renesans': '#8ca629',
 298         u'Barok': '#a6820a',
 299         u'Oświecenie': '#f2802e',
 300         u'Romantyzm': '#db4b16',
 301         u'Pozytywizm': '#961060',
 302         u'Modernizm': '#7784e0',
 303         u'Dwudziestolecie międzywojenne': '#3044cf',
 304         u'Współczesność': '#06393d',
 305     }
 306     set_title_color = True
 307
 308     kind_box_position = {
 309         u'Liryka': 'top',
 310         u'Epika': 'bottom',
 311     }
 312
 313     def __init__(self, book_info, format=None, width=None, height=None,
 314                  bleed=0, cover_logo=None):
 315         super(WLCover, self).__init__(book_info, format=format, width=width,
 316                                       height=height)
 317         # Set box position.
 318         self.box_position = book_info.cover_box_position or \
 319             self.kind_box_position.get(book_info.kind, self.box_position)
 320         # Set bar color.
 321         if book_info.cover_bar_color == 'none':
 322             self.bar_width = 0
 323         else:
 324             self.bar_color = book_info.cover_bar_color or \
 325                 self.get_variable_color(book_info) or self.bar_color
 326         # Set title color.
 327         if self.set_title_color:
 328             self.title_color = self.get_variable_color(book_info) or self.title_color
 329
 330         self.bleed = bleed
 331         self.box_top_margin += bleed
 332         self.box_bottom_margin += bleed
 333         self.bar_width += bleed
 334
 335         if book_info.cover_url:
 336             url = book_info.cover_url
 337             bg_src = None
 338             if bg_src is None:
 339                 bg_src = URLOpener().open(url)
 340             self.background_img = BytesIO(bg_src.read())
 341             bg_src.close()
 342
 343     def get_variable_color(self, book_info):
 344         return self.epoch_colors.get(book_info.epoch, None)
 345
 346     def pretty_authors(self):
 347         return [a.upper() for a in self.authors]
 348
 349     def add_box(self, img):
 350         if self.box_position == 'none':
 351             return img
 352
 353         metr = Metric(self, self.scale)
 354
 355         # Write author name.
 356         box = TextBox(metr.title_box_width, metr.height,
 357                       padding_y=metr.box_padding_y,
 358                       padding_x=metr.box_padding_x,
 359                       bar_width=metr.box_bar_width,
 360                       bar_color=self.bar_color,
 361                       )
 362         author_font = ImageFont.truetype(
 363             self.author_font_ttf, metr.author_font_size,
 364             layout_engine=ImageFont.LAYOUT_BASIC)
 365         for pa in self.pretty_authors():
 366             box.text(pa, font=author_font, line_height=metr.author_lineskip,
 367                      color=self.author_color, shadow_color=self.author_shadow,
 368                      centering=self.author_centering
 369                      )
 370
 371         box.skip(metr.box_above_line)
 372         box.draw.line(
 373             (metr.box_line_left, box.height, metr.box_line_right, box.height),
 374             fill=self.author_color, width=metr.box_line_width
 375         )
 376         box.skip(metr.box_below_line)
 377
 378         # Write title.
 379         title_font = ImageFont.truetype(
 380             self.title_font_ttf, metr.title_font_size,
 381             layout_engine=ImageFont.LAYOUT_BASIC)
 382         box.text(self.pretty_title(),
 383                  line_height=metr.title_lineskip,
 384                  font=title_font,
 385                  color=self.title_color,
 386                  shadow_color=self.title_shadow,
 387                  centering=self.title_centering
 388                  )
 389
 390         box_img = box.image()
 391
 392         # Find box position.
 393         if self.box_position == 'bottom' or box_img.size[1] + metr.box_top_margin + metr.box_bottom_margin > metr.height:
 394             box_top = metr.height - metr.box_bottom_margin - box_img.size[1]
 395         elif self.box_position == 'top':
 396             box_top = metr.box_top_margin
 397         else:   # Middle.
 398             box_top = (metr.height - box_img.size[1]) // 2
 399
 400         box_left = metr.bar_width + (
 401             metr.width - metr.bar_width - box_img.size[0]
 402         ) // 2
 403
 404         # Draw the white box.
 405         img_draw = ImageDraw.Draw(img)
 406         img_draw.rectangle(
 407             (
 408                 box_left,
 409                 box_top,
 410                 box_left + box_img.size[0],
 411                 box_top + box_img.size[1]
 412             ),
 413             fill='#fff'
 414         )
 415         # Paste the contents into the white box.
 416         img.paste(box_img, (box_left, box_top), box_img)
 417         if self.box_bar_width:
 418             img_draw.rectangle(
 419                 (
 420                     box_left,
 421                     box_top,
 422                     box_left + metr.box_bar_width,
 423                     box_top + box_img.size[1]
 424                 ),
 425                 fill=self.bar_color
 426             )
 427         return img
 428
 429     def add_cut_lines(self, img):
 430         line_ratio = 0.5
 431         if self.bleed == 0:
 432             return img
 433         metr = Metric(self, self.scale)
 434         draw = ImageDraw.Draw(img)
 435         for corner_x, corner_y in (
 436                 (0, 0), (metr.width, 0),
 437                 (0, metr.height), (metr.width, metr.height)
 438                 ):
 439             dir_x = 1 if corner_x == 0 else -1
 440             dir_y = 1 if corner_y == 0 else -1
 441             for offset in (-1, 0, 1):
 442                 draw.line(
 443                     (
 444                         corner_x,
 445                         corner_y + dir_y * metr.bleed + offset,
 446                         corner_x + dir_x * metr.bleed * line_ratio,
 447                         corner_y + dir_y * metr.bleed + offset
 448                     ),
 449                     fill='black' if offset == 0 else 'white',
 450                     width=1
 451                 )
 452                 draw.line(
 453                     (
 454                         corner_x + dir_x * metr.bleed + offset,
 455                         corner_y,
 456                         corner_x + dir_x * metr.bleed + offset,
 457                         corner_y + dir_y * metr.bleed * line_ratio
 458                     ),
 459                     fill='black' if offset == 0 else 'white',
 460                     width=1
 461                 )
 462         return img
 463
 464     def image(self):
 465         metr = Metric(self, self.scale)
 466         img = Image.new('RGB', (metr.width, metr.height),
 467                         self.background_color)
 468         draw = ImageDraw.Draw(img)
 469
 470         draw.rectangle((0, 0, metr.bar_width, metr.height),
 471                        fill=self.bar_color)
 472
 473         if self.background_img:
 474             src = Image.open(self.background_img)
 475             trg_size = (metr.width - metr.bar_width, metr.height)
 476             if src.size[0] * trg_size[1] < src.size[1] * trg_size[0]:
 477                 resized = (
 478                     trg_size[0],
 479                     int(round(src.size[1] * trg_size[0] / src.size[0]))
 480                 )
 481                 cut = (resized[1] - trg_size[1]) // 2
 482                 src = src.resize(resized, Image.ANTIALIAS)
 483                 src = src.crop((0, cut, src.size[0], src.size[1] - cut))
 484             else:
 485                 resized = (
 486                     int(round(src.size[0] * trg_size[1] / src.size[1])),
 487                     trg_size[1],
 488                 )
 489                 cut = (resized[0] - trg_size[0]) // 2
 490                 src = src.resize(resized, Image.ANTIALIAS)
 491                 src = src.crop((cut, 0, src.size[0] - cut, src.size[1]))
 492
 493             img.paste(src, (metr.bar_width, 0))
 494             del src
 495
 496         img = self.add_box(img)
 497
 498         # img = self.add_cut_lines(img)
 499
 500         return img
 501
 502
 503 class WLNoBoxCover(WLCover):
 504     def add_box(self, img):
 505         return img
 506
 507
 508 class LogoWLCover(WLCover):
 509     gradient_height = 90
 510     gradient_easing = 90
 511     gradient_logo_height = 57
 512     gradient_logo_max_width = 200
 513     gradient_logo_margin_right = 30
 514     gradient_logo_spacing = 40
 515     gradient_color = '#000'
 516     gradient_opacity = .6
 517     gradient_logos = [
 518         'res/wl-logo-white.png',
 519         'res/fnp-logo-white.png',
 520     ]
 521     annotation = None
 522     annotation_height = 10
 523
 524     logos_right = True
 525     gradient_logo_centering = False
 526
 527
 528     def __init__(self, book_info, *args, cover_logo=None, **kwargs):
 529         super(LogoWLCover, self).__init__(book_info, *args, **kwargs)
 530         self.gradient_height += self.bleed
 531         self.gradient_logo_margin_right += self.bleed
 532
 533         self.additional_cover_logos = [
 534             BytesIO(URLOpener().open(cover_logo_url).read())
 535             for cover_logo_url in book_info.cover_logo_urls
 536         ]
 537         self.end_cover_logos = []
 538         if cover_logo:
 539             self.additional_cover_logos.append(
 540                     open(cover_logo, 'rb')
 541                     )
 542
 543     @property
 544     def has_gradient_logos(self):
 545         return self.gradient_logos or self.additional_cover_logos or self.end_cover_logos or self.annotation
 546
 547     def add_gradient_logos(self, img, metr):
 548         gradient = Image.new(
 549             'RGBA',
 550             (metr.width - metr.bar_width, metr.gradient_height),
 551             self.gradient_color
 552         )
 553         gradient_mask = Image.new(
 554             'L',
 555             (metr.width - metr.bar_width, metr.gradient_height)
 556         )
 557         draw = ImageDraw.Draw(gradient_mask)
 558         for line in range(0, metr.gradient_easing):
 559             draw.line(
 560                 (0, line, metr.width - metr.bar_width, line),
 561                 fill=int(
 562                     255 * self.gradient_opacity * line / metr.gradient_easing
 563                 )
 564             )
 565         draw.rectangle((
 566             0, metr.gradient_easing,
 567             metr.width - metr.bar_width, metr.gradient_height
 568         ), fill=int(255 * self.gradient_opacity))
 569
 570
 571         img.paste(gradient,
 572                   (metr.bar_width, metr.height - metr.gradient_height),
 573                   mask=gradient_mask)
 574
 575         if self.logos_right:
 576             cursor = metr.width - metr.gradient_logo_margin_right
 577         else:
 578             cursor = metr.gradient_logo_margin_right
 579         logo_top = int(
 580             metr.height
 581             - metr.gradient_height / 2
 582             - metr.gradient_logo_height / 2 - metr.bleed / 2
 583         )
 584
 585         logos = [
 586             get_resource(logo_path)
 587             for logo_path in self.gradient_logos
 588         ]
 589
 590         logos = self.additional_cover_logos + logos + self.end_cover_logos
 591
 592         if self.logos_right:
 593             logos.reverse()
 594
 595         logos = [
 596             Image.open(logo_bytes).convert('RGBA')
 597             for logo_bytes in logos
 598         ]
 599
 600         # See if logos fit into the gradient. If not, scale down accordingly.
 601         space_for_logos = (
 602             metr.width
 603             - metr.bar_width
 604             - 2 * metr.gradient_logo_margin_right
 605         )
 606         widths = [
 607             min(
 608                 metr.gradient_logo_max_width,
 609                 logo.size[0] * metr.gradient_logo_height / logo.size[1]
 610             )
 611             for logo in logos]
 612         taken_space = sum(widths) + (len(logos) - 1) * metr.gradient_logo_spacing
 613         if taken_space > space_for_logos:
 614             logo_scale = space_for_logos / taken_space
 615         else:
 616             logo_scale = 1
 617             if self.gradient_logo_centering:
 618                 cursor += int((space_for_logos - taken_space) / 2)
 619         logo_scale = (
 620             space_for_logos / taken_space
 621             if taken_space > space_for_logos else 1
 622         )
 623         #logo_top += int(metr.gradient_logo_height * (1 - logo_scale) / 2)
 624
 625         for i, logo in enumerate(logos):
 626             if i == -1:
 627                 L_scale = 1
 628             else:
 629                 L_scale = logo_scale
 630             L_top = logo_top + int(metr.gradient_logo_height * (1 - L_scale) / 2)
 631
 632             logo = logo.resize(
 633                 (
 634                     int(round(widths[i] * L_scale)),
 635                     int(round(
 636                         logo.size[1] * widths[i] / logo.size[0] * L_scale
 637                     ))
 638                 ),
 639                 Image.ANTIALIAS)
 640             if self.logos_right:
 641                 cursor -= logo.size[0]
 642
 643             img.paste(
 644                 logo,
 645                 (
 646                     cursor,
 647                     L_top
 648                     #int(round(logo_top + (metr.gradient_logo_height - logo.size[1]) * L_scale / 2))
 649                 ),
 650                 mask=logo
 651             )
 652             if self.logos_right:
 653                 cursor -= int(round(metr.gradient_logo_spacing * logo_scale))
 654             else:
 655                 cursor += int(round(metr.gradient_logo_spacing * logo_scale)) + logo.size[0]
 656
 657         if self.annotation:
 658             img2 = Image.new('RGBA', (metr.height, metr.height), color=None)
 659             draw = ImageDraw.Draw(img2)
 660             author_font = ImageFont.truetype(
 661                 self.author_font_ttf, metr.annotation_height,
 662                 layout_engine=ImageFont.LAYOUT_BASIC)
 663             draw.text((self.annotation_height, self.annotation_height), self.annotation, font=author_font, fill='#FFFFFF')
 664             img2.show()
 665             img2 = img2.rotate(90)
 666             img2.show()
 667             img.putalpha(0)
 668             img.alpha_composite(img2, (0, 0))
 669             img = img.convert('RGB')
 670
 671         return img
 672
 673     def image(self):
 674         img = super(LogoWLCover, self).image()
 675         metr = Metric(self, self.scale)
 676         if self.has_gradient_logos:
 677             img = self.add_gradient_logos(img, metr)
 678         return img
 679
 680
 681 class LegimiCover(LogoWLCover):
 682     width = 210
 683     height = 297
 684     bar_width = 0
 685     # Other bar
 686
 687     author_font_ttf = get_resource('fonts/Lora-Regular.ttf')
 688     author_font_size = 15
 689     author_lineskip = 19.5
 690     author_centering = False
 691     title_font_ttf = get_resource('fonts/Lora-Bold.ttf')
 692     title_font_size = 15
 693     title_lineskip = 19.5
 694     title_centering = False
 695
 696     title_box_width = 210
 697
 698     box_bottom_margin = 20
 699     box_padding_x = 20
 700     box_padding_y = 10   # do baseline
 701     box_above_line = 6
 702     box_below_line = 0
 703     box_line_left = 0
 704     box_line_right = 0
 705
 706     box_line_width = 0
 707
 708     box_bar_width = 20
 709
 710     #logo_top = 15
 711     #logo_width = 140
 712
 713     bar_color = '#000'
 714     box_position = 'bottom'
 715     background_color = '#444'
 716     author_color = '#000'
 717     title_color = '#000'
 718     set_title_color = False
 719
 720     kind_box_position = {}
 721
 722     box_bottom_margin_logos_add = 10
 723     gradient_height = 30
 724     gradient_easing = 0
 725     gradient_logo_height = 20
 726     gradient_logo_max_width = 200
 727     gradient_logo_margin_right = 10
 728     gradient_logo_spacing = 20
 729     gradient_color = '#000'
 730     gradient_opacity = .5
 731     gradient_logos = [
 732         'res/wl-logo-white.png',
 733     ]
 734     logos_right = False
 735     gradient_logo_centering = True
 736
 737     genre_colors = {
 738         'Artykuł': '#bf001a',
 739         'Artykuł naukowy': '#bf001a',
 740         'Dziennik': '#bf001a',
 741         'Esej': '#bf001a',
 742         'Felieton': '#bf001a',
 743         'Kronika': '#bf001a',
 744         'List': '#bf001a',
 745         'Manifest': '#bf001a',
 746         'Odczyt': '#bf001a',
 747         'Pamiętnik': '#bf001a',
 748         'Poradnik': '#bf001a',
 749         'Praca naukowa': '#bf001a',
 750         'Publicystyka': '#bf001a',
 751         'Reportaż': '#bf001a',
 752         'Reportaż podróżniczy': '#bf001a',
 753         'Rozprawa': '#bf001a',
 754         'Rozprawa polityczna': '#bf001a',
 755         'Traktat': '#bf001a',
 756     }
 757     kind_colors = {
 758         'Epika': '#9bbb2b',
 759         'Liryka': '#3e626f',
 760         'Dramat': '#ecbe00',
 761     }
 762
 763     def __init__(self, *args, **kwargs):
 764         super().__init__(*args, **kwargs)
 765         if self.has_gradient_logos:
 766             self.box_bottom_margin += self.box_bottom_margin_logos_add
 767
 768     def pretty_authors(self):
 769         return self.authors
 770
 771     def get_variable_color(self, book_info):
 772         for genre in book_info.genres:
 773             if genre in self.genre_colors:
 774                 return self.genre_colors[genre]
 775         for kind in book_info.kinds:
 776             if kind in self.kind_colors:
 777                 return self.kind_colors[kind]
 778
 779 class LegimiCornerCover(LegimiCover):
 780     gradient_logos = []
 781     corner_width = 120
 782     corner_image = get_resource('res/book-band.png')
 783
 784     def image(self):
 785         image = super().image()
 786         metr = Metric(self, self.scale)
 787
 788         if self.corner_image:
 789             corner = Image.open(self.corner_image).convert('RGBA')
 790             corner = corner.resize(
 791                 (
 792                     int(round(metr.corner_width)),
 793                     int(round(corner.size[1] / corner.size[0] * metr.corner_width))
 794                 )
 795             )
 796             image.paste(corner, (
 797                 metr.width - int(round(metr.corner_width)),
 798                 0,
 799             ), corner)
 800         return image
 801
 802 class LegimiAudiobookCover(LegimiCornerCover):
 803     corner_width = 82.5
 804     corner_image = get_resource('res/audiobook-sticker.png')
 805     height = 210
 806
 807
 808 class EbookpointCover(LogoWLCover):
 809     gradient_logo_height = 58
 810     gradient_logo_spacing = 25
 811     gradient_logos = [
 812         'res/ebookpoint-logo-white.png',
 813         'res/wl-logo-white.png',
 814         'res/fnp-logo-white.png',
 815     ]
 816
 817
 818 class VirtualoCover(Cover):
 819     width = 600
 820     height = 730
 821     author_top = 73
 822     title_top = 73
 823     logo_bottom = 25
 824     logo_width = 250
 825
 826
 827 class PrestigioCover(Cover):
 828     width = 580
 829     height = 783
 830     background_img = get_resource('res/cover-prestigio.png')
 831
 832     author_top = 446
 833     author_margin_left = 118
 834     author_margin_right = 62
 835     author_lineskip = 60
 836     author_color = '#fff'
 837     author_shadow = '#000'
 838     author_font_ttf = get_resource('fonts/JunicodeWL-Italic.ttf')
 839     author_font_size = 50
 840
 841     title_top = 0
 842     title_margin_left = 118
 843     title_margin_right = 62
 844     title_lineskip = 60
 845     title_color = '#fff'
 846     title_shadow = '#000'
 847     title_font_ttf = get_resource('fonts/JunicodeWL-Italic.ttf')
 848     title_font_size = 50
 849
 850     def pretty_title(self):
 851         return u"„%s”" % self.title
 852
 853
 854 class BookotekaCover(Cover):
 855     width = 2140
 856     height = 2733
 857     background_img = get_resource('res/cover-bookoteka.png')
 858
 859     author_top = 480
 860     author_margin_left = 307
 861     author_margin_right = 233
 862     author_lineskip = 156
 863     author_color = '#d9d919'
 864     author_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
 865     author_font_size = 130
 866
 867     title_top = 400
 868     title_margin_left = 307
 869     title_margin_right = 233
 870     title_lineskip = 168
 871     title_color = '#d9d919'
 872     title_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
 873     title_font_size = 140
 874
 875     format = 'PNG'
 876
 877
 878 class GandalfCover(Cover):
 879     width = 600
 880     height = 730
 881     background_img = get_resource('res/cover-gandalf.png')
 882     author_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
 883     author_font_size = 30
 884     title_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
 885     title_font_size = 40
 886     logo_bottom = 25
 887     logo_width = 250
 888     format = 'PNG'
 889
 890
 891 class KMLUCover(LogoWLCover):
 892     gradient_logo_height = 58
 893     gradient_logo_spacing = 25
 894     gradient_logos = [
 895         'res/kmlu-logo-white.png',
 896         'res/wl-logo-white.png',
 897         'res/fnp-logo-white.png',
 898     ]
 899
 900
 901 class MPWCover(LogoWLCover):
 902     gradient_logo_height = 57
 903     gradient_logo_spacing = 25
 904     gradient_logos = [
 905         'res/mpw-logo-white.png',
 906         'res/wl-logo-white.png',
 907         'res/fnp-logo-white.png',
 908     ]
 909
 910
 911 class AtriumCover(LogoWLCover):
 912     gradient_logo_height = 58
 913     gradient_logo_spacing = 25
 914     gradient_logos = [
 915         'res/atrium-logo.png',
 916         'res/wl-logo-white.png',
 917         'res/fnp-logo-white.png',
 918     ]
 919
 920
 921 class BNCover(LogoWLCover):
 922     gradient_logos = [
 923         'res/dofinansowano.png',
 924         'res/MKIDN.jpg',
 925         'res/BN.png',
 926         'res/wl-logo-white.png',
 927     ]
 928 #    annotation = 'Zadanie „Udostępnienie publikacji w formatach cyfrowych” w ramach Narodowego Programu Rozwoju Czytelnictwa. Dofinansowano ze środków Ministra Kultury, Dziedzictwa Narodowego i Sportu.'
 929
 930
 931 COVER_CLASSES = {
 932     'default': LogoWLCover,
 933     'kmlu': KMLUCover,
 934     'mpw': MPWCover,
 935     'atrium': AtriumCover,
 936     'bn': BNCover,
 937     'legimi': LegimiCover,
 938     'legimi-corner': LegimiCornerCover,
 939     'legimi-audiobook': LegimiAudiobookCover,
 940 }
 941
 942
 943 def make_cover(book_info, *args, **kwargs):
 944     if 'cover_class' in kwargs:
 945         cover_class_name = kwargs.pop('cover_class')
 946     else:
 947         cover_class_name = book_info.cover_class
 948     cover_class = COVER_CLASSES[cover_class_name]
 949     return cover_class(book_info, *args, **kwargs)