chore: clear typehinting errors

dzgui/managers/contextmenu.py

@@ -8,6 +8,8 @@ from dzgui.util.clip import copy_clipboard
 from dzgui.util.open_links import open_workshop_page
 
 from dzgui.views.dialogs.note import NoteDialog
+from dzgui.views.trees.tree_mods import ModTreeView
+from dzgui.views.trees.tree_log import LogTreeView
 from dzgui.views.trees.tree_servers import ServerTreeView
 
 
@@ -19,13 +21,12 @@ logger = logging.getLogger(APP_NAME)
 
 if TYPE_CHECKING:
     from dzgui.controllers.mc import Controller
-    from dzgui.views.trees.tree_servers import ServerTreeView
 
 
 class ContextMenuManager:
     def __init__(
         self,
-        treeview: "ServerTreeView",
+        treeview: LogTreeView | ModTreeView | ServerTreeView,
         controller: "Controller",
     ) -> None:
         self.controller = controller
@@ -49,28 +50,40 @@ class ContextMenuManager:
                 self.open_mod_page()
             case ContextMenu.SET_FAV:
                 name = self.treeview.get_value_at_index(0)
-                record = self.treeview.get_record_string()
+                record_str = self.treeview.get_record_string()
                 simple = self.treeview.get_simplified_ip()
-                self.controller.set_fav(name, record, simple)
+                self.controller.set_fav(name, record_str, simple)
 
             # THREADED
             case ContextMenu.ADD_SERVER:
                 record = self.treeview.get_record()
+                if record is None:
+                    return
                 self.controller.add_by_record(record)
             case ContextMenu.CONNECT:
                 record = self.treeview.get_record()
+                if record is None:
+                    return
                 self.controller.connect_by_record(record)
             case ContextMenu.REFRESH_PLAYERS:
                 record = self.treeview.get_record()
+                if record is None:
+                    return
                 self.controller.refresh_players(record)
             case ContextMenu.REMOVE_HISTORY:
                 record = self.treeview.get_record()
+                if record is None:
+                    return
                 self.controller.remove_from_history(record)
             case ContextMenu.REMOVE_SERVER:
                 record = self.treeview.get_record()
+                if record is None:
+                    return
                 self.controller.remove_by_record(record)
             case ContextMenu.SHOW_DETAILS:
                 record = self.treeview.get_record()
+                if record is None:
+                    return
                 self.controller.get_details(record)
             case ContextMenu.SHOW_MODS:
                 record = self.treeview.get_record()
@@ -89,11 +102,15 @@ class ContextMenuManager:
         copy_clipboard(name)
 
     def open_mod_page(self) -> None:
+        if not isinstance(self.treeview, ModTreeView):
+            return
         mod = self.treeview.get_selected_mod()
         cmd = self.controller.query_config(Preferences.CLIENT)
         open_workshop_page(mod, cmd)
 
     def copy_log(self) -> None:
+        if not isinstance(self.treeview, LogTreeView):
+            return
         log = self.treeview.concatenate_rows()
         if log is None:
             return