Add generic exception handler to bigquery. (log2timeline#842)

Also, fix linter warning.
jkppr · Mar 13, 2024 · 911122f · 911122f
1 parent 8bf31b8
commit 911122f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/dftimewolf/lib/collectors/bigquery.py b/dftimewolf/lib/collectors/bigquery.py
@@ -76,6 +76,11 @@ def Process(self, container: containers.BigQueryQuery
         )
       self.ModuleError(str(exception), critical=True)
 
+    except Exception as error:  # pylint: disable=broad-except
+      self.ModuleError(
+          f'Unknown exception encountered: {str(error)}',
+          critical=True)
+
     out_container: Union[containers.DataFrame, containers.File]
     if container.pandas_output:
       out_container = containers.DataFrame(

diff --git a/dftimewolf/lib/exporters/timesketch.py b/dftimewolf/lib/exporters/timesketch.py
@@ -258,6 +258,7 @@ def Process(self, container: containers.File) -> None:
             critical=False)
       if streamer.response and container.description:
         streamer.timeline.description = container.description
+  # pytype: enable=signature-mismatch
 
   def GetThreadOnContainerType(self) -> Type[interface.AttributeContainer]:
     return containers.File