class EnvironmentEditor(_EnvironmentEditor): def __init__(self, conn, suite, name='EnvironmentEditor'): self.menu = make_menu(['update', 'create'], self.env_command) _EnvironmentEditor.__init__(self, conn, name=name) self.suite = suite self.traitparent = TraitParent(self.conn, self.suite) self.current_trait = None self.reset_rows() w, h = map(int, config.get_list('env_editor_size', section='management_gui')) self.set_size_request(w, h) def reset_rows(self): rows = self.main.select(fields=['trait'], table=ujoin(self.suite, 'traits')) self.set_rows(rows) self.set_row_select(self.trait_selected) def trait_selected(self, listbox, row, column, event): trait = listbox.get_selected_data()[0].trait self.select_trait(trait) def select_trait(self, trait): self.current_trait = trait environ = TraitEnvironment(self.conn, self.suite, trait) self.environ = self.traitparent.get_environment([self.current_trait]) self.sections = dict([(k, v.keys()) for k,v in self.environ]) self.remove_page('Environment') self.record = RecordBox(environ, name=self.current_trait) self.append_page(self.record, 'Environment') self.pages['Environment'].connect('button_press_event', self.box_pressed) #self.menu.connect('populate', self.populate_menu, self.menu) self.record.connect_entries('populate-popup', self.populate_menu, self.menu) def make_new_variable(self, *args): d = self.dialogs['create'] environ = TraitEnvironment(self.conn, self.suite, self.current_trait) environ.update(dict([(d['name'], d['value'])])) def remove_item(self, *args): d = self.dialogs['remove'] name = d.get_selected_data()[0].name environ = TraitEnvironment(self.conn, self.suite, self.current_trait) del environ[name] dialogs.Message('%s deleted' %name) self.destroy_dialog(d) def update_environment(self): environ = TraitEnvironment(self.conn, self.suite, self.current_trait) environ.update(self.record) def env_command(self, menuitem, command): if command == 'update': dialogs.Message('updating %s' % self.current_trait) self.update_environment() elif command == 'create': self.ask_new_entry() else: dialogs.Message('ack')
class TemplateEnvironment(ScrolledWindow): def __init__(self, conn): ScrolledWindow.__init__(self) self.conn = conn self.editor = TemplateEditor() self.menu = SimpleMenu() self.editor.connect('populate-popup', self.populate_menu, self.menu) self.add_with_viewport(self.editor) self.show() def set_suite(self, suite): self.suite = suite self.__set_suitecursors__() def __set_suitecursors__(self): self.traitparent = TraitParent(self.conn, self.suite) self.traittemplate = TraitTemplate(self.conn, self.suite) self.traitpackage = TraitPackage(self.conn, self.suite) def set_trait(self, trait): self.trait = trait self.traitparent.set_trait(trait) self.traittemplate.set_trait(trait) self.traitpackage.set_trait(trait) self.__set_environment__() def __set_environment__(self): self.environ = dict(self.traitparent.get_environment([self.trait])) self.sections = dict([(k, v.keys()) for k, v in self.environ.items()]) def populate_menu(self, widget, mainmenu, tmenu): self.__set_environment__() populate_menu(mainmenu, '_define_', ['_create_'], self.make_new_tag) for section, vars in self.sections.items(): populate_menu(mainmenu, section, vars, self.make_tag) def make_tag(self, menuitem, name, parent): parentname = parent.get_name() tagname = ujoin(parentname, name) self.editor.make_tag(tagname) def make_new_tag(self, menuitem, name, parent): parentname = parent.get_name() if (parentname, name) == ('_define_', '_create_'): self.editor.create_new_tag(self.create_entry) def create_entry(self, *args): var = dict(self.editor.dialogs['create'].items()) debug('var is %s' % var) self.environ[self.trait][var['name']] = var['value'] tagname = ujoin(self.trait, var['name']) self.editor.make_tag(tagname) self.editor.destroy_dialog(self.editor.dialogs['create']) def preview(self, *args): subs = self.traitparent.get_superdict(self.environ.keys()) self.editor.buffer.set_subs(subs) self.editor.preview(*args) def set_text(self, text): self.editor.buffer.set_text(text) def get_text(self): return self.editor.get_text()
class TemplateEnvironment(ScrolledWindow): def __init__(self, conn): ScrolledWindow.__init__(self) self.conn = conn self.editor = TemplateEditor() self.menu = SimpleMenu() self.editor.connect('populate-popup', self.populate_menu, self.menu) self.add_with_viewport(self.editor) self.show() def set_suite(self, suite): self.suite = suite self.__set_suitecursors__() def __set_suitecursors__(self): self.traitparent = TraitParent(self.conn, self.suite) self.traittemplate = TraitTemplate(self.conn, self.suite) self.traitpackage = TraitPackage(self.conn, self.suite) def set_trait(self, trait): self.trait = trait self.traitparent.set_trait(trait) self.traittemplate.set_trait(trait) self.traitpackage.set_trait(trait) self.__set_environment__() def __set_environment__(self): self.environ = dict(self.traitparent.get_environment([self.trait])) self.sections = dict([(k, v.keys()) for k,v in self.environ.items()]) def populate_menu(self, widget, mainmenu, tmenu): self.__set_environment__() populate_menu(mainmenu, '_define_', ['_create_'], self.make_new_tag) for section, vars in self.sections.items(): populate_menu(mainmenu, section, vars, self.make_tag) def make_tag(self, menuitem, name, parent): parentname = parent.get_name() tagname = ujoin(parentname, name) self.editor.make_tag(tagname) def make_new_tag(self, menuitem, name, parent): parentname = parent.get_name() if (parentname, name) == ('_define_', '_create_'): self.editor.create_new_tag(self.create_entry) def create_entry(self, *args): var = dict(self.editor.dialogs['create'].items()) debug('var is %s' % var) self.environ[self.trait][var['name']] = var['value'] tagname = ujoin(self.trait, var['name']) self.editor.make_tag(tagname) self.editor.destroy_dialog(self.editor.dialogs['create']) def preview(self, *args): subs = self.traitparent.get_superdict(self.environ.keys()) self.editor.buffer.set_subs(subs) self.editor.preview(*args) def set_text(self, text): self.editor.buffer.set_text(text) def get_text(self): return self.editor.get_text()
class EnvironmentEditor(_EnvironmentEditor): def __init__(self, conn, suite, name='EnvironmentEditor'): self.menu = make_menu(['update', 'create'], self.env_command) _EnvironmentEditor.__init__(self, conn, name=name) self.suite = suite self.traitparent = TraitParent(self.conn, self.suite) self.current_trait = None self.reset_rows() w, h = map( int, config.get_list('env_editor_size', section='management_gui')) self.set_size_request(w, h) def reset_rows(self): rows = self.main.select(fields=['trait'], table=ujoin(self.suite, 'traits')) self.set_rows(rows) self.set_row_select(self.trait_selected) def trait_selected(self, listbox, row, column, event): trait = listbox.get_selected_data()[0].trait self.select_trait(trait) def select_trait(self, trait): self.current_trait = trait environ = TraitEnvironment(self.conn, self.suite, trait) self.environ = self.traitparent.get_environment([self.current_trait]) self.sections = dict([(k, v.keys()) for k, v in self.environ]) self.remove_page('Environment') self.record = RecordBox(environ, name=self.current_trait) self.append_page(self.record, 'Environment') self.pages['Environment'].connect('button_press_event', self.box_pressed) #self.menu.connect('populate', self.populate_menu, self.menu) self.record.connect_entries('populate-popup', self.populate_menu, self.menu) def make_new_variable(self, *args): d = self.dialogs['create'] environ = TraitEnvironment(self.conn, self.suite, self.current_trait) environ.update(dict([(d['name'], d['value'])])) def remove_item(self, *args): d = self.dialogs['remove'] name = d.get_selected_data()[0].name environ = TraitEnvironment(self.conn, self.suite, self.current_trait) del environ[name] dialogs.Message('%s deleted' % name) self.destroy_dialog(d) def update_environment(self): environ = TraitEnvironment(self.conn, self.suite, self.current_trait) environ.update(self.record) def env_command(self, menuitem, command): if command == 'update': dialogs.Message('updating %s' % self.current_trait) self.update_environment() elif command == 'create': self.ask_new_entry() else: dialogs.Message('ack')