Minor EPUB/MOBI fixes.
[librarian.git] / 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)