Upload 2 files

app.py

@@ -25,6 +25,7 @@ from pathlib import Path
 from cinema_api_client import fetch_canonical_movie_names as shared_fetch_canonical_movie_names
 from historical_sessions import load_history_df, summarize_total_box_office_by_movies
 from r2_config_sync import ensure_settings_file_from_r2, ensure_settings_file_synced, upload_settings_file_to_r2
+from tms_proxy import build_tms_url, get_tms_proxy_base_url, tms_verify_ssl, with_tms_proxy_headers
 # --- 新增：加载环境变量 ---
 from dotenv import load_dotenv
 load_dotenv()  # 加载本地 .env 文件
@@ -2253,6 +2254,7 @@ def display_pdf(base64_pdf):
 @st.cache_data(show_spinner=False, ttl=600)
 def fetch_and_process_server_movies(priority_movie_titles=None):
     if priority_movie_titles is None: priority_movie_titles = []
+    tms_proxy_base_url = get_tms_proxy_base_url()
 
     # 获取环境变量
     app_secret = os.getenv("TMS_APP_SECRET")
@@ -2289,7 +2291,11 @@ def fetch_and_process_server_movies(priority_movie_titles=None):
     # 使用变量
     token_json_data = {'appId': 'hd', 'appSecret': app_secret, 'timeStamp': int(time.time() * 1000)}
     # 动态构建 URL
-    token_url = f'https://tms.hengdianfilm.com/cinema-api/admin/generateToken?token=hd&murl=?token=hd&murl=ticket={ticket}'
+    token_url = build_tms_url(
+        f'https://tms.hengdianfilm.com/cinema-api/admin/generateToken?token=hd&murl=?token=hd&murl=ticket={ticket}',
+        tms_proxy_base_url,
+    )
+    token_headers = with_tms_proxy_headers(token_headers, tms_proxy_base_url)
 
     response = requests.post(token_url, headers=token_headers, json=token_json_data, timeout=10)
 
@@ -2324,8 +2330,18 @@ def fetch_and_process_server_movies(priority_movie_titles=None):
         list_json_data = {'THEATER_ID': theater_id, 'SOURCE': 'SERVER', 'ASSERT_TYPE': 2, 'PAGE_CAPACITY': 20,
                           'PAGE_INDEX': page_index}
 
-        list_url = 'https://tms.hengdianfilm.com/cinema-api/cinema/server/dcp/list'
-        response = requests.post(list_url, params=list_params, headers=list_headers, json=list_json_data, verify=False)
+        list_url = build_tms_url(
+            'https://tms.hengdianfilm.com/cinema-api/cinema/server/dcp/list',
+            tms_proxy_base_url,
+        )
+        list_headers = with_tms_proxy_headers(list_headers, tms_proxy_base_url)
+        response = requests.post(
+            list_url,
+            params=list_params,
+            headers=list_headers,
+            json=list_json_data,
+            verify=tms_verify_ssl(default=False, proxy_url=tms_proxy_base_url),
+        )
         response.raise_for_status()
         movie_data = response.json()
         if movie_data.get("RSPCD") != "000000": raise Exception(f"获取影片列表失败: {movie_data.get('RSPMSG')}")

tms_proxy.py

@@ -0,0 +1,47 @@
+import os
+from urllib.parse import urlparse
+
+
+TMS_ORIGIN = "https://tms.hengdianfilm.com"
+TMS_PROXY_URL_ENV = "TMS_CF_WORKER_URL"
+TMS_PROXY_TOKEN_ENV = "TMS_CF_WORKER_TOKEN"
+TMS_PROXY_TOKEN_HEADER = "X-TMS-Proxy-Token"
+
+
+def get_tms_proxy_base_url(proxy_url=None):
+    raw = proxy_url if proxy_url is not None else os.getenv(TMS_PROXY_URL_ENV, "")
+    raw = str(raw or "").strip()
+    if not raw:
+        return ""
+    if "://" not in raw:
+        raw = f"https://{raw}"
+    return raw.rstrip("/")
+
+
+def build_tms_url(origin_url, proxy_url=None):
+    proxy_base_url = get_tms_proxy_base_url(proxy_url)
+    if not proxy_base_url:
+        return origin_url
+
+    parsed = urlparse(origin_url)
+    origin = f"{parsed.scheme}://{parsed.netloc}"
+    if origin != TMS_ORIGIN:
+        return origin_url
+
+    query = f"?{parsed.query}" if parsed.query else ""
+    return f"{proxy_base_url}{parsed.path}{query}"
+
+
+def with_tms_proxy_headers(headers=None, proxy_url=None):
+    headers = dict(headers or {})
+    if get_tms_proxy_base_url(proxy_url):
+        proxy_token = os.getenv(TMS_PROXY_TOKEN_ENV, "").strip()
+        if proxy_token:
+            headers[TMS_PROXY_TOKEN_HEADER] = proxy_token
+    return headers
+
+
+def tms_verify_ssl(default=False, proxy_url=None):
+    if get_tms_proxy_base_url(proxy_url):
+        return True
+    return default