diff --git a/dzgui/views/trees/tree_mods.py b/dzgui/views/trees/tree_mods.py index 17f12fd..108f9ef 100644 --- a/dzgui/views/trees/tree_mods.py +++ b/dzgui/views/trees/tree_mods.py @@ -70,6 +70,8 @@ class ModTreeView(ModsMixin, ContextMixin, TreeView): # type: ignore def get_selected_mod(self) -> str: path = self.get_focused_row_path() model = self.get_model() + if model is None: + raise AttributeError("Trying to call a method on a non-existent model") tree_iter = model.get_iter(path) mod = model.get(tree_iter, 2)[0] return str(mod) diff --git a/dzgui/views/trees/tree_server_mods.py b/dzgui/views/trees/tree_server_mods.py index f527164..d3eb436 100644 --- a/dzgui/views/trees/tree_server_mods.py +++ b/dzgui/views/trees/tree_server_mods.py @@ -61,6 +61,8 @@ class ServerModTreeView(ContextMixin, TreeView): # type: ignore def get_selected_mod(self) -> str: path = self.get_focused_row_path() model = self.get_model() + if model is None: + raise AttributeError("Trying to call a method on a non-existent model") tree_iter = model.get_iter(path) # FIXME: https://docs.gtk.org/gtk3/method.TreeModel.get.html mod = model.get(tree_iter, 1)[0]