import json, os, sys, urllib.parse, urllib.request, urllib.error def _cfg(): return json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) def _inputs(): return json.loads(os.environ.get("INTEGRATION_INPUTS", "{}")) def _base(cfg): return (str(cfg.get("base_url") or "https://api.us.onelogin.com")).rstrip("/") def _token(cfg): body = json.dumps({"grant_type": "client_credentials"}).encode("utf-8") auth = "client_id:" + str(cfg.get("client_id", "")) + ", client_secret:" + str(cfg.get("client_secret", "")) req = urllib.request.Request(_base(cfg) + "/auth/oauth2/v2/token", data=body, headers={"Authorization": auth, "Content-Type": "application/json", "Accept": "application/json"}, method="POST") with urllib.request.urlopen(req, timeout=60) as r: tok = json.loads(r.read()) at = tok.get("access_token") if not at: raise Exception("Token request failed: " + json.dumps(tok)) return at def request(method, path, cfg, token, body=None, params=None): url = _base(cfg) + path if params: clean = {k: v for k, v in params.items() if v not in (None, "")} if clean: url += "?" + urllib.parse.urlencode(clean) data = json.dumps(body).encode("utf-8") if body is not None else None headers = {"Authorization": "bearer " + token, "Accept": "application/json"} if data is not None: headers["Content-Type"] = "application/json" req = urllib.request.Request(url, data=data, headers=headers, method=method) with urllib.request.urlopen(req, timeout=60) as r: raw = r.read() return json.loads(raw) if raw else {} def _run(fn): try: cfg = _cfg() inputs = _inputs() token = _token(cfg) print(json.dumps(fn(cfg, token, inputs))) 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) def main(cfg, token, inputs): user_id = inputs.get("user_id") event_type_id = inputs.get("event_type_id") return request("GET", "/api/1/events", cfg, token, params={"user_id": user_id, "event_type_id": event_type_id}) _run(main)