X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/f318053fb3349c5364cfb866b2a3d33c2423e12a..f2f48f5d59a7a5d247786e4a5e19ed05c4f751dc:/librarian/cover.py diff --git a/librarian/cover.py b/librarian/cover.py index dfd451b..efb2974 100644 --- a/librarian/cover.py +++ b/librarian/cover.py @@ -7,7 +7,7 @@ import re from PIL import Image, ImageFont, ImageDraw, ImageFilter, ImageEnhance from StringIO import StringIO from librarian import get_resource, OutputFile, URLOpener - +from os import path class Metric(object): """Gets metrics from an object, scaling it by a factor.""" @@ -435,3 +435,27 @@ class GandalfCover(Cover): logo_bottom = 25 logo_width = 250 format = 'PNG' + +class FutureOfCopyrightCover(Cover): + width = 402 + height = 595 + background_img = 'cover.png' + format = 'PNG' + + def save(self, dest): + dest.write(open(self.background_img).read()) + + +class CoverFromFile(Cover): + def __init__(self, filename, *args, **kwargs): + super(CoverFromFile, self).__init__(*args, **kwargs) + + self.background_img = filename + + self.img = Image.open(filename) + (self.width, self.height) = self.img.size + + self.format = path.splitext(filename)[-1][1:].upper() + + def image(self): + return self.img