X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/0f3e12398ddbbb07a71eeafb3f960631777cabdc..637475880b35561e349a4a5df3048256cb58fb76:/librarian/cover.py diff --git a/librarian/cover.py b/librarian/cover.py index ff01841..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.""" @@ -437,10 +437,25 @@ class GandalfCover(Cover): format = 'PNG' class FutureOfCopyrightCover(Cover): - width = 420 + 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