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