import json, os, sys, urllib.request, urllib.parse, urllib.error def _cfg(): s = json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) base = str(s.get("base_url") or "https://www.virustotal.com/api/v3").rstrip("/") headers = {"x-apikey": s.get("api_key", ""), "Accept": "application/json"} return base, headers def request(path): base, headers = _cfg() req = urllib.request.Request(base + "/" + path.lstrip("/"), headers=headers, method="GET") with urllib.request.urlopen(req, timeout=60) as r: raw = r.read() return json.loads(raw) if raw else {} def run(): request("ip_addresses/8.8.8.8") print(json.dumps({"ok": True})) try: run() except urllib.error.HTTPError as e: detail = e.read().decode("utf-8", "replace") msg = "API key is not valid." if e.code in (401, 403) else "HTTP " + str(e.code) print(json.dumps({"ok": False, "error": msg, "detail": detail})) sys.exit(1) except Exception as e: print(json.dumps({"ok": False, "error": str(e)})) sys.exit(1)