Upload new_api_proxy.py

new_api_proxy.py

@@ -0,0 +1,34 @@
+import os
+
+
+NEW_API_PROXY_URL_ENV = "NEW_API_PROXY_URL"
+NEW_API_PROXY_TOKEN_ENV = "NEW_API_PROXY_TOKEN"
+NEW_API_PROXY_TOKEN_HEADER = "X-New-API-Proxy-Token"
+
+
+def get_new_api_proxy_base_url(proxy_url=None):
+    raw = proxy_url if proxy_url is not None else os.getenv(NEW_API_PROXY_URL_ENV, "")
+    raw = str(raw or "").strip().strip('"').strip("'")
+    if not raw:
+        return ""
+    if "://" not in raw:
+        raw = f"https://{raw}"
+    return raw.rstrip("/")
+
+
+def build_new_api_proxy_url(path, proxy_url=None):
+    proxy_base_url = get_new_api_proxy_base_url(proxy_url)
+    if not proxy_base_url:
+        return ""
+    if not path.startswith("/"):
+        path = f"/{path}"
+    return f"{proxy_base_url}{path}"
+
+
+def with_new_api_proxy_headers(headers=None, proxy_url=None):
+    headers = dict(headers or {})
+    if get_new_api_proxy_base_url(proxy_url):
+        proxy_token = os.getenv(NEW_API_PROXY_TOKEN_ENV, "").strip()
+        if proxy_token:
+            headers[NEW_API_PROXY_TOKEN_HEADER] = proxy_token
+    return headers