import json, os, sys, urllib.request, urllib.parse, urllib.error def _cfg(): s = json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) base = str(s.get("url") or "https://api.sekoia.io").rstrip("/") headers = {"Authorization": "Bearer " + s.get("api_key", ""), "Content-Type": "application/json", "Accept": "application/json"} return base, headers def _inputs(): return json.loads(os.environ.get("INTEGRATION_INPUTS", "{}")) def request(method, path, params=None): base, headers = _cfg() url = base + "/" + path.lstrip("/") if params: clean = {k: v for k, v in params.items() if v not in (None, "")} if clean: url += "?" + urllib.parse.urlencode(clean, doseq=True) req = urllib.request.Request(url, headers=headers, method=method) with urllib.request.urlopen(req, timeout=90) as r: raw = r.read() return json.loads(raw) if raw else {} def _range(value): # A bare date is the lower bound (",now"); a full range is passed through. v = str(value) return v if "," in v else (v + ",now") def run(): inp = _inputs() params = { "offset": inp.get("offset") or 0, "direction": inp.get("direction") or "asc", "sort": inp.get("sort_by") or "created_at", } if inp.get("limit"): params["limit"] = inp["limit"] if inp.get("status"): params["match[status_name]"] = inp["status"] if inp.get("created_at"): params["date[created_at]"] = _range(inp["created_at"]) if inp.get("updated_at"): params["date[updated_at]"] = _range(inp["updated_at"]) if inp.get("urgency"): params["range[urgency]"] = inp["urgency"] if inp.get("alerts_type"): params["match[type_value]"] = inp["alerts_type"] print(json.dumps(request("GET", "/v1/sic/alerts", params=params))) try: run() except urllib.error.HTTPError as e: print(json.dumps({"error": "HTTP " + str(e.code), "detail": e.read().decode("utf-8", "replace")})) sys.exit(1) except Exception as e: print(json.dumps({"error": str(e)})) sys.exit(1)