65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
|
# /// script
|
||
|
|
# requires-python = ">=3.12"
|
||
|
|
# dependencies = ["requests<3"]
|
||
|
|
# ///
|
||
|
|
|
||
|
|
import json
|
||
|
|
import os
|
||
|
|
import requests
|
||
|
|
|
||
|
|
MANIFEST_URL = "https://null.formulatrix.dev/Contest/stitch-a-ton-answer/raw/branch/main/answer.json"
|
||
|
|
API_ENDPOINT_URL = os.getenv("CONTEST_HOST")+ os.getenv("CONTEST_ENDPOINT")
|
||
|
|
DOWNLOAD_DIR = os.getenv("CONTEST_OUTPUT")
|
||
|
|
|
||
|
|
def fetch_manifest(url):
|
||
|
|
print(f"fetching manifest from {url}...")
|
||
|
|
try:
|
||
|
|
response = requests.get(url)
|
||
|
|
response.raise_for_status()
|
||
|
|
return response.json()
|
||
|
|
except requests.exceptions.RequestException as e:
|
||
|
|
print(f" ERROR: Could not fetch manifest. Reason: {e}")
|
||
|
|
except json.JSONDecodeError:
|
||
|
|
print(" ERROR: Failed to parse JSON from manifest. The content may not be valid JSON.")
|
||
|
|
return None
|
||
|
|
|
||
|
|
def download_image(url, payload, filepath):
|
||
|
|
try:
|
||
|
|
# Use a session for potential connection pooling
|
||
|
|
with requests.Session() as s:
|
||
|
|
with s.post(url, json=payload, stream=True) as response:
|
||
|
|
response.raise_for_status()
|
||
|
|
|
||
|
|
# Write the content to the file in chunks
|
||
|
|
with open(filepath, 'wb') as f:
|
||
|
|
for chunk in response.iter_content(chunk_size=8192):
|
||
|
|
f.write(chunk)
|
||
|
|
|
||
|
|
print(f" SUCCESS: Saved '{os.path.basename(filepath)}'")
|
||
|
|
|
||
|
|
except requests.exceptions.HTTPError as e:
|
||
|
|
print(f" ERROR: HTTP error for '{os.path.basename(filepath)}': {e.response.status_code} {e.response.reason}")
|
||
|
|
except requests.exceptions.RequestException as e:
|
||
|
|
print(f" ERROR: Could not download '{os.path.basename(filepath)}'. Reason: {e}")
|
||
|
|
|
||
|
|
def main():
|
||
|
|
os.makedirs(DOWNLOAD_DIR, exist_ok=True)
|
||
|
|
|
||
|
|
manifest_data = fetch_manifest(MANIFEST_URL)
|
||
|
|
if not manifest_data:
|
||
|
|
return
|
||
|
|
|
||
|
|
for test_case in manifest_data.get("tests", []):
|
||
|
|
filename = test_case.get("name")
|
||
|
|
request_payload = test_case.get("request")
|
||
|
|
|
||
|
|
if not filename or not request_payload:
|
||
|
|
print(f"Skipping invalid test case: {test_case}")
|
||
|
|
continue
|
||
|
|
|
||
|
|
print(f"Requesting '{filename}'...")
|
||
|
|
output_filepath = os.path.join(DOWNLOAD_DIR, filename)
|
||
|
|
download_image(API_ENDPOINT_URL, request_payload, output_filepath)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|