x/themblem
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add timeouts to requests to prevent hanging

Browse Source
This commit is contained in:
Fam Zheng 2025-12-27 17:11:14 +00:00
parent 59425ca9fb
commit 73529bec16
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
emblem5/ai/fetch-scans.py
View File

@ -48,7 +48,7 @@ class ScanDataFetcher(object):
local_scan_data = self.load_local_scan_data() local_scan_data = self.load_local_scan_data()
logger.info(f'local_scan_data: {len(local_scan_data)}') logger.info(f'local_scan_data: {len(local_scan_data)}')
url = 'https://themblem.com/api/v1/scan-data-labels/' url = 'https://themblem.com/api/v1/scan-data-labels/'
r = requests.get(url, headers=self.make_headers()) r = requests.get(url, headers=self.make_headers(), timeout=30)
data = r.json() data = r.json()
fetch_backlog = [] fetch_backlog = []
scan_ids_set = set(scan_ids) if scan_ids else None scan_ids_set = set(scan_ids) if scan_ids else None
@ -104,7 +104,7 @@ class ScanDataFetcher(object):
frame_img_url = f'https://themblem.com/api/v1/oss-image/?token={self.token}&name={quote(scan["image"])}' frame_img_url = f'https://themblem.com/api/v1/oss-image/?token={self.token}&name={quote(scan["image"])}'
frame_img_file = os.path.join(scan_dir, 'frame.jpg') frame_img_file = os.path.join(scan_dir, 'frame.jpg')
if not os.path.exists(frame_img_file): if not os.path.exists(frame_img_file):
r = requests.get(frame_img_url) r = requests.get(frame_img_url, timeout=30)
r.raise_for_status() # Raise an exception for bad status codes r.raise_for_status() # Raise an exception for bad status codes
with open(frame_img_file, 'wb') as f: with open(frame_img_file, 'wb') as f:
f.write(r.content) f.write(r.content)