Add devtools & live reload

piggy/devtools.py

@@ -0,0 +1,23 @@
+from flask import Flask
+
+
+def inject_devtools(app: Flask):
+    # Inject livereload script into HTML responses
+    @app.after_request
+    def after_request(response):
+        if response.status_code != 200:
+            return response
+
+        mimetype = response.mimetype or ""
+        if not mimetype.startswith("text/html"):
+            return response
+
+        if not isinstance(response.response, list):
+            return response
+
+        script = '<script src="http://localhost:35729/livereload.js"></script>'
+        body = b"".join(response.response).decode()
+        body = body.replace("</body>", f"{script}</body>")
+        response.response = [body.encode("utf8")]
+        response.content_length = len(response.response[0])
+        return response

pyproject.toml

@@ -19,6 +19,7 @@ py-modules = []
 [project.optional-dependencies]
 dev = [
     "ruff",
+    "livereload",
 ]
 
 [tool.ruff]

run.py

@@ -22,13 +22,22 @@ def checkout_branch(branch):
 
 
 if __name__ == "__main__":
+    from livereload import Server
+    from piggy.devtools import inject_devtools
+
     # checkout_branch("test-output")
     os.environ["FLASK_DEBUG"] = "1"
     os.environ["USE_CACHE"] = "0"
+
     app = create_app()
-    app.config["TEMPLATES_AUTO_RELOAD"] = True
+    inject_devtools(app)
+
     run_tailwind(reload=True)
-    app.run(port=5001)
+
+    server = Server(app)
+    server.watch("piggy/**/*", ignore=lambda *_: False)
+    server.watch("piggybank/*", ignore=lambda *_: False)
+    server.serve(host="127.0.0.1", port=5001, debug=False)
 else:
     checkout_branch("output")
     run_tailwind()