class FamilyParentAssigner(BaseAssigner): def __init__(self, parent, family): self.app = get_application_pointer() self.conn = self.app.conn self.family = Family(self.conn) self.family.set_family(family) BaseAssigner.__init__(self, parent, name='FamilyParentAssigner', udbuttons=False) self.connect(self, SIGNAL('okClicked()'), self.slotAssignParents) def initView(self): family = self.family.current all_fams = self.family.all_families() assigned_parents = self.family.parents() avail_parents = [f for f in all_fams if f not in assigned_parents + [family]] abox = self.listBox.availableListBox() sbox = self.listBox.selectedListBox() for family in assigned_parents: self._add_family_to_listbox(sbox, family) for family in avail_parents: self._add_family_to_listbox(abox, family) self.already_assigned = assigned_parents def _add_family_to_listbox(self, box, family): item = QListBoxText(box, family) item.family = family def slotAssignParents(self): sbox = self.listBox.selectedListBox() families =[str(sbox.item(n).text()) for n in range(sbox.numRows())] # since we're using a SimpleRelation class for # family.parents, we don't need to worry about # inserting families that are already there self.family.delete_parents() self.family.insert_parents(families)
class FamilyParentAssigner(BaseAssigner): def __init__(self, parent, family): self.app = get_application_pointer() self.conn = self.app.conn self.family = Family(self.conn) self.family.set_family(family) BaseAssigner.__init__(self, parent, name='FamilyParentAssigner', udbuttons=False) self.connect(self, SIGNAL('okClicked()'), self.slotAssignParents) def initView(self): family = self.family.current all_fams = self.family.all_families() assigned_parents = self.family.parents() avail_parents = [ f for f in all_fams if f not in assigned_parents + [family] ] abox = self.listBox.availableListBox() sbox = self.listBox.selectedListBox() for family in assigned_parents: self._add_family_to_listbox(sbox, family) for family in avail_parents: self._add_family_to_listbox(abox, family) self.already_assigned = assigned_parents def _add_family_to_listbox(self, box, family): item = QListBoxText(box, family) item.family = family def slotAssignParents(self): sbox = self.listBox.selectedListBox() families = [str(sbox.item(n).text()) for n in range(sbox.numRows())] # since we're using a SimpleRelation class for # family.parents, we don't need to worry about # inserting families that are already there self.family.delete_parents() self.family.insert_parents(families)
class FamilyBrowser(ListNoteBook): def __init__(self, conn): self.menu = make_menu(['delete'], self.modify_family) ListNoteBook.__init__(self) self.conn = conn self.family = Family(self.conn) self.var_menu = make_menu(['edit', 'nothing', 'nothing', 'drop'], self.var_menu_selected) self.parent_menu = make_menu(['drop'], self.modify_parent) self.reset_rows() self.append_page(ScrollCList(rcmenu=self.var_menu), 'environment') self.append_page(ScrollCList(rcmenu=self.parent_menu), 'parents') self.set_size_request(400, 300) def modify_parent(self, menuitem, action): if action == 'drop': parents = self._get_listbox_col_('parents', 'family') self.trait.delete_parents(parents) self.__set_pages(self.current_family) def modify_family(self, menuitem, action): if action == 'delete': trait = self.listbox.get_selected_data()[0].family self.trait.delete_family(family) self.reset_rows() def reset_rows(self): self.set_rows(self.family.family_rows()) self.set_row_select(self.family_selected) def __set_droptargets__(self, pages): set_receive_targets(pages['environment'].listbox, self.drop_variable, TARGETS.get('variable', 'flavor')) set_receive_targets(pages['parents'].listbox, self.drop_family, TARGETS.get('family', 'flavor')) def set_package(self, menu_item, action): raise Error, 'this call (set_package) is not needed in Family' packages = self._get_listbox_col_('packages', 'package') trait = self.current_family self.trait.set_action(action, packages) self.__set_pages(self.current_trait) def var_menu_selected(self, menu_item, action): if action == 'edit': config = self.family.getVariablesConfig(self.current_family) newconfig = config.edit() config.update(newconfig) self.select_family(self.current_family) elif action == 'drop': pages = dict(self.pages) listbox = pages['environment'].listbox rows = listbox.get_selected_data() for row in rows: self.family.deleteVariable(row.trait, row.name, self.current_family) def pop_mymenu(self, widget, event, menu): if right_click_pressed(event): menu.popup(None, None, None, event.button, event.time) def family_selected(self, listbox, row, column, event): family = listbox.get_selected_data()[0].family self.select_family(family) def select_family(self, family): self.current_family = family self.family.set_family(family) self.__set_pages(self.current_family) def __set_pages(self, family): pages = dict(self.pages) pages['environment'].set_rows(self.family.environment_rows()) pages['environment'].set_select_mode('multi') pages['parents'].set_rows(self.family.parents(), ['family']) pages['parents'].set_select_mode('multi') self.__set_droptargets__(pages) def drop_family(self, listbox, context, x, y, selection, targettype, time): families = Set(selection.data.split('^&^')) self.family.insert_parents(families) self.__set_pages(self.current_family) def drop_variable(self, listbox, context, x, y, selection, targettype, time): table = 'family_environment' trips = [x.split('<=>') for x in selection.data.split('^&^')] rows = [dict(trait=x[0], name=x[1], value=x[2]) for x in trips] for r in trips: trait, name, value = r self.family.insertVariable(trait, name, value, family=self.current_family) self.__set_pages(self.current_family) def _get_listbox_col_(self, page, field): pages = dict(self.pages) return [row[field] for row in pages[page].listbox.get_selected_data()] def make_families_window(self): rows = self.family.family_rows() FamiliesWindow(rows) def make_variables_window(self): rows = self.family.get_all_defaults() VariablesWindow(rows)