Load a different cover template for different paper sizes

1 files changed, 25 insertions, 2 deletions

diff --git a/planner_generator.py b/planner_generator.py
index 66f5890..ea802e4 100755
--- a/planner_generator.py
+++ b/planner_generator.py
@@ -38,6 +38,7 @@ class Generator:
             self,
             year: Optional[int] = None,
             template: Optional[str] = None,
+            cover_template: Optional[str] = None,
             out_file: Optional[str] = None,
             build_dir: Optional[str] = "build",
             latitude: Optional[float] = None,
@@ -52,7 +53,7 @@ class Generator:
         self.longitude = longitude
         self.timezone = timezone
 
-        self.out_file = out_file or self.default_template + ".pdf"
+        self.out_file = out_file or (template or self.default_template) + ".pdf"
 
         env = jinja2.Environment()
         self.templates_dir = "templates"
@@ -61,13 +62,29 @@ class Generator:
             template = self.default_template
         self.template_recto = loader.load(env, template + "-r.svg")
         self.template_verso = loader.load(env, template + "-v.svg")
-        self.template_cover = loader.load(env, "cover-A6-r.svg")
+        self.template_cover = loader.load(
+            env,
+            self._get_cover_name(template, cover_template)
+        )
         self.build_dir = build_dir
         self.page_fname = os.path.join(
             self.build_dir,
             template + "-{year}-{page:03}.svg"
         )
 
+    def _get_cover_name(self, template, cover_template):
+        if cover_template:
+            return cover_template
+        if "A6" in template:
+            return "cover-A6-r.svg"
+        if "A5" in template:
+            return "cover-A5-r.svg"
+        if "A4" in template:
+            return "cover-A4-r.svg"
+        return "cover-A6-r.svg"
+
+
+
     def render_page(self, page: int, **kw):
         # page counts starts with 0
         if page == 0:
@@ -314,6 +331,11 @@ class Command:
             help="Base name of the template (without -[rv].svg)",
         )
         parser.add_argument(
+            "--cover-template",
+            default=None,
+            help="Full name of the template (including -[rv].svg)",
+        )
+        parser.add_argument(
             "--latitude",
             default=None,
             type=float,
@@ -350,6 +372,7 @@ class Command:
             generator(
                 year=self.args.year,
                 template=self.args.template,
+                cover_template=self.args.cover_template,
                 latitude=self.args.latitude,
                 longitude=self.args.longitude,
                 timezone=self.args.timezone,