style: format code with Black

This commit fixes the style issues introduced in 9d87d9b according to the output
from Black.

Details: #17

Author: deepsource-autofix[bot]

codeinsight_sdk/client.py

@@ -8,51 +8,69 @@
 
 logger = logging.getLogger(__name__)
 
+
 class CodeInsightClient:
     """Client for the code insight API."""
 
-    def __init__(self,
-                 base_url: str,
-                 api_token: str,
-                 timeout: int = 60,
-                 verify_ssl: bool = True,
-                 experimental: bool = False
-                 ):
-
+    def __init__(
+        self,
+        base_url: str,
+        api_token: str,
+        timeout: int = 60,
+        verify_ssl: bool = True,
+        experimental: bool = False,
+    ):
         self.base_url = base_url
         self.api_url = f"{base_url}/codeinsight/api"
         self.__api_token = api_token
         self.__api_headers = {
-            'Content-Type': 'application/json',
+            "Content-Type": "application/json",
             "Authorization": "Bearer %s" % self.__api_token,
             "User-Agent": "codeinsight_sdk_python",
         }
         self.__timeout = timeout
         self.__verify_ssl = verify_ssl
         self.experimental_enabled = experimental
 
-    def request(self, method, url_part: str, params: dict = None, body: any = None, data: any = None, content_type: str = None):
+    def request(
+        self,
+        method,
+        url_part: str,
+        params: dict = None,
+        body: any = None,
+        data: any = None,
+        content_type: str = None,
+    ):
         url = f"{self.api_url}/{url_part}"
 
         # Iterate over params and remove any that are None (Empty)
-        if(params):
+        if params:
             for k, v in list(params.items()):
                 if v is None:
                     del params[k]
 
         # Update headers if content_type is specified
         headers = self.__api_headers
         if content_type:
-            headers['Content-Type'] = content_type
-
-        response = requests.request(method, url,
-                                     headers=self.__api_headers, params=params, json=body, data=data,
-                                     timeout=self.__timeout, verify=self.__verify_ssl)
+            headers["Content-Type"] = content_type
+
+        response = requests.request(
+            method,
+            url,
+            headers=self.__api_headers,
+            params=params,
+            json=body,
+            data=data,
+            timeout=self.__timeout,
+            verify=self.__verify_ssl,
+        )
 
         if not response.ok:
-            logger.error(f"Error: {response.status_code} - {response.reason}", exc_info=True)
+            logger.error(
+                f"Error: {response.status_code} - {response.reason}", exc_info=True
+            )
             logger.error(response.text)
-            raise CodeInsightError(response)   
+            raise CodeInsightError(response)
 
         return response
 
@@ -75,10 +93,8 @@ def experimental(self) -> ExperimentalHandler:
         else:
             return ExperimentalHandler(self)
 
-
     # Coming soon...?
 
-
     def vulnerabilites(self):
         raise NotImplementedError
 
@@ -105,4 +121,3 @@ def jobs(self):
 
     def components(self):
         raise NotImplementedError
-

codeinsight_sdk/experimental.py

@@ -1,6 +1,7 @@
 from .handler import Handler
 from .models import ProjectInventoryItem
 
+
 class ExperimentalHandler(Handler):
     def __init__(self, client):
         super().__init__(client)
@@ -9,7 +10,9 @@ def get(self):
         # Do nothing, there is no get for this handler
         pass
 
-    def get_project_vulnerabilities(self, project_id:int) -> list[ProjectInventoryItem]:
+    def get_project_vulnerabilities(
+        self, project_id: int
+    ) -> list[ProjectInventoryItem]:
         """
         Get all vulnerabilities for a project.
 
@@ -21,7 +24,9 @@ def get_project_vulnerabilities(self, project_id:int) -> list[ProjectInventoryIt
         """
         # First we get the inventory summary for the project with vulnerability summary
         # Then we iterate over the inventory items and calling the inventory vulnerability endpoint for each item with a vulnerability
-        inventory = self.client.projects.get_inventory_summary(project_id, vulnerabilitySummary=True)
+        inventory = self.client.projects.get_inventory_summary(
+            project_id, vulnerabilitySummary=True
+        )
 
         # Iterate over the inventory items, find which have vulnerabilities.
         item: ProjectInventoryItem
@@ -31,13 +36,14 @@ def get_project_vulnerabilities(self, project_id:int) -> list[ProjectInventoryIt
                 continue
 
             # If the item has a vulnerability, get the vulnerability details for this item and append it
-            if sum(item.vulnerabilitySummary[0]['CvssV3'].values()) > 0:
-
-                vul_detail = self.client.inventories.get_inventory_vulnerabilities(item.id)
+            if sum(item.vulnerabilitySummary[0]["CvssV3"].values()) > 0:
+                vul_detail = self.client.inventories.get_inventory_vulnerabilities(
+                    item.id
+                )
                 item.vulnerabilities = vul_detail
                 vuln_items.append(item)
             else:
                 # If the item has no vulnerabilities, skip it
                 continue
 
-        return vuln_items
+        return vuln_items

codeinsight_sdk/handlers/inventory.py

@@ -1,8 +1,9 @@
 from ..models import ProjectInventoryItem, Vulnerability
 from ..handler import Handler
 
+
 class InventoryHandler(Handler):
-    """ Handles operations related to inventories."""
+    """Handles operations related to inventories."""
 
     def __init__(self, client):
         super().__init__(client)
@@ -21,13 +22,13 @@ def get(self, inventoryId: int) -> list[ProjectInventoryItem]:
         path = f"inventories/{inventoryId}"
         resp = self.client.request("GET", url_part=path)
         inventory = []
-        for inv_item in resp.json()['data']:
+        for inv_item in resp.json()["data"]:
             inventory.append(ProjectInventoryItem.from_dict(inv_item))
         return inventory
 
-    def get_inventory_vulnerabilities(self, inventoryId: int,
-                                      limit: int = 25,
-                                      offset: int = 1) -> list[Vulnerability]:
+    def get_inventory_vulnerabilities(
+        self, inventoryId: int, limit: int = 25, offset: int = 1
+    ) -> list[Vulnerability]:
         """
         Get all vulnerabilities for an inventory item.
 
@@ -44,7 +45,7 @@ def get_inventory_vulnerabilities(self, inventoryId: int,
         # TODO - Iterate pages
 
         inventory_vuls: list(Vulnerability) = []
-        for v in resp.json()['data']:
+        for v in resp.json()["data"]:
             inventory_vuls.append(Vulnerability.from_dict(v))
 
-        return inventory_vuls
+        return inventory_vuls