Initial commit

core/middleware/prod_middleware.py

@@ -0,0 +1,20 @@
+from fastapi.requests import Request
+
+from config import conf
+from core.app import app
+
+
+@app.middleware("http")
+async def prod_middleware(request: Request, call_next):
+    if (
+            request.url.hostname == "scm.haco.dc"
+            or request.url.hostname == "scm.hacoupian.net"
+            or request.url.hostname == "rc.hacoupian.net"
+    ):
+        conf.server.is_prod = True
+    else:
+        conf.server.is_prod = False
+    response = await call_next(request)
+    response.headers["is-host"] = request.url.hostname
+    response.headers["is-prod"] = str(conf.server.is_prod)
+    return response