buttons and extra metainfo on comic generation

1 files changed, 9 insertions, 2 deletions

diff --git a/modules/comic_generate.py b/modules/comic_generate.py
index 76705fa..ef539ba 100644
--- a/modules/comic_generate.py
+++ b/modules/comic_generate.py
@@ -14,6 +14,7 @@ IMAGE_FILE_NAME = "comic.png"
 TITLE_KEY = "title"
 DESCRIPTION_KEY = "description"
 PUBLISH_DATE_KEY = "publish_date"
+EXTRA_ATTR_KEY = "extra_attributes"
 
 
 class Issue:
@@ -39,6 +40,9 @@ class Issue:
     def get_link(self) -> str:
         return f"{Config.COMIC_ROOT_URL}/issues/{self.index}"
 
+    def extra_attributes(self) -> str:
+        return self.meta_data[EXTRA_ATTR_KEY] if EXTRA_ATTR_KEY in self.meta_data.keys() else ""
+
 
 def get_issues(path: str, local: bool) -> list[Issue]:
     return_list = []
@@ -68,13 +72,16 @@ def generate(jinja_env: jinja2.Environment, output_root_path: str, local: bool):
     issue_template = jinja_env.get_template("issue.html.j2")
     last_render = None
 
-    for issue in issues:
+    last_issue_idx = len(issues) - 1
+    for idx, issue in enumerate(issues):
         output_dir = os.path.join(output_root_path, "issues", issue.index)
         os.makedirs(output_dir, exist_ok=True)
 
         ctx = {
             "issue": issue,
-            "url": issue.get_link()
+            "url": issue.get_link(),
+            "previous": "/#" if idx == 0 else f"/issues/{str(idx)}",
+            "next": "/#" if idx == last_issue_idx else f"/issues/{str(idx + 2)}"
         }
 
         last_render = issue_template.render(ctx)