minor style fixes to EPUBs

[librarian.git] / librarian / cover.py

diff --git a/librarian/cover.py b/librarian/cover.py
index dfd451b..efb2974 100644 (file)
--- 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