class FamilyDiffer(VBox): def __init__(self, conn, name='FamilyDiffer'): VBox.__init__(self) self.set_name(name) self.conn = conn self.view = TwinScrollCList(name=name) self.cursor = StatementCursor(self.conn) self.lfamily = Family(self.conn) self.rfamily = Family(self.conn) self.add(self.view) self.udbar = UDBar() self.pack_end(self.udbar, 0, 0, 0) self.show() self.udbar.ubutton.connect('clicked', self.update_pressed) self.udbar.dbutton.connect('clicked', self.diff_selection) def update_pressed(self, button): self.update_lists() def update_lists(self): rows = self.lfamily.family_rows() self.view.lbox.set_rows(rows) self.view.rbox.set_rows(rows) def diff_selection(self, *args): lrow = self.view.lbox.get_selected_data()[0] rrow = self.view.rbox.get_selected_data()[0] lfam, rfam = lrow.family, rrow.family lcfg = FamilyVariablesConfig(self.conn, lfam) rcfg = FamilyVariablesConfig(self.conn, rfam) lcfg.diff(rcfg)
class FamilyMainWindow(SimpleSplitWindow): def __init__(self, app, parent): SimpleSplitWindow.__init__(self, app, parent, FamilyView, 'FamilyMainWindow') self.app = app self.initActions() self.initMenus() self.initToolbar() self.conn = app.conn self.cfg = app.cfg self.cursor = StatementCursor(self.conn) self.family = Family(self.conn) self.refreshListView() self.resize(600, 800) self.setCaption('paella families') def initActions(self): collection = self.actionCollection() def initMenus(self): mainMenu = KPopupMenu(self) menus = [mainMenu] self.menuBar().insertItem('&Main', mainMenu) self.menuBar().insertItem('&Help', self.helpMenu('')) def initToolbar(self): toolbar = self.toolBar() def initlistView(self): self.listView.setRootIsDecorated(True) self.listView.addColumn('group') def refreshListView(self): for row in self.family.family_rows(): item = KListViewItem(self.listView, row.family) item.family = row.family def selectionChanged(self): current = self.listView.currentItem() if hasattr(current, 'family'): self.view.set_family(current.family) if hasattr(current, 'trait'): print 'trait is', current.trait self.view.set_trait(current.trait) if hasattr(current, 'suite'): print 'suite is', current.suite if hasattr(current, 'widget'): print 'widget is', current.widget
class FamilyList(KListView): def __init__(self, app, parent, name='FamilyList'): KListView.__init__(self, parent, name) dbwidget(self, app) self.family = Family(self.conn) self.setRootIsDecorated(True) self.addColumn('family') self.refreshlistView() def refreshlistView(self): self.clear() rows = self.family.family_rows() for row in rows: item = KListViewItem(self, row.family) item.family = row.family def getData(self): item = self.currentItem() return FamilyVariablesConfig(self.conn, item.family) def updateData(self, data): pass
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): 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 = FamilyVariablesConfig(self.conn, 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() data = dict(family=self.current_family) for row in rows: data['trait'] = row.trait data['name'] = row.name clause = reduce(and_, [Eq(k, v) for k,v in data.items()]) self.family.env.cursor.delete(clause=clause) 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] print rows for r in trips: clause = Eq('family', self.current_family) clause &= Eq('trait', r[0]) & Eq('name', r[1]) trows = self.family.cursor.select(table=table, clause=clause) print 'trows', trows if len(trows) == 0: data = dict(trait=r[0], name=r[1], value=r[2]) data['family'] = self.current_family print 'data', data self.family.cursor.insert(table=table, data=data) 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)