Adding new endpoints

Author: dacoburn

example-socket-export.py

@@ -0,0 +1,54 @@
+import json
+import os
+import logging
+from socketdev import socketdev
+logging.basicConfig(level=logging.INFO)
+
+
+sdk = socketdev(token=os.getenv("SOCKET_SECURITY_API_KEY"))
+
+orgs = sdk.org.get()
+if len(orgs) > 0:
+    org_id = None
+    org_slug = None
+    for org_key in orgs['organizations']:
+        org = orgs['organizations'][org_key]
+        org_id = org_key
+        org_slug = org['slug']
+else:
+    print("Something went wrong with getting org info")
+    exit(1)
+per_page = 100
+response = sdk.repos.get(org_slug, per_page=per_page)
+next_page = response.get("nextPage")
+repos = response.get("results")
+while next_page is not None and next_page != 0:
+    response = sdk.repos.get(org_slug, per_page=per_page, page=next_page)
+    next_page = response.get("nextPage")
+    repos.extend(response.get("results"))
+    if len(response.get("results", [])) == 0:
+        break
+
+# repos = repos[:20]
+head_full_scans_ids = []
+for repo in repos:
+    head_full_scans_id = repo.get("head_full_scan_id")
+    if head_full_scans_id:
+        head_full_scans_ids.append(head_full_scans_id)
+
+socket_results = {}
+for head_full_scan_id in head_full_scans_ids:
+    full_scan_metadata = sdk.fullscans.metadata(org_slug=org_slug, full_scan_id=head_full_scan_id)
+    full_scan_result = {
+        "repo": full_scan_metadata.get("repo"),
+        "branch": full_scan_metadata.get("branch"),
+        "commit_hash": full_scan_metadata.get("commit_hash"),
+        "commit_message": full_scan_metadata.get("commit_message"),
+        "pull_request_url": full_scan_metadata.get("pull_request_url"),
+        "committers": full_scan_metadata.get("committers"),
+        "created_at": full_scan_metadata.get("created_at"),
+        "results": sdk.fullscans.stream(org_slug=org_slug, full_scan_id=head_full_scan_id) or None
+    }
+    socket_results[head_full_scan_id] = full_scan_result
+
+print(json.dumps(socket_results, indent=4))