Add server header

BattlesnakeOfficial · Aug 22, 2022 · cbadaee · cbadaee
1 parent 36d8864
commit cbadaee
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/server.py b/server.py
@@ -7,7 +7,7 @@
 
 
 def run_server(handlers: typing.Dict):
-    app = Flask("My Battlesnake")
+    app = Flask("Battlesnake")
 
     @app.get("/")
     def on_info():
@@ -30,6 +30,13 @@ def on_end():
         handlers["end"](game_state)
         return "ok"
 
+    @app.after_request
+    def identify_server(response):
+        response.headers.set(
+            "server", "battlesnake/github/starter-snake-python"
+        )
+        return response
+
     host = "0.0.0.0"
     port = int(os.environ.get("PORT", "8000"))