-# -*- coding: utf-8 -*-
-#
# This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
-# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+# Copyright © Fundacja Wolne Lektury. See NOTICE for more information.
#
-from __future__ import unicode_literals
-
import re
from PIL import Image, ImageFont, ImageDraw, ImageFilter
-from six import BytesIO
+from io import BytesIO
from librarian import get_resource, OutputFile, URLOpener
-class Metric(object):
+class Metric:
"""Gets metrics from an object, scaling it by a factor."""
def __init__(self, obj, scale):
self._obj = obj
return src
-class TextBox(object):
+class TextBox:
"""Creates an Image with a series of centered strings."""
SHADOW_X = 3
break
line = parts[0]
- if line[-2] == ' ':
+ if len(line) > 2 and line[-2] == ' ':
line = line[:-2]
line_width = self.draw.textsize(line, font=font)[0]
return image
-class Cover(object):
+class Cover:
"""Abstract base class for cover images generator."""
width = 600
height = 800
}
def __init__(self, book_info, format=None, width=None, height=None, cover_logo=None):
+ self.book_info = book_info
+
+ self.predesigned = False
+ if book_info.cover_class == 'image':
+ self.predesigned = True
+
+ # TODO: deprecated
+ if book_info.cover_box_position == 'none':
+ self.predesigned = True
+
self.authors = [auth.readable() for auth in book_info.authors]
self.title = book_info.title
if format is not None:
self.format = format
+ self.set_size(width, height)
+ self.cover_logo = cover_logo
+
+ def set_size(self, width, height):
if width and height:
self.height = int(round(height * self.width / width))
scale = max(float(width or 0) / self.width,
def pretty_authors(self):
"""Allows for decorating authors' names."""
- return [self.authors]
+ return self.authors
def pretty_title(self):
"""Allows for decorating title."""
author_font = ImageFont.truetype(
self.author_font_ttf, metr.author_font_size,
- layout_engine=ImageFont.LAYOUT_BASIC)
+ layout_engine=ImageFont.Layout.BASIC)
for pa in self.pretty_authors():
tbox.text(pa, self.author_color, author_font, metr.author_lineskip,
self.author_shadow)
)
title_font = ImageFont.truetype(
self.title_font_ttf, metr.title_font_size,
- layout_engine=ImageFont.LAYOUT_BASIC)
+ layout_engine=ImageFont.Layout.BASIC)
tbox.text(self.pretty_title(), self.title_color, title_font,
metr.title_lineskip, self.title_shadow)
text_img = tbox.image()
img = img.resize((
int(round(img.size[0] * self.scale_after)),
int(round(img.size[1] * self.scale_after))),
- Image.ANTIALIAS)
+ Image.Resampling.LANCZOS)
return img
def mime_type(self):
background_color = '#444'
author_color = '#444'
background_img = get_resource('res/cover.png')
+ background_top = False
format = 'JPEG'
epoch_colors = {
- u'Starożytność': '#9e3610',
- u'Średniowiecze': '#564c09',
- u'Renesans': '#8ca629',
- u'Barok': '#a6820a',
- u'Oświecenie': '#f2802e',
- u'Romantyzm': '#db4b16',
- u'Pozytywizm': '#961060',
- u'Modernizm': '#7784e0',
- u'Dwudziestolecie międzywojenne': '#3044cf',
- u'Współczesność': '#06393d',
+ 'Starożytność': '#9e3610',
+ 'Średniowiecze': '#564c09',
+ 'Renesans': '#8ca629',
+ 'Barok': '#a6820a',
+ 'Oświecenie': '#f2802e',
+ 'Romantyzm': '#db4b16',
+ 'Pozytywizm': '#961060',
+ 'Modernizm': '#7784e0',
+ 'Dwudziestolecie międzywojenne': '#3044cf',
+ 'Współczesność': '#06393d',
}
set_title_color = True
kind_box_position = {
- u'Liryka': 'top',
- u'Epika': 'bottom',
+ 'Liryka': 'top',
+ 'Epika': 'bottom',
}
def __init__(self, book_info, format=None, width=None, height=None,
bleed=0, cover_logo=None):
super(WLCover, self).__init__(book_info, format=format, width=width,
height=height)
+
+ self.slug = book_info.url.slug
# Set box position.
self.box_position = book_info.cover_box_position or \
self.kind_box_position.get(book_info.kind, self.box_position)
self.bar_width = 0
else:
self.bar_color = book_info.cover_bar_color or \
- self.epoch_colors.get(book_info.epoch, self.bar_color)
+ self.get_variable_color(book_info) or self.bar_color
# Set title color.
if self.set_title_color:
- self.title_color = self.epoch_colors.get(book_info.epoch,
- self.title_color)
+ self.title_color = self.get_variable_color(book_info) or self.title_color
self.bleed = bleed
self.box_top_margin += bleed
if book_info.cover_url:
url = book_info.cover_url
bg_src = None
- if bg_src is None:
- bg_src = URLOpener().open(url)
- self.background_img = BytesIO(bg_src.read())
- bg_src.close()
+ while True:
+ try:
+ if bg_src is None:
+ import requests
+ bg_src = requests.get(url, timeout=5)
+ self.background_img = BytesIO(bg_src.content)
+ bg_src.close()
+ except Exception as e:
+ bg_src = None
+ print(e)
+ import time
+ time.sleep(1)
+ else:
+ break
+
+ def get_variable_color(self, book_info):
+ return self.epoch_colors.get(book_info.epoch, None)
def pretty_authors(self):
return [a.upper() for a in self.authors]
metr = Metric(self, self.scale)
# Write author name.
- box = TextBox(metr.title_box_width, metr.height,
+ box = TextBox(metr.title_box_width - 2 * self.bleed, metr.height,
padding_y=metr.box_padding_y,
padding_x=metr.box_padding_x,
bar_width=metr.box_bar_width,
)
author_font = ImageFont.truetype(
self.author_font_ttf, metr.author_font_size,
- layout_engine=ImageFont.LAYOUT_BASIC)
+ layout_engine=ImageFont.Layout.BASIC)
for pa in self.pretty_authors():
box.text(pa, font=author_font, line_height=metr.author_lineskip,
color=self.author_color, shadow_color=self.author_shadow,
# Write title.
title_font = ImageFont.truetype(
self.title_font_ttf, metr.title_font_size,
- layout_engine=ImageFont.LAYOUT_BASIC)
+ layout_engine=ImageFont.Layout.BASIC)
box.text(self.pretty_title(),
line_height=metr.title_lineskip,
font=title_font,
box_top = (metr.height - box_img.size[1]) // 2
box_left = metr.bar_width + (
- metr.width - metr.bar_width - box_img.size[0]
+ metr.width - metr.bar_width - box_img.size[0] - self.bleed
) // 2
# Draw the white box.
trg_size[0],
int(round(src.size[1] * trg_size[0] / src.size[0]))
)
- cut = (resized[1] - trg_size[1]) // 2
- src = src.resize(resized, Image.ANTIALIAS)
+ if self.background_top:
+ cut = 0
+ else:
+ cut = (resized[1] - trg_size[1]) // 2
+ src = src.resize(resized, Image.Resampling.LANCZOS)
src = src.crop((0, cut, src.size[0], src.size[1] - cut))
else:
resized = (
trg_size[1],
)
cut = (resized[0] - trg_size[0]) // 2
- src = src.resize(resized, Image.ANTIALIAS)
+ src = src.resize(resized, Image.Resampling.LANCZOS)
src = src.crop((cut, 0, src.size[0] - cut, src.size[1]))
img.paste(src, (metr.bar_width, 0))
logos_right = True
gradient_logo_centering = False
+ qrcode = None
+
def __init__(self, book_info, *args, cover_logo=None, **kwargs):
super(LogoWLCover, self).__init__(book_info, *args, **kwargs)
]
self.end_cover_logos = []
if cover_logo:
- self.end_cover_logos.append(
+ self.additional_cover_logos.append(
open(cover_logo, 'rb')
)
@property
def has_gradient_logos(self):
- return self.gradient_logos or self.additional_cover_logos or self.end_cover_logos or self.annotation
+ return self.gradient_logos or self.additional_cover_logos or self.end_cover_logos or self.annotation or self.qrcode is not None
def add_gradient_logos(self, img, metr):
gradient = Image.new(
logo_top = int(
metr.height
- metr.gradient_height / 2
- - metr.gradient_logo_height / 2 - metr.bleed / 2
+ - metr.gradient_logo_height / 2 - metr.bleed
)
logos = [
logos = self.additional_cover_logos + logos + self.end_cover_logos
- if self.logos_right:
- logos.reverse()
-
logos = [
Image.open(logo_bytes).convert('RGBA')
for logo_bytes in logos
]
+ if self.qrcode is not None:
+ import qrcode
+ logos.append(qrcode.make(f'https://wolnelektury.pl/katalog/lektura/{self.slug}/?{self.qrcode}'))
+
+ if self.logos_right:
+ logos.reverse()
+
# See if logos fit into the gradient. If not, scale down accordingly.
space_for_logos = (
metr.width
- metr.bar_width
- - 2 * metr.gradient_logo_margin_right
+ - 2 * metr.gradient_logo_margin_right + self.bleed
)
widths = [
min(
logo.size[1] * widths[i] / logo.size[0] * L_scale
))
),
- Image.ANTIALIAS)
+ Image.Resampling.LANCZOS)
if self.logos_right:
cursor -= logo.size[0]
draw = ImageDraw.Draw(img2)
author_font = ImageFont.truetype(
self.author_font_ttf, metr.annotation_height,
- layout_engine=ImageFont.LAYOUT_BASIC)
+ layout_engine=ImageFont.Layout.BASIC)
draw.text((self.annotation_height, self.annotation_height), self.annotation, font=author_font, fill='#FFFFFF')
img2.show()
img2 = img2.rotate(90)
]
logos_right = False
gradient_logo_centering = True
+ background_top = True
+
+ genre_colors = {
+ 'Artykuł': '#bf001a',
+ 'Artykuł naukowy': '#bf001a',
+ 'Dziennik': '#bf001a',
+ 'Esej': '#bf001a',
+ 'Felieton': '#bf001a',
+ 'Kronika': '#bf001a',
+ 'List': '#bf001a',
+ 'Manifest': '#bf001a',
+ 'Odczyt': '#bf001a',
+ 'Pamiętnik': '#bf001a',
+ 'Poradnik': '#bf001a',
+ 'Praca naukowa': '#bf001a',
+ 'Publicystyka': '#bf001a',
+ 'Reportaż': '#bf001a',
+ 'Reportaż podróżniczy': '#bf001a',
+ 'Rozprawa': '#bf001a',
+ 'Rozprawa polityczna': '#bf001a',
+ 'Traktat': '#bf001a',
+ }
+ kind_colors = {
+ 'Epika': '#9bbb2b',
+ 'Liryka': '#3e626f',
+ 'Dramat': '#ecbe00',
+ }
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
if self.has_gradient_logos:
self.box_bottom_margin += self.box_bottom_margin_logos_add
-
-
+
def pretty_authors(self):
return self.authors
+ def get_variable_color(self, book_info):
+ for genre in book_info.genres:
+ if genre in self.genre_colors:
+ return self.genre_colors[genre]
+ for kind in book_info.kinds:
+ if kind in self.kind_colors:
+ return self.kind_colors[kind]
class LegimiCornerCover(LegimiCover):
gradient_logos = []
title_font_size = 50
def pretty_title(self):
- return u"„%s”" % self.title
+ return "„%s”" % self.title
class BookotekaCover(Cover):
# annotation = 'Zadanie „Udostępnienie publikacji w formatach cyfrowych” w ramach Narodowego Programu Rozwoju Czytelnictwa. Dofinansowano ze środków Ministra Kultury, Dziedzictwa Narodowego i Sportu.'
+class FactoryCover(LogoWLCover):
+ gradient_logos = [
+ 'res/factory.jpg',
+ 'res/wl-logo-white.png',
+ ]
+ qrcode = 'pk_campaign=factory22'
+ format = 'PNG'
+
+ def __init__(self, *args, **kwargs):
+ kwargs.setdefault('bleed', 10)
+ return super().__init__(*args, **kwargs)
+
+ def ext(self):
+ return 'pdf'
+
+ def output_file(self, *args, **kwargs):
+ imgfile = super().output_file(*args, **kwargs)
+ import subprocess
+ import tempfile
+ with tempfile.TemporaryDirectory(prefix='factory-', dir='.') as d:
+ import os
+ import shutil
+ with open(d + '/cover.png', 'wb') as f:
+ f.write(imgfile.get_bytes())
+ shutil.copy(get_resource('res/factory-cover.svg'), d)
+ subprocess.call(['inkscape', f'--export-filename={d}/cover.pdf', d + '/factory-cover.svg'])
+ with open(d + '/cover.pdf', 'rb') as f:
+ return OutputFile.from_bytes(f.read())
+
+
+
+from librarian.covers.marquise import MarquiseCover, LabelMarquiseCover
+
COVER_CLASSES = {
- 'default': LogoWLCover,
+ 'legacy': LogoWLCover,
'kmlu': KMLUCover,
'mpw': MPWCover,
'atrium': AtriumCover,
'legimi': LegimiCover,
'legimi-corner': LegimiCornerCover,
'legimi-audiobook': LegimiAudiobookCover,
+ 'factory': FactoryCover,
+ 'm': MarquiseCover,
+ 'm-label': LabelMarquiseCover,
+ 'default': MarquiseCover,
}
def make_cover(book_info, *args, **kwargs):
+ cover_class_name = None
if 'cover_class' in kwargs:
cover_class_name = kwargs.pop('cover_class')
- else:
- cover_class_name = book_info.cover_class
+ if not cover_class_name:
+ cover_class_name = 'default'
cover_class = COVER_CLASSES[cover_class_name]
return cover_class(book_info, *args, **kwargs)
fnp / librarian.git / blobdiff