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.book_info = book_info
 156         self.authors = [auth.readable() for auth in book_info.authors]
 157         self.title = book_info.title
 158         if format is not None:
 159             self.format = format
 160         self.set_size(width, height)
 161
 162     def set_size(self, width, height):
 163         if width and height:
 164             self.height = int(round(height * self.width / width))
 165         scale = max(float(width or 0) / self.width,
 166                     float(height or 0) / self.height)
 167         if scale >= 1:
 168             self.scale = scale
 169         elif scale:
 170             self.scale_after = scale
 171
 172     def pretty_authors(self):
 173         """Allows for decorating authors' names."""
 174         return self.authors
 175
 176     def pretty_title(self):
 177         """Allows for decorating title."""
 178         return self.title
 179
 180     def image(self):
 181         metr = Metric(self, self.scale)
 182         img = Image.new('RGB', (metr.width, metr.height),
 183                         self.background_color)
 184
 185         if self.background_img:
 186             background = Image.open(self.background_img)
 187             img.paste(background, None, background)
 188             del background
 189
 190         # WL logo
 191         if metr.logo_width:
 192             logo = Image.open(get_resource('res/wl-logo.png'))
 193             logo = logo.resize((
 194                 metr.logo_width,
 195                 int(round(logo.size[1] * metr.logo_width / logo.size[0]))
 196             ))
 197             img.paste(logo, (
 198                 (metr.width - metr.logo_width) // 2,
 199                 img.size[1] - logo.size[1] - metr.logo_bottom
 200             ))
 201
 202         top = metr.author_top
 203         tbox = TextBox(
 204             metr.width - metr.author_margin_left - metr.author_margin_right,
 205             metr.height - top,
 206             )
 207
 208         author_font = ImageFont.truetype(
 209             self.author_font_ttf, metr.author_font_size,
 210             layout_engine=ImageFont.LAYOUT_BASIC)
 211         for pa in self.pretty_authors():
 212             tbox.text(pa, self.author_color, author_font, metr.author_lineskip,
 213                       self.author_shadow)
 214         text_img = tbox.image()
 215         img.paste(text_img, (metr.author_margin_left, top), text_img)
 216
 217         top += text_img.size[1] + metr.title_top
 218         tbox = TextBox(
 219             metr.width - metr.title_margin_left - metr.title_margin_right,
 220             metr.height - top,
 221             )
 222         title_font = ImageFont.truetype(
 223             self.title_font_ttf, metr.title_font_size,
 224             layout_engine=ImageFont.LAYOUT_BASIC)
 225         tbox.text(self.pretty_title(), self.title_color, title_font,
 226                   metr.title_lineskip, self.title_shadow)
 227         text_img = tbox.image()
 228         img.paste(text_img, (metr.title_margin_left, top), text_img)
 229
 230         return img
 231
 232     def final_image(self):
 233         img = self.image()
 234         if self.scale_after != 1:
 235             img = img.resize((
 236                     int(round(img.size[0] * self.scale_after)),
 237                     int(round(img.size[1] * self.scale_after))),
 238                 Image.ANTIALIAS)
 239         return img
 240
 241     def mime_type(self):
 242         return self.mime_types[self.format]
 243
 244     def ext(self):
 245         return self.exts[self.format]
 246
 247     def save(self, *args, **kwargs):
 248         default_kwargs = {
 249                 'format': self.format,
 250                 'quality': 95,
 251         }
 252         default_kwargs.update(kwargs)
 253         return self.final_image().save(*args, **default_kwargs)
 254
 255     def output_file(self, *args, **kwargs):
 256         imgstr = BytesIO()
 257         self.save(imgstr, *args, **kwargs)
 258         return OutputFile.from_bytes(imgstr.getvalue())
 259
 260
 261 class WLCover(Cover):
 262     """Wolne Lektury cover without logos."""
 263     width = 600
 264     height = 833
 265     uses_dc_cover = True
 266     author_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
 267     author_font_size = 20
 268     author_lineskip = 30
 269     author_centering = True
 270     title_font_ttf = get_resource('fonts/DejaVuSerif-Bold.ttf')
 271     title_font_size = 30
 272     title_lineskip = 40
 273     title_box_width = 350
 274     title_centering = True
 275
 276     box_top_margin = 100
 277     box_bottom_margin = 100
 278     box_padding_y = 20
 279     box_above_line = 10
 280     box_below_line = 15
 281     box_line_left = 75
 282     box_line_right = 275
 283     box_line_width = 2
 284     box_padding_x = 0
 285     box_bar_width = 0
 286
 287     logo_top = 15
 288     logo_width = 140
 289
 290     bar_width = 35
 291     bar_color = '#000'
 292     box_position = 'middle'
 293     background_color = '#444'
 294     author_color = '#444'
 295     background_img = get_resource('res/cover.png')
 296     background_top = False
 297     format = 'JPEG'
 298
 299     epoch_colors = {
 300         u'Starożytność': '#9e3610',
 301         u'Średniowiecze': '#564c09',
 302         u'Renesans': '#8ca629',
 303         u'Barok': '#a6820a',
 304         u'Oświecenie': '#f2802e',
 305         u'Romantyzm': '#db4b16',
 306         u'Pozytywizm': '#961060',
 307         u'Modernizm': '#7784e0',
 308         u'Dwudziestolecie międzywojenne': '#3044cf',
 309         u'Współczesność': '#06393d',
 310     }
 311     set_title_color = True
 312
 313     kind_box_position = {
 314         u'Liryka': 'top',
 315         u'Epika': 'bottom',
 316     }
 317
 318     def __init__(self, book_info, format=None, width=None, height=None,
 319                  bleed=0, cover_logo=None):
 320         super(WLCover, self).__init__(book_info, format=format, width=width,
 321                                       height=height)
 322
 323         self.slug = book_info.url.slug
 324         # Set box position.
 325         self.box_position = book_info.cover_box_position or \
 326             self.kind_box_position.get(book_info.kind, self.box_position)
 327         # Set bar color.
 328         if book_info.cover_bar_color == 'none':
 329             self.bar_width = 0
 330         else:
 331             self.bar_color = book_info.cover_bar_color or \
 332                 self.get_variable_color(book_info) or self.bar_color
 333         # Set title color.
 334         if self.set_title_color:
 335             self.title_color = self.get_variable_color(book_info) or self.title_color
 336
 337         self.bleed = bleed
 338         self.box_top_margin += bleed
 339         self.box_bottom_margin += bleed
 340         self.bar_width += bleed
 341
 342         if book_info.cover_url:
 343             url = book_info.cover_url
 344             bg_src = None
 345             while True:
 346                 try:
 347                     if bg_src is None:
 348                         import requests
 349                         bg_src = requests.get(url, timeout=5)
 350                     self.background_img = BytesIO(bg_src.content)
 351                     bg_src.close()
 352                 except Exception as e:
 353                     bg_src = None
 354                     print(e)
 355                     import time
 356                     time.sleep(1)
 357                 else:
 358                     break
 359
 360     def get_variable_color(self, book_info):
 361         return self.epoch_colors.get(book_info.epoch, None)
 362
 363     def pretty_authors(self):
 364         return [a.upper() for a in self.authors]
 365
 366     def add_box(self, img):
 367         if self.box_position == 'none':
 368             return img
 369
 370         metr = Metric(self, self.scale)
 371
 372         # Write author name.
 373         box = TextBox(metr.title_box_width - 2 * self.bleed, metr.height,
 374                       padding_y=metr.box_padding_y,
 375                       padding_x=metr.box_padding_x,
 376                       bar_width=metr.box_bar_width,
 377                       bar_color=self.bar_color,
 378                       )
 379         author_font = ImageFont.truetype(
 380             self.author_font_ttf, metr.author_font_size,
 381             layout_engine=ImageFont.LAYOUT_BASIC)
 382         for pa in self.pretty_authors():
 383             box.text(pa, font=author_font, line_height=metr.author_lineskip,
 384                      color=self.author_color, shadow_color=self.author_shadow,
 385                      centering=self.author_centering
 386                      )
 387
 388         box.skip(metr.box_above_line)
 389         box.draw.line(
 390             (metr.box_line_left, box.height, metr.box_line_right, box.height),
 391             fill=self.author_color, width=metr.box_line_width
 392         )
 393         box.skip(metr.box_below_line)
 394
 395         # Write title.
 396         title_font = ImageFont.truetype(
 397             self.title_font_ttf, metr.title_font_size,
 398             layout_engine=ImageFont.LAYOUT_BASIC)
 399         box.text(self.pretty_title(),
 400                  line_height=metr.title_lineskip,
 401                  font=title_font,
 402                  color=self.title_color,
 403                  shadow_color=self.title_shadow,
 404                  centering=self.title_centering
 405                  )
 406
 407         box_img = box.image()
 408
 409         # Find box position.
 410         if self.box_position == 'bottom' or box_img.size[1] + metr.box_top_margin + metr.box_bottom_margin > metr.height:
 411             box_top = metr.height - metr.box_bottom_margin - box_img.size[1]
 412         elif self.box_position == 'top':
 413             box_top = metr.box_top_margin
 414         else:   # Middle.
 415             box_top = (metr.height - box_img.size[1]) // 2
 416
 417         box_left = metr.bar_width + (
 418             metr.width - metr.bar_width - box_img.size[0] - self.bleed
 419         ) // 2
 420
 421         # Draw the white box.
 422         img_draw = ImageDraw.Draw(img)
 423         img_draw.rectangle(
 424             (
 425                 box_left,
 426                 box_top,
 427                 box_left + box_img.size[0],
 428                 box_top + box_img.size[1]
 429             ),
 430             fill='#fff'
 431         )
 432         # Paste the contents into the white box.
 433         img.paste(box_img, (box_left, box_top), box_img)
 434         if self.box_bar_width:
 435             img_draw.rectangle(
 436                 (
 437                     box_left,
 438                     box_top,
 439                     box_left + metr.box_bar_width,
 440                     box_top + box_img.size[1]
 441                 ),
 442                 fill=self.bar_color
 443             )
 444         return img
 445
 446     def add_cut_lines(self, img):
 447         line_ratio = 0.5
 448         if self.bleed == 0:
 449             return img
 450         metr = Metric(self, self.scale)
 451         draw = ImageDraw.Draw(img)
 452         for corner_x, corner_y in (
 453                 (0, 0), (metr.width, 0),
 454                 (0, metr.height), (metr.width, metr.height)
 455                 ):
 456             dir_x = 1 if corner_x == 0 else -1
 457             dir_y = 1 if corner_y == 0 else -1
 458             for offset in (-1, 0, 1):
 459                 draw.line(
 460                     (
 461                         corner_x,
 462                         corner_y + dir_y * metr.bleed + offset,
 463                         corner_x + dir_x * metr.bleed * line_ratio,
 464                         corner_y + dir_y * metr.bleed + offset
 465                     ),
 466                     fill='black' if offset == 0 else 'white',
 467                     width=1
 468                 )
 469                 draw.line(
 470                     (
 471                         corner_x + dir_x * metr.bleed + offset,
 472                         corner_y,
 473                         corner_x + dir_x * metr.bleed + offset,
 474                         corner_y + dir_y * metr.bleed * line_ratio
 475                     ),
 476                     fill='black' if offset == 0 else 'white',
 477                     width=1
 478                 )
 479         return img
 480
 481     def image(self):
 482         metr = Metric(self, self.scale)
 483         img = Image.new('RGB', (metr.width, metr.height),
 484                         self.background_color)
 485         draw = ImageDraw.Draw(img)
 486
 487         draw.rectangle((0, 0, metr.bar_width, metr.height),
 488                        fill=self.bar_color)
 489
 490         if self.background_img:
 491             src = Image.open(self.background_img)
 492             trg_size = (metr.width - metr.bar_width, metr.height)
 493             if src.size[0] * trg_size[1] < src.size[1] * trg_size[0]:
 494                 resized = (
 495                     trg_size[0],
 496                     int(round(src.size[1] * trg_size[0] / src.size[0]))
 497                 )
 498                 if self.background_top:
 499                     cut = 0
 500                 else:
 501                     cut = (resized[1] - trg_size[1]) // 2
 502                 src = src.resize(resized, Image.ANTIALIAS)
 503                 src = src.crop((0, cut, src.size[0], src.size[1] - cut))
 504             else:
 505                 resized = (
 506                     int(round(src.size[0] * trg_size[1] / src.size[1])),
 507                     trg_size[1],
 508                 )
 509                 cut = (resized[0] - trg_size[0]) // 2
 510                 src = src.resize(resized, Image.ANTIALIAS)
 511                 src = src.crop((cut, 0, src.size[0] - cut, src.size[1]))
 512
 513             img.paste(src, (metr.bar_width, 0))
 514             del src
 515
 516         img = self.add_box(img)
 517
 518         # img = self.add_cut_lines(img)
 519
 520         return img
 521
 522
 523 class WLNoBoxCover(WLCover):
 524     def add_box(self, img):
 525         return img
 526
 527
 528 class LogoWLCover(WLCover):
 529     gradient_height = 90
 530     gradient_easing = 90
 531     gradient_logo_height = 57
 532     gradient_logo_max_width = 200
 533     gradient_logo_margin_right = 30
 534     gradient_logo_spacing = 40
 535     gradient_color = '#000'
 536     gradient_opacity = .6
 537     gradient_logos = [
 538         'res/wl-logo-white.png',
 539         'res/fnp-logo-white.png',
 540     ]
 541     annotation = None
 542     annotation_height = 10
 543
 544     logos_right = True
 545     gradient_logo_centering = False
 546
 547     qrcode = None
 548
 549
 550     def __init__(self, book_info, *args, cover_logo=None, **kwargs):
 551         super(LogoWLCover, self).__init__(book_info, *args, **kwargs)
 552         self.gradient_height += self.bleed
 553         self.gradient_logo_margin_right += self.bleed
 554
 555         self.additional_cover_logos = [
 556             BytesIO(URLOpener().open(cover_logo_url).read())
 557             for cover_logo_url in book_info.cover_logo_urls
 558         ]
 559         self.end_cover_logos = []
 560         if cover_logo:
 561             self.additional_cover_logos.append(
 562                     open(cover_logo, 'rb')
 563                     )
 564
 565     @property
 566     def has_gradient_logos(self):
 567         return self.gradient_logos or self.additional_cover_logos or self.end_cover_logos or self.annotation or self.qrcode is not None
 568
 569     def add_gradient_logos(self, img, metr):
 570         gradient = Image.new(
 571             'RGBA',
 572             (metr.width - metr.bar_width, metr.gradient_height),
 573             self.gradient_color
 574         )
 575         gradient_mask = Image.new(
 576             'L',
 577             (metr.width - metr.bar_width, metr.gradient_height)
 578         )
 579         draw = ImageDraw.Draw(gradient_mask)
 580         for line in range(0, metr.gradient_easing):
 581             draw.line(
 582                 (0, line, metr.width - metr.bar_width, line),
 583                 fill=int(
 584                     255 * self.gradient_opacity * line / metr.gradient_easing
 585                 )
 586             )
 587         draw.rectangle((
 588             0, metr.gradient_easing,
 589             metr.width - metr.bar_width, metr.gradient_height
 590         ), fill=int(255 * self.gradient_opacity))
 591
 592
 593         img.paste(gradient,
 594                   (metr.bar_width, metr.height - metr.gradient_height),
 595                   mask=gradient_mask)
 596
 597         if self.logos_right:
 598             cursor = metr.width - metr.gradient_logo_margin_right
 599         else:
 600             cursor = metr.gradient_logo_margin_right
 601         logo_top = int(
 602             metr.height
 603             - metr.gradient_height / 2
 604             - metr.gradient_logo_height / 2 - metr.bleed
 605         )
 606
 607         logos = [
 608             get_resource(logo_path)
 609             for logo_path in self.gradient_logos
 610         ]
 611
 612         logos = self.additional_cover_logos + logos + self.end_cover_logos
 613
 614         logos = [
 615             Image.open(logo_bytes).convert('RGBA')
 616             for logo_bytes in logos
 617         ]
 618
 619         if self.qrcode is not None:
 620             import qrcode
 621             logos.append(qrcode.make(f'https://wolnelektury.pl/katalog/lektura/{self.slug}/?{self.qrcode}'))
 622
 623         if self.logos_right:
 624             logos.reverse()
 625
 626         # See if logos fit into the gradient. If not, scale down accordingly.
 627         space_for_logos = (
 628             metr.width
 629             - metr.bar_width
 630             - 2 * metr.gradient_logo_margin_right + self.bleed
 631         )
 632         widths = [
 633             min(
 634                 metr.gradient_logo_max_width,
 635                 logo.size[0] * metr.gradient_logo_height / logo.size[1]
 636             )
 637             for logo in logos]
 638         taken_space = sum(widths) + (len(logos) - 1) * metr.gradient_logo_spacing
 639         if taken_space > space_for_logos:
 640             logo_scale = space_for_logos / taken_space
 641         else:
 642             logo_scale = 1
 643             if self.gradient_logo_centering:
 644                 cursor += int((space_for_logos - taken_space) / 2)
 645         logo_scale = (
 646             space_for_logos / taken_space
 647             if taken_space > space_for_logos else 1
 648         )
 649         #logo_top += int(metr.gradient_logo_height * (1 - logo_scale) / 2)
 650
 651         for i, logo in enumerate(logos):
 652             if i == -1:
 653                 L_scale = 1
 654             else:
 655                 L_scale = logo_scale
 656             L_top = logo_top + int(metr.gradient_logo_height * (1 - L_scale) / 2)
 657
 658             logo = logo.resize(
 659                 (
 660                     int(round(widths[i] * L_scale)),
 661                     int(round(
 662                         logo.size[1] * widths[i] / logo.size[0] * L_scale
 663                     ))
 664                 ),
 665                 Image.ANTIALIAS)
 666             if self.logos_right:
 667                 cursor -= logo.size[0]
 668
 669             img.paste(
 670                 logo,
 671                 (
 672                     cursor,
 673                     L_top
 674                     #int(round(logo_top + (metr.gradient_logo_height - logo.size[1]) * L_scale / 2))
 675                 ),
 676                 mask=logo
 677             )
 678             if self.logos_right:
 679                 cursor -= int(round(metr.gradient_logo_spacing * logo_scale))
 680             else:
 681                 cursor += int(round(metr.gradient_logo_spacing * logo_scale)) + logo.size[0]
 682
 683         if self.annotation:
 684             img2 = Image.new('RGBA', (metr.height, metr.height), color=None)
 685             draw = ImageDraw.Draw(img2)
 686             author_font = ImageFont.truetype(
 687                 self.author_font_ttf, metr.annotation_height,
 688                 layout_engine=ImageFont.LAYOUT_BASIC)
 689             draw.text((self.annotation_height, self.annotation_height), self.annotation, font=author_font, fill='#FFFFFF')
 690             img2.show()
 691             img2 = img2.rotate(90)
 692             img2.show()
 693             img.putalpha(0)
 694             img.alpha_composite(img2, (0, 0))
 695             img = img.convert('RGB')
 696
 697         return img
 698
 699     def image(self):
 700         img = super(LogoWLCover, self).image()
 701         metr = Metric(self, self.scale)
 702         if self.has_gradient_logos:
 703             img = self.add_gradient_logos(img, metr)
 704         return img
 705
 706
 707 class LegimiCover(LogoWLCover):
 708     width = 210
 709     height = 297
 710     bar_width = 0
 711     # Other bar
 712
 713     author_font_ttf = get_resource('fonts/Lora-Regular.ttf')
 714     author_font_size = 15
 715     author_lineskip = 19.5
 716     author_centering = False
 717     title_font_ttf = get_resource('fonts/Lora-Bold.ttf')
 718     title_font_size = 15
 719     title_lineskip = 19.5
 720     title_centering = False
 721
 722     title_box_width = 210
 723
 724     box_bottom_margin = 20
 725     box_padding_x = 20
 726     box_padding_y = 10   # do baseline
 727     box_above_line = 6
 728     box_below_line = 0
 729     box_line_left = 0
 730     box_line_right = 0
 731
 732     box_line_width = 0
 733
 734     box_bar_width = 20
 735
 736     #logo_top = 15
 737     #logo_width = 140
 738
 739     bar_color = '#000'
 740     box_position = 'bottom'
 741     background_color = '#444'
 742     author_color = '#000'
 743     title_color = '#000'
 744     set_title_color = False
 745
 746     kind_box_position = {}
 747
 748     box_bottom_margin_logos_add = 10
 749     gradient_height = 30
 750     gradient_easing = 0
 751     gradient_logo_height = 20
 752     gradient_logo_max_width = 200
 753     gradient_logo_margin_right = 10
 754     gradient_logo_spacing = 20
 755     gradient_color = '#000'
 756     gradient_opacity = .5
 757     gradient_logos = [
 758         'res/wl-logo-white.png',
 759     ]
 760     logos_right = False
 761     gradient_logo_centering = True
 762     background_top = True
 763
 764     genre_colors = {
 765         'Artykuł': '#bf001a',
 766         'Artykuł naukowy': '#bf001a',
 767         'Dziennik': '#bf001a',
 768         'Esej': '#bf001a',
 769         'Felieton': '#bf001a',
 770         'Kronika': '#bf001a',
 771         'List': '#bf001a',
 772         'Manifest': '#bf001a',
 773         'Odczyt': '#bf001a',
 774         'Pamiętnik': '#bf001a',
 775         'Poradnik': '#bf001a',
 776         'Praca naukowa': '#bf001a',
 777         'Publicystyka': '#bf001a',
 778         'Reportaż': '#bf001a',
 779         'Reportaż podróżniczy': '#bf001a',
 780         'Rozprawa': '#bf001a',
 781         'Rozprawa polityczna': '#bf001a',
 782         'Traktat': '#bf001a',
 783     }
 784     kind_colors = {
 785         'Epika': '#9bbb2b',
 786         'Liryka': '#3e626f',
 787         'Dramat': '#ecbe00',
 788     }
 789
 790     def __init__(self, *args, **kwargs):
 791         super().__init__(*args, **kwargs)
 792         if self.has_gradient_logos:
 793             self.box_bottom_margin += self.box_bottom_margin_logos_add
 794
 795     def pretty_authors(self):
 796         return self.authors
 797
 798     def get_variable_color(self, book_info):
 799         for genre in book_info.genres:
 800             if genre in self.genre_colors:
 801                 return self.genre_colors[genre]
 802         for kind in book_info.kinds:
 803             if kind in self.kind_colors:
 804                 return self.kind_colors[kind]
 805
 806 class LegimiCornerCover(LegimiCover):
 807     gradient_logos = []
 808     corner_width = 120
 809     corner_image = get_resource('res/book-band.png')
 810
 811     def image(self):
 812         image = super().image()
 813         metr = Metric(self, self.scale)
 814
 815         if self.corner_image:
 816             corner = Image.open(self.corner_image).convert('RGBA')
 817             corner = corner.resize(
 818                 (
 819                     int(round(metr.corner_width)),
 820                     int(round(corner.size[1] / corner.size[0] * metr.corner_width))
 821                 )
 822             )
 823             image.paste(corner, (
 824                 metr.width - int(round(metr.corner_width)),
 825                 0,
 826             ), corner)
 827         return image
 828
 829 class LegimiAudiobookCover(LegimiCornerCover):
 830     corner_width = 82.5
 831     corner_image = get_resource('res/audiobook-sticker.png')
 832     height = 210
 833
 834
 835 class EbookpointCover(LogoWLCover):
 836     gradient_logo_height = 58
 837     gradient_logo_spacing = 25
 838     gradient_logos = [
 839         'res/ebookpoint-logo-white.png',
 840         'res/wl-logo-white.png',
 841         'res/fnp-logo-white.png',
 842     ]
 843
 844
 845 class VirtualoCover(Cover):
 846     width = 600
 847     height = 730
 848     author_top = 73
 849     title_top = 73
 850     logo_bottom = 25
 851     logo_width = 250
 852
 853
 854 class PrestigioCover(Cover):
 855     width = 580
 856     height = 783
 857     background_img = get_resource('res/cover-prestigio.png')
 858
 859     author_top = 446
 860     author_margin_left = 118
 861     author_margin_right = 62
 862     author_lineskip = 60
 863     author_color = '#fff'
 864     author_shadow = '#000'
 865     author_font_ttf = get_resource('fonts/JunicodeWL-Italic.ttf')
 866     author_font_size = 50
 867
 868     title_top = 0
 869     title_margin_left = 118
 870     title_margin_right = 62
 871     title_lineskip = 60
 872     title_color = '#fff'
 873     title_shadow = '#000'
 874     title_font_ttf = get_resource('fonts/JunicodeWL-Italic.ttf')
 875     title_font_size = 50
 876
 877     def pretty_title(self):
 878         return u"„%s”" % self.title
 879
 880
 881 class BookotekaCover(Cover):
 882     width = 2140
 883     height = 2733
 884     background_img = get_resource('res/cover-bookoteka.png')
 885
 886     author_top = 480
 887     author_margin_left = 307
 888     author_margin_right = 233
 889     author_lineskip = 156
 890     author_color = '#d9d919'
 891     author_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
 892     author_font_size = 130
 893
 894     title_top = 400
 895     title_margin_left = 307
 896     title_margin_right = 233
 897     title_lineskip = 168
 898     title_color = '#d9d919'
 899     title_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
 900     title_font_size = 140
 901
 902     format = 'PNG'
 903
 904
 905 class GandalfCover(Cover):
 906     width = 600
 907     height = 730
 908     background_img = get_resource('res/cover-gandalf.png')
 909     author_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
 910     author_font_size = 30
 911     title_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
 912     title_font_size = 40
 913     logo_bottom = 25
 914     logo_width = 250
 915     format = 'PNG'
 916
 917
 918 class KMLUCover(LogoWLCover):
 919     gradient_logo_height = 58
 920     gradient_logo_spacing = 25
 921     gradient_logos = [
 922         'res/kmlu-logo-white.png',
 923         'res/wl-logo-white.png',
 924         'res/fnp-logo-white.png',
 925     ]
 926
 927
 928 class MPWCover(LogoWLCover):
 929     gradient_logo_height = 57
 930     gradient_logo_spacing = 25
 931     gradient_logos = [
 932         'res/mpw-logo-white.png',
 933         'res/wl-logo-white.png',
 934         'res/fnp-logo-white.png',
 935     ]
 936
 937
 938 class AtriumCover(LogoWLCover):
 939     gradient_logo_height = 58
 940     gradient_logo_spacing = 25
 941     gradient_logos = [
 942         'res/atrium-logo.png',
 943         'res/wl-logo-white.png',
 944         'res/fnp-logo-white.png',
 945     ]
 946
 947
 948 class BNCover(LogoWLCover):
 949     gradient_logos = [
 950         'res/dofinansowano.png',
 951         'res/MKIDN.jpg',
 952         'res/BN.png',
 953         'res/wl-logo-white.png',
 954     ]
 955 #    annotation = 'Zadanie „Udostępnienie publikacji w formatach cyfrowych” w ramach Narodowego Programu Rozwoju Czytelnictwa. Dofinansowano ze środków Ministra Kultury, Dziedzictwa Narodowego i Sportu.'
 956
 957
 958 class FactoryCover(LogoWLCover):
 959     gradient_logos = [
 960         'res/factory.jpg',
 961         'res/wl-logo-white.png',
 962     ]
 963     qrcode = 'pk_campaign=factory22'
 964     format = 'PNG'
 965
 966     def __init__(self, *args, **kwargs):
 967         kwargs.setdefault('bleed', 10)
 968         return super().__init__(*args, **kwargs)
 969
 970     def ext(self):
 971         return 'pdf'
 972
 973     def output_file(self, *args, **kwargs):
 974         imgfile = super().output_file(*args, **kwargs)
 975         import subprocess
 976         import tempfile
 977         with tempfile.TemporaryDirectory(prefix='factory-', dir='.') as d:
 978             import os
 979             import shutil
 980             with open(d + '/cover.png', 'wb') as f:
 981                 f.write(imgfile.get_bytes())
 982             shutil.copy(get_resource('res/factory-cover.svg'), d)
 983             subprocess.call(['inkscape', f'--export-filename={d}/cover.pdf', d + '/factory-cover.svg'])
 984             with open(d + '/cover.pdf', 'rb') as f:
 985                 return OutputFile.from_bytes(f.read())
 986
 987
 988
 989 from librarian.covers.marquise import MarquiseCover, LabelMarquiseCover
 990
 991 COVER_CLASSES = {
 992     'default': LogoWLCover,
 993     'kmlu': KMLUCover,
 994     'mpw': MPWCover,
 995     'atrium': AtriumCover,
 996     'bn': BNCover,
 997     'legimi': LegimiCover,
 998     'legimi-corner': LegimiCornerCover,
 999     'legimi-audiobook': LegimiAudiobookCover,
1000     'factory': FactoryCover,
1001     'm': MarquiseCover,
1002     'm-label': LabelMarquiseCover,
1003 }
1004
1005
1006 def make_cover(book_info, *args, **kwargs):
1007     if 'cover_class' in kwargs:
1008         cover_class_name = kwargs.pop('cover_class')
1009     else:
1010         cover_class_name = book_info.cover_class
1011     cover_class = COVER_CLASSES[cover_class_name]
1012     return cover_class(book_info, *args, **kwargs)