better fix for #13

pyarchinit · Oct 6, 2019 · 5f8f9d1 · 5f8f9d1
1 parent bbcf276
commit 5f8f9d1
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/select_by_relationship_plugin.py b/select_by_relationship_plugin.py
@@ -51,6 +51,8 @@ def __init__(self, iface):
         self.iface = iface
         self.sFr = None
 
+        self.iface.newProjectCreated.connect(self.resetRelation)
+
         self.actions = []
         self.menu = self.tr(u'&Select by relationship')
         self.toolbar = self.iface.addToolBar(u'SelectByRelationship')
@@ -188,6 +190,12 @@ def unload(self):
             self.sFr.disable()
             del self.sFr
 
+    def resetRelation(self):
+        if self.sFr:
+            self.sFr.relationsChecked = {}
+        self.comboRelations._checkedItems = set()
+        self.comboRelations.setCheckedItems(set())
+
     def populateComboRelations(self):
         self.comboRelations.clear()
         relations = self.sFr.manager.relations().keys()