Add 404 Page when theme not found (#777)

getpelican · Jul 30, 2024 · ef37632 · ef37632
1 parent 74fb998
commit ef37632
Show file tree

Hide file tree

Showing 3 changed files with 49 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -19,4 +19,4 @@ Thumbs.db
 # Preview artifacts #
 #####################
 _pelican
-_output
+_output
diff --git a/build-theme-previews.py b/build-theme-previews.py
@@ -59,7 +59,7 @@ def install_translator(self, generator):
 <style>
 
 h1 {
-  margin 20px auto;
+  margin: 20px auto;
   text-align: center;
 }
 
@@ -134,6 +134,40 @@ def install_translator(self, generator):
 </html>
 """
 
+HTML_404 = """\
+<!DOCTYPE html>
+<html>
+<head>
+<title>Not Found</title>
+</head>
+<style>
+h1 {
+  margin: 20px auto;
+  text-align: center;
+}
+p {text-align: center;}
+h3 {text-align: center;}
+a {color: black;}
+.center {
+  display: block;
+  margin-left: auto;
+  margin-right: auto;
+  width: 50%;
+}
+</style>
+<body>
+
+<h1>Not Found</h1>
+<h3>What you’re looking for isn’t here. <br> It may have moved, or something unfortunate may have befallen it.</h3>
+
+<p>Double-check the URL and try again, or <a href="https://www.pelicanthemes.com">head home</a>.</p>
+
+<img src="pelican-sad.png" alt="sad pelican" class="center">
+
+</body>
+</html>
+"""
+
 
 def setup_folders(args):
     theme_root = os.path.abspath(os.path.dirname(__file__))
@@ -235,6 +269,17 @@ def write_index_files(output_root, success, fail):
     logger.info(f"failed {len(fail)} themes")
 
 
+def write_404_file(output_root):
+    logger.info("generating 404 page file...")
+    with open(os.path.join(output_root, "404.html"), "w") as outfile:
+        outfile.write(HTML_404)
+
+    pelican_pic_path = os.path.join(output_root, "pelican-sad.png")
+    subprocess.call(["cp", "pelican-sad.png", pelican_pic_path])
+
+    logger.info("wrote 404 page file")
+
+
 def parse_args(argv=None):
     parser = argparse.ArgumentParser()
     parser.add_argument(
@@ -273,7 +318,8 @@ def main(argv=None):
     theme_root, samples_root, output_root, screenshot_root = setup_folders(args)
     success, fail = build_theme_previews(theme_root, samples_root, output_root, screenshot_root)
     write_index_files(output_root, success, fail)
+    write_404_file(output_root)
 
 
 if __name__ == "__main__":
-    main()
+    main()
diff --git a/pelican-sad.png b/pelican-sad.png