fix: Add null check for headers

2024-10-08 19:03:02 +11:00 · 2024-10-08 19:03:02 +11:00 · 99b63592d0
commit 99b63592d0
parent 6c004d14bd
1 changed files with 16 additions and 15 deletions
--- a/server.py
+++ b/server.py
@ -359,21 +359,22 @@ def index():
            loaded = True

    # Check if crawler
-    if "Googlebot" not in request.headers.get(
-        "User-Agent"
-    ) and "Bingbot" not in request.headers.get("User-Agent"):
-        # Check if cookie is set
-        if not request.cookies.get("loaded") and not loaded:
-            # Set cookie
-            resp = make_response(
-                render_template("loading.html").replace(
-                    "https://nathan.woodburn.au/loading", "https://nathan.woodburn.au/"
-                ),
-                200,
-                {"Content-Type": "text/html"},
-            )
-            resp.set_cookie("loaded", "true", max_age=604800)
-            return resp
+    if request.headers:
+        if "Googlebot" not in request.headers.get(
+            "User-Agent"
+        ) and "Bingbot" not in request.headers.get("User-Agent"):
+            # Check if cookie is set
+            if not request.cookies.get("loaded") and not loaded:
+                # Set cookie
+                resp = make_response(
+                    render_template("loading.html").replace(
+                        "https://nathan.woodburn.au/loading", "https://nathan.woodburn.au/"
+                    ),
+                    200,
+                    {"Content-Type": "text/html"},
+                )
+                resp.set_cookie("loaded", "true", max_age=604800)
+                return resp

    global handshake_scripts
    global projects