clean up crash on modulenotfound, also code tidying and duckdb upgrade

Author: nickzoic

countess/core/cmd.py

@@ -2,10 +2,8 @@
 import getopt
 import logging
 import logging.handlers
-import multiprocessing
 import re
 import sys
-import time
 from typing import Iterable, Union
 
 from countess import VERSION
@@ -95,5 +93,6 @@ def main() -> None:
 
     run(sys.argv[1:])
 
+
 if __name__ == "__main__":
     main()  # pragma: no cover

countess/core/config.py

@@ -3,7 +3,6 @@
 import logging
 import os.path
 import re
-import sys
 from configparser import ConfigParser
 
 from countess.core.pipeline import PipelineGraph, PipelineNode
@@ -69,7 +68,6 @@ def read_config(
 
 
 def config_to_graph(cp: ConfigParser, base_dir: str = ".") -> PipelineGraph:
-
     pipeline_graph = PipelineGraph()
     nodes_by_name: dict[str, PipelineNode] = {}
 

countess/core/plugins.py

@@ -69,11 +69,15 @@ def get_plugin_classes() -> Iterable[Type["BasePlugin"]]:
     return plugin_classes
 
 
-def load_plugin(module_name: str, class_name: str, plugin_name: Optional[str] = None) -> "BasePlugin":
-    module = importlib.import_module(module_name)
-    plugin_class = getattr(module, class_name)
-    assert issubclass(plugin_class, BasePlugin)
-    return plugin_class(plugin_name)
+def load_plugin(module_name: str, class_name: str, plugin_name: Optional[str] = None) -> Optional["BasePlugin"]:
+    try:
+        module = importlib.import_module(module_name)
+        plugin_class = getattr(module, class_name)
+        assert issubclass(plugin_class, BasePlugin)
+        return plugin_class(plugin_name)
+    except ModuleNotFoundError:
+        logger.error("Module Not Found: %s", module_name)
+        return None
 
 
 class BasePlugin(HasSubParametersMixin):

countess/gui/main.py

@@ -15,7 +15,7 @@
 from duckdb import DuckDBPyRelation
 
 from countess import VERSION
-from countess.core.config import export_config_graphviz, read_config, write_config, config_to_graph, graph_to_config
+from countess.core.config import config_to_graph, export_config_graphviz, graph_to_config, read_config, write_config
 from countess.core.pipeline import PipelineGraph
 from countess.core.plugins import get_plugin_classes
 from countess.gui.config import PluginConfigurator
@@ -345,7 +345,7 @@ def __init__(self, graph: PipelineGraph):
         self.button.grid(row=2, column=0, sticky=tk.EW)
 
         config = graph_to_config(graph)
-        self.process = multiprocessing.Process(target=runwindow_subprocess_target, args=[ config ])
+        self.process = multiprocessing.Process(target=runwindow_subprocess_target, args=[config])
         self.process.start()
 
         self.poll()

pyproject.toml

@@ -19,7 +19,7 @@ classifiers = [
 ]
 dependencies = [
     'biobear~=0.23.2',
-    'duckdb>=1.2.2',
+    'duckdb>=1.3.1',
     'fqfa~=1.3.1',
     'more_itertools~=9.1.0',
     'numpy~=1.26',