New covers.

[librarian.git] / src / librarian / covers / widgets / image.py

diff --git a/src/librarian/covers/widgets/image.py b/src/librarian/covers/widgets/image.py
new file mode 100644 (file)
index 0000000..39879d4
--- /dev/null
+++ b/src/librarian/covers/widgets/image.py
@@ -0,0 +1,29 @@
+import PIL.Image
+from librarian import get_resource
+from .base import Widget
+
+
+class ImageWidget(Widget):
+    def build(self, w, h):
+        img = PIL.Image.open(self.image_path)
+        img = img.resize((round(img.size[0] / img.size[1] * h), h))
+        return img
+    
+    
+class WLLogo(ImageWidget):
+    @property
+    def image_path(self):
+        if self.cover.color_scheme['text'] == '#fff':
+            return get_resource('res/cover/logo_WL_invert.png')
+        else:
+            return get_resource('res/cover/logo_WL.png')
+
+
+class Label(ImageWidget):
+    @property
+    def image_path(self):
+        if self.cover.is_very_bright:
+            return get_resource('res/cover/label_WLpolecaja.szary.png')
+        else:
+            return get_resource('res/cover/label_WLpolecaja.png')
+