Nix Bot
2a6e2d192b
All checks were successful
EcoArk Wheel Pipeline CI / build (push) Successful in 2m36s
126 lines
4.3 KiB
Python
126 lines
4.3 KiB
Python
#!/usr/bin/env python3
|
|
"""Upload artifact to Gitea Actions runtime, falling back to local save.
|
|
|
|
Usage:
|
|
upload_artifacts.py <artifact_name> <file_or_glob> [<file_or_glob>...]
|
|
"""
|
|
|
|
import glob
|
|
import json
|
|
import os
|
|
import sys
|
|
import urllib.request
|
|
import urllib.error
|
|
|
|
HERE = os.path.dirname(os.path.abspath(__file__))
|
|
ARTIFACTS_DIR = os.path.join(os.path.dirname(HERE), "artifacts")
|
|
|
|
|
|
def _upload_via_runtime(artifact_name, payload, runtime_url, runtime_token):
|
|
"""Upload via Gitea Actions runtime API."""
|
|
url = f"{runtime_url}/_apis/artifact/build/{artifact_name}/{artifact_name}/upload"
|
|
headers = {
|
|
"Authorization": f"Bearer {runtime_token}",
|
|
"Content-Type": "application/octet-stream",
|
|
"Accept": "*/*",
|
|
}
|
|
req = urllib.request.Request(url, data=payload, headers=headers, method="PUT")
|
|
with urllib.request.urlopen(req) as resp:
|
|
return resp.status in (200, 201, 204)
|
|
|
|
|
|
def _save_locally(artifact_name, payload):
|
|
"""Fallback: save to local filesystem."""
|
|
os.makedirs(ARTIFACTS_DIR, exist_ok=True)
|
|
dest = os.path.join(ARTIFACTS_DIR, f"{artifact_name}.zip")
|
|
with open(dest, "wb") as f:
|
|
f.write(payload)
|
|
return dest
|
|
|
|
|
|
def _resolve_files(patterns):
|
|
"""Resolve glob patterns into actual file paths."""
|
|
files = []
|
|
for p in patterns:
|
|
expanded = glob.glob(p)
|
|
if expanded:
|
|
files.extend(expanded)
|
|
else:
|
|
# Maybe it's a literal path
|
|
files.append(p)
|
|
return files
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) < 3:
|
|
print(f"Usage: {sys.argv[0]} <artifact_name> <file_or_glob> [<file_or_glob>...]")
|
|
sys.exit(1)
|
|
|
|
artifact_name = sys.argv[1]
|
|
file_or_globs = sys.argv[2:]
|
|
|
|
# Resolve all files
|
|
files = _resolve_files(file_or_globs)
|
|
|
|
# Check that at least one file exists
|
|
missing = [f for f in files if not os.path.isfile(f)]
|
|
for f in missing:
|
|
print(f"[upload-artifact] WARNING: file not found (skipped): {f}")
|
|
|
|
existing = [f for f in files if os.path.isfile(f)]
|
|
if not existing:
|
|
print(f"[upload-artifact] ERROR: no valid files found in: {' '.join(file_or_globs)}")
|
|
sys.exit(1)
|
|
|
|
# For individual upload: save each file separately
|
|
if "wheels-individual" in artifact_name:
|
|
for f in existing:
|
|
name = os.path.basename(f)
|
|
individual_name = f"{artifact_name}-{name}"
|
|
with open(f, "rb") as fh:
|
|
payload = fh.read()
|
|
runtime_url = os.environ.get("ACTIONS_RUNTIME_URL", "")
|
|
runtime_token = os.environ.get("ACTIONS_RUNTIME_TOKEN", "")
|
|
if runtime_url and runtime_token:
|
|
try:
|
|
_upload_via_runtime(individual_name, payload, runtime_url, runtime_token)
|
|
print(f"[upload-artifact] Uploaded {f} ({len(payload)} bytes)")
|
|
continue
|
|
except Exception as e:
|
|
print(f"[upload-artifact] Runtime upload failed for {f}: {e}")
|
|
dest = _save_locally(individual_name, payload)
|
|
print(f"[upload-artifact] Saved {f} to {dest} ({len(payload)} bytes)")
|
|
print(f"[upload-artifact] Done — {len(existing)} file(s) uploaded.")
|
|
return
|
|
|
|
# Bundle upload: read first/only file
|
|
file_path = existing[0]
|
|
if len(existing) > 1:
|
|
print(f"[upload-artifact] WARNING: bundle upload uses first file only: {existing[0]}")
|
|
|
|
with open(file_path, "rb") as f:
|
|
payload = f.read()
|
|
|
|
runtime_url = os.environ.get("ACTIONS_RUNTIME_URL", "")
|
|
runtime_token = os.environ.get("ACTIONS_RUNTIME_TOKEN", "")
|
|
|
|
if runtime_url and runtime_token:
|
|
try:
|
|
ok = _upload_via_runtime(artifact_name, payload, runtime_url, runtime_token)
|
|
if ok:
|
|
print(f"[upload-artifact] Uploaded {file_path} ({len(payload)} bytes)")
|
|
return
|
|
except urllib.error.URLError as e:
|
|
print(f"[upload-artifact] Runtime upload failed (URLError): {e.reason}")
|
|
except Exception as e:
|
|
print(f"[upload-artifact] Runtime upload failed: {e}")
|
|
|
|
# Fallback
|
|
dest = _save_locally(artifact_name, payload)
|
|
print(f"[upload-artifact] Saved artifact to {dest} ({len(payload)} bytes)")
|
|
print("[upload-artifact] Done — 1 file(s) in '{}'.".format(artifact_name))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|