Choose spouse if not defined

svn: r873

src/AddSpouse.py

@@ -187,3 +187,132 @@ class AddSpouse:
             index = index + 1
         self.spouse_list.thaw()
 
+
+class SetSpouse:
+    """
+    Displays the AddSpouse dialog, allowing the user to create a new
+    family with the passed person as one spouse, and another person to
+    be selected.
+    """
+    def __init__(self,db,person,family,update,addperson):
+        """
+        Displays the AddSpouse dialog box.
+
+        db - database to which to add the new family
+        person - the current person, will be one of the parents
+        update - function that updates the family display
+        addperson - function that adds a person to the person view
+        """
+        self.db = db
+        self.update = update
+        self.person = person
+        self.family = family
+        self.addperson = addperson
+
+        self.glade = libglade.GladeXML(const.gladeFile, "spouseDialog")
+
+        self.rel_combo = self.glade.get_widget("rel_combo")
+        self.relation_type = self.glade.get_widget("rel_type")
+        self.spouse_list = self.glade.get_widget("spouseList")
+        self.relation_def = self.glade.get_widget("reldef")
+        self.top = self.glade.get_widget("spouseDialog")
+
+        self.name_list = self.db.getPersonMap().values()
+        self.name_list.sort(sort.by_last_name)
+        self.rel_combo.set_popdown_strings(const.familyRelations)
+        title = _("Choose Spouse/Partner of %s") % GrampsCfg.nameof(person)
+        self.glade.get_widget("spouseTitle").set_text(title)
+
+        self.glade.signal_autoconnect({
+            "on_select_spouse_clicked" : self.select_spouse_clicked,
+            "on_new_spouse_clicked"    : self.new_spouse_clicked,
+            "on_rel_type_changed"      : self.relation_type_changed,
+            "on_combo_insert_text"     : Utils.combo_insert_text,
+            "destroy_passed_object"    : Utils.destroy_passed_object
+            })
+
+        self.relation_type.set_text(_("Married"))
+
+    def new_spouse_clicked(self,obj):
+        """
+        Called when the spouse to be added does not exist, and needs
+        to be created and added to the database
+        """
+        import QuickAdd
+
+        relation = const.save_frel(self.relation_type.get_text())
+        if relation == "Partners":
+            if self.person.getGender() == RelLib.Person.male:
+                gen = "male"
+            else:
+                gen = "female"
+        elif self.person.getGender() == RelLib.Person.male:
+            gen = "female"
+        else:
+            gen = "male"
+        QuickAdd.QuickAdd(self.db,gen,self.update_list)
+
+    def update_list(self,person):
+        self.name_list.append(person)
+        self.name_list.sort(sort.by_last_name)
+        self.addperson(person)
+        self.relation_type_changed(self.relation_type)
+
+    def select_spouse_clicked(self,obj):
+        """
+        Called when the spouse to be added already exists and has been
+        selected from the list.
+        """
+        if len(self.spouse_list.selection) == 0:
+            return
+        row = self.spouse_list.selection[0]
+        spouse = self.spouse_list.get_row_data(row)
+
+        # don't do anything if the marriage already exists
+        for f in self.person.getFamilyList():
+            if spouse == f.getMother() or spouse == f.getFather():
+                Utils.destroy_passed_object(obj)
+                return
+
+        Utils.modified()
+        if self.family.getFather() == self.person:
+            self.family.setMother(spouse)
+        else:
+            self.family.setFather(spouse)
+            
+        spouse.addFamily(self.family)
+
+        self.family.setRelationship(const.save_frel(self.relation_type.get_text()))
+        Utils.destroy_passed_object(obj)
+        self.update(self.family)
+
+    def relation_type_changed(self,obj):
+        """
+        Called whenever the relationship type changes. Rebuilds the
+        the potential spouse list.
+        """
+        text = obj.get_text()
+        self.relation_def.set_text(const.relationship_def(text))
+    
+        # determine the gender of the people to be loaded into
+        # the potential spouse list. If Partners is selected, use
+        # the same gender as the current person.
+        gender = self.person.getGender()
+        if text == _("Partners"):
+            if gender == RelLib.Person.male:
+                gender = RelLib.Person.female
+            else:
+                gender = RelLib.Person.male
+	
+        index = 0
+        self.spouse_list.clear()
+        self.spouse_list.freeze()
+        for person in self.name_list:
+            if person.getGender() == gender:
+                continue
+            name = person.getPrimaryName().getName()
+            self.spouse_list.append([name,Utils.birthday(person)])
+            self.spouse_list.set_row_data(index,person)
+            index = index + 1
+        self.spouse_list.thaw()
+