feat: serve 404 page locally (#160)

Resolves #159 Signed-off-by: Mike Fiedler <[email protected]>
pypi · Mar 28, 2024 · d645a31 · d645a31
1 parent 5756f29
commit d645a31
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/inspector/main.py b/inspector/main.py
@@ -64,6 +64,9 @@ def versions(project_name):
     resp = requests.get(f"https://pypi.org/pypi/{project_name}/json")
     pypi_project_url = f"https://pypi.org/project/{project_name}"
 
+    # Self-host 404 page to mitigate iframe embeds
+    if resp.status_code == 404:
+        return render_template("404.html")
     if resp.status_code != 200:
         return redirect(pypi_project_url, 307)
 

diff --git a/inspector/templates/404.html b/inspector/templates/404.html
@@ -0,0 +1,5 @@
+{% extends 'base.html' %}
+
+{% block body %}
+<h1>NOT FOUND</h1>
+{% endblock %}