add timeout

app.py

@@ -7,6 +7,9 @@ from datetime import datetime
 from zoneinfo import ZoneInfo
 from tzlocal import get_localzone
 
+TIMEOUT = 10
+API = os.getenv("api_bili")
+
 
 def timestamp(naive_time: datetime = None, target_tz=ZoneInfo("Asia/Shanghai")):
     if not naive_time:
@@ -23,7 +26,7 @@ def download_file(url, video_id, cache_dir="./__pycache__"):
 
     os.makedirs(cache_dir)
     local_file = f"{cache_dir}/{video_id}.mp4"
-    response = requests.get(url, stream=True)
+    response = requests.get(url, stream=True, timeout=TIMEOUT)
     if response.status_code == 200:
         with open(local_file, "wb") as file:
             for chunk in response.iter_content(chunk_size=8192):
@@ -54,7 +57,7 @@ def get_real_url(short_url):
             "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/53.0.2785.116 Safari/537.36"
         },
         allow_redirects=True,
-        timeout=10,
+        timeout=TIMEOUT,
     ).url.split("/?")[0]
 
 
@@ -69,7 +72,7 @@ def infer(video_url: str, p: int):
         if "b23.tv" in video_url:
             video_url = get_real_url(video_url)
 
-        response = requests.get(os.getenv("api_bili"), params={"url": video_url})
+        response = requests.get(API, params={"url": video_url}, timeout=TIMEOUT)
         response_json = response.json()
         retcode = response_json["code"]
         if retcode == 1: