21 lines
607 B
Python
21 lines
607 B
Python
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
|