Coding style edits (#772)

log2timeline · Aug 3, 2023 · 22c4b09 · 22c4b09
1 parent c41269d
commit 22c4b09
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 11 deletions.
diff --git a/dftimewolf/lib/processors/turbinia_base.py b/dftimewolf/lib/processors/turbinia_base.py
@@ -412,26 +412,31 @@ def TurbiniaWait(self,
       time.sleep(interval)
       try:
         if self.RefreshClientCredentials():
-          # pylint: disable=line-too-long
-          self.requests_api_instance = turbinia_requests_api.TurbiniaRequestsApi(
-              self.client)
+          self.requests_api_instance = (
+              turbinia_requests_api.TurbiniaRequestsApi(self.client)
+          )
         request_data = self.requests_api_instance.get_request_status(request_id)
         status = request_data.get('status')
         failed_tasks = request_data.get('failed_tasks')
         successful_tasks = request_data.get('successful_tasks')
         task_count = request_data.get('task_count')
         progress = math.ceil(
-            ((failed_tasks + successful_tasks) / task_count) * 100)
+            ((failed_tasks + successful_tasks) / task_count) * 100
+        )
         self.PublishMessage(
-            f'Turbinia request {request_id} is {status}. Progress: {progress}%')
+            f'Turbinia request {request_id} is {status}. Progress: {progress}%'
+        )
 
         for task in request_data.get('tasks', []):
           current_saved_paths = task.get('saved_paths', [])
           if not current_saved_paths:
             continue
           for path in current_saved_paths:
-            if path not in processed_paths and self._isInterestingPath(
-                path) and path.startswith(self.output_path):
+            if (
+                path not in processed_paths
+                and self._isInterestingPath(path)
+                and path.startswith(self.output_path)
+            ):
               processed_paths.add(path)
               yield task, path
 
@@ -442,11 +447,12 @@ def TurbiniaWait(self,
   def TurbiniaFinishReport(self, request_id: str) -> str:
     """This method generates a report for a Turbinia request."""
     if self.RefreshClientCredentials():
-      # pylint: disable=line-too-long
       self.requests_api_instance = turbinia_requests_api.TurbiniaRequestsApi(
-          self.client)
+          self.client
+      )
     request_data = self.requests_api_instance.get_request_status(request_id)
     if request_data:
       report: str = turbinia_formatter.RequestMarkdownReport(
-          request_data=request_data).generate_markdown()
+          request_data=request_data
+      ).generate_markdown()
     return report
diff --git a/tests/lib/processors/turbinia_base.py b/tests/lib/processors/turbinia_base.py
@@ -7,7 +7,7 @@
 
 import turbinia_api_lib
 
-from google.auth.credentials import Credentials
+from google.oauth2.credentials import Credentials
 
 from dftimewolf.lib.processors import turbinia_base
 from dftimewolf import config