style: format code with Black

This commit fixes the style issues introduced in 1b12fb2 according to the output
from Black.

Details: None

Author: deepsource-autofix[bot]

codeinsight_sdk/client.py

@@ -8,101 +8,116 @@
 
 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
+            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)
+        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
 
     @property
     def projects(self) -> ProjectHandler:
         return ProjectHandler(self)
-    
+
     @property
     def reports(self) -> ReportHandler:
         return ReportHandler(self)
-    
+
     @property
     def inventories(self):
         return InventoryHandler(self)
-    
+
     @property
     def experimental(self) -> ExperimentalHandler:
         if self.experimental_enabled == False:
             raise CodeInsightError("Experimental API is not enabled for this instance")
         else:
             return ExperimentalHandler(self)
-    
-    
+
     # Coming soon...?
 
-    
     def vulnerabilites(self):
         raise NotImplementedError
-    
+
     def users(self):
         raise NotImplementedError
-    
+
     def licenses(self):
         raise NotImplementedError
-    
+
     def tasks(self):
         raise NotImplementedError
-    
+
     def rules(self):
         raise NotImplementedError
-    
+
     def files(self):
         raise NotImplementedError
-    
+
     def folders(self):
         raise NotImplementedError
-    
+
     def jobs(self):
         raise NotImplementedError
-    
+
     def components(self):
         raise NotImplementedError
-

codeinsight_sdk/exceptions.py

@@ -1,24 +1,29 @@
-
 import requests
 import json
 
-class GenericError(Exception): #pragma: no cover
+
+class GenericError(Exception):  # pragma: no cover
     """Generic error class, catch-all for most code insight API errors."""
+
     pass
 
-class NotYetImplementedError(Exception): #pragma: no cover
+
+class NotYetImplementedError(Exception):  # pragma: no cover
     """Error class for API features that have not yet been implemented."""
+
     pass
 
+
 class CodeInsightError(GenericError):
     """Error class for code insight API errors."""
+
     def __init__(self, response: requests.Response):
         try:
             resp = response.json()
             self.code = response.status_code
-            self.message = resp['Error: ']
-            self.arguments = resp['Arguments: ']
-            self.error = resp['Key: ']
+            self.message = resp["Error: "]
+            self.arguments = resp["Arguments: "]
+            self.error = resp["Key: "]
             self.add_note(f"Arguments: {self.arguments}")
             super().__init__("Error: %s - %s" % (self.code, self.message))
 

codeinsight_sdk/experimental.py

@@ -1,43 +1,49 @@
 from .handler import Handler
 from .models import ProjectInventoryItem
 
+
 class ExperimentalHandler(Handler):
     def __init__(self, client):
         super().__init__(client)
-    
+
     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.
-        
+
         Args:
             project_id (int): The project id.
-        
+
         Returns:
             dict: The vulnerabilities.
         """
         # 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
         vuln_items: list(ProjectInventoryItem) = []
         for item in inventory:
             if item.vulnerabilitySummary is None:
                 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/handler.py

@@ -4,6 +4,7 @@
 from .models import Project, ProjectInventory, ProjectInventoryItem, Report
 from .exceptions import CodeInsightError
 
+
 class Handler(abc.ABC):
     def __init__(self, client):
         self.client = client
@@ -12,4 +13,3 @@ def __init__(self, client):
     @abc.abstractmethod
     def get(self):
         pass
-

codeinsight_sdk/handlers/__init__.py

@@ -1,3 +1,3 @@
 from .inventory import InventoryHandler
 from .project import ProjectHandler
-from .report import ReportHandler
+from .report import ReportHandler

codeinsight_sdk/handlers/inventory.py

@@ -1,13 +1,14 @@
 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)
         self.cls = ProjectInventoryItem
-    
+
     def get(self, inventoryId: int) -> list[ProjectInventoryItem]:
         """
         Get an inventory item by id.
@@ -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