re-flake8'd the source

abowersox-cb · abowersox-cb · commit c330a01201fd · 2019-12-03T15:44:46.000-07:00
diff --git a/src/cbapi/psc/query.py b/src/cbapi/psc/query.py
@@ -991,31 +991,31 @@ def facets(self, fieldlist, max_rows=0):
         resp = self._cb.post_object(url, body=request)
         result = resp.json()
         return result.get("results", [])
-    
+
     def _update_status(self, status, remediation, comment):
-        request = { "state": status, "criteria": self._build_criteria(), "query": self._query_builder._collapse()}
+        request = {"state": status, "criteria": self._build_criteria(), "query": self._query_builder._collapse()}
         if remediation is not None:
             request["remediation_state"] = remediation
         if comment is not None:
             request["comment"] = comment
         resp = self._cb.post_object(self._bulkupdate_url.format(self._cb.credentials.org_key), body=request)
         output = resp.json()
         return output["request_id"]
-         
+
     def update(self, remediation=None, comment=None):
         """
         Update all alerts matching the given query. The alerts will be left in an OPEN state after this request.
-        
+
         :param remediation str: The remediation state to set for all alerts.
         :param comment str: The comment to set for all alerts.
         :return: The request ID, which may be used to select a WorkflowStatus object.
         """
         return self._update_status("OPEN", remediation, comment)
-    
+
     def dismiss(self, remediation=None, comment=None):
         """
         Dismiss all alerts matching the given query. The alerts will be left in a DISMISSED state after this request.
-        
+
         :param remediation str: The remediation state to set for all alerts.
         :param comment str: The comment to set for all alerts.
         :return: The request ID, which may be used to select a WorkflowStatus object.
@@ -1071,7 +1071,7 @@ class CBAnalyticsAlertSearchQuery(BaseAlertSearchQuery):
     valid_sensor_actions = ["POLICY_NOT_APPLIED", "ALLOW", "ALLOW_AND_LOG", "TERMINATE", "DENY"]
     valid_threat_cause_vectors = ["EMAIL", "WEB", "GENERIC_SERVER", "GENERIC_CLIENT", "REMOTE_DRIVE",
                                   "REMOVABLE_MEDIA", "UNKNOWN", "APP_STORE", "THIRD_PARTY"]
-    
+
     def __init__(self, doc_class, cb):
         super().__init__(doc_class, cb)
         self._bulkupdate_url = "/appservices/v6/orgs/{0}/alerts/cbanalytics/workflow/_criteria"
diff --git a/src/cbapi/psc/rest_api.py b/src/cbapi/psc/rest_api.py
@@ -136,11 +136,11 @@ def alert_search_suggestions(self, query):
         url = "/appservices/v6/orgs/{0}/alerts/search_suggestions".format(self.credentials.org_key)
         output = self.get_object(url, query_params)
         return output["suggestions"]
-    
+
     def _bulk_threat_update_status(self, threat_ids, status, remediation, comment):
         if not all(isinstance(t, str) for t in threat_ids):
             raise ApiError("One or more invalid threat ID values")
-        request = { "state": status, "threat_id": threat_ids}
+        request = {"state": status, "threat_id": threat_ids}
         if remediation is not None:
             request["remediation_state"] = remediation
         if comment is not None:
@@ -149,28 +149,27 @@ def _bulk_threat_update_status(self, threat_ids, status, remediation, comment):
         resp = self.post_object(url, body=request)
         output = resp.json()
         return output["request_id"]
-    
+
     def bulk_threat_update(self, threat_ids, remediation=None, comment=None):
         """
         Update the alert status of alerts associated with multiple threat IDs.
         The alerts will be left in an OPEN state after this request.
-        
+
         :param threat_ids list: List of string threat IDs.
         :param remediation str: The remediation state to set for all alerts.
         :param comment str: The comment to set for all alerts.
         :return: The request ID, which may be used to select a WorkflowStatus object.
         """
         return self._bulk_threat_update_status(threat_ids, "OPEN", remediation, comment)
-        
+
     def bulk_threat_dismiss(self, threat_ids, remediation=None, comment=None):
         """
         Dismiss the alerts associated with multiple threat IDs.
         The alerts will be left in a DISMISSED state after this request.
-        
+
         :param threat_ids list: List of string threat IDs.
         :param remediation str: The remediation state to set for all alerts.
         :param comment str: The comment to set for all alerts.
         :return: The request ID, which may be used to select a WorkflowStatus object.
         """
         return self._bulk_threat_update_status(threat_ids, "DISMISSED", remediation, comment)
-        
