Orlando/add new endpoints (#2)

* added the sbom report output in a dict format

* added sbom function

---------

Co-authored-by: Orlando Barrera II <[email protected]>
Co-authored-by: Orlando Barrera II <[email protected]>

Author: 3 people

.gitignore

@@ -5,3 +5,4 @@ venv
 .DS_Store
 *.zip
 *.pyc
+test.py

socketdev/__init__.py

@@ -1,12 +1,14 @@
 import logging
 import requests
 import base64
+
 from socketdev.dependencies import Dependencies
 from socketdev.npm import NPM
 from socketdev.openapi import OpenAPI
 from socketdev.org import Orgs
 from socketdev.quota import Quota
 from socketdev.report import Report
+from socketdev.sbom import Sbom
 from socketdev.repositories import Repositories
 from socketdev.settings import Settings
 from socketdev.socket_classes import Dependency, Org, Response
@@ -86,5 +88,6 @@ def __init__(self, token: str):
         self.org = Orgs()
         self.quota = Quota()
         self.report = Report()
+        self.sbom = Sbom()
         self.repositories = Repositories()
         self.settings = Settings()

socketdev/sbom/__init__.py

@@ -0,0 +1,23 @@
+import socketdev
+import json
+
+class Sbom:
+    @staticmethod
+    def get_sbom_data(report_id: str) -> list:
+        path = f"sbom/view/{report_id}"
+        response = socketdev.do_request(path=path)
+        if response.status_code == 200:
+            sbom = []
+            sbom_dict = {}
+            data = response.text
+            data.strip('"')
+            data.strip()
+            for line in data.split("\n"):
+                if line != '"' and line != "" and line is not None:
+                    item = json.loads(line)
+                    sbom.append(item)
+            for key, val in enumerate(sbom):
+                sbom_dict[val['id']] = val
+        else:
+            sbom_dict = {}
+        return sbom_dict