def __init__(self, parent): """Prepare the groups dialog""" # Load the user interface self.ui = GtkBuilderLoader(get_ui_file('groups.glade')) if not preferences.get(preferences.DETACHED_WINDOWS): self.ui.dialog_groups.set_transient_for(parent) # Restore the saved size and position settings.positions.restore_window_position(self.ui.dialog_groups, SECTION_WINDOW_NAME) # Initialize actions for widget in self.ui.get_objects_by_type(Gtk.Action): # Connect the actions accelerators widget.connect_accelerator() # Set labels widget.set_label(text(widget.get_label())) # Initialize tooltips for widget in self.ui.get_objects_by_type(Gtk.Button): action = widget.get_related_action() if action: widget.set_tooltip_text(action.get_label().replace('_', '')) # Initialize column headers for widget in self.ui.get_objects_by_type(Gtk.TreeViewColumn): widget.set_title(text(widget.get_title())) # Load the groups self.model = ModelGroups(self.ui.store_groups) self.selected_iter = None # Sort the data in the models self.model.model.set_sort_column_id( self.ui.column_name.get_sort_column_id(), Gtk.SortType.ASCENDING) # Connect signals from the glade file to the module functions self.ui.connect_signals(self)
def __init__(self, parent): """Prepare the groups dialog""" # Load the user interface self.ui = GtkBuilderLoader(get_ui_file('groups.glade')) if not preferences.get(preferences.DETACHED_WINDOWS): self.ui.dialog_groups.set_transient_for(parent) # Restore the saved size and position settings.positions.restore_window_position( self.ui.dialog_groups, SECTION_WINDOW_NAME) # Initialize actions for widget in self.ui.get_objects_by_type(Gtk.Action): # Connect the actions accelerators widget.connect_accelerator() # Set labels widget.set_label(text(widget.get_label())) # Initialize tooltips for widget in self.ui.get_objects_by_type(Gtk.Button): action = widget.get_related_action() if action: widget.set_tooltip_text(action.get_label().replace('_', '')) # Initialize column headers for widget in self.ui.get_objects_by_type(Gtk.TreeViewColumn): widget.set_title(text(widget.get_title())) # Load the groups self.model = ModelGroups(self.ui.store_groups) self.selected_iter = None # Sort the data in the models self.model.model.set_sort_column_id( self.ui.column_name.get_sort_column_id(), Gtk.SortType.ASCENDING) # Connect signals from the glade file to the module functions self.ui.connect_signals(self)
def __init__(self, application): self.application = application # Load settings settings.settings = settings.Settings(FILE_SETTINGS, False) settings.positions = settings.Settings(FILE_WINDOWS_POSITION, False) settings.services = settings.Settings(FILE_SERVICES, False) preferences.preferences = preferences.Preferences() # Load services for key in settings.services.get_sections(): model_services.services[key] = ServiceInfo( name=key, description=settings.services.get(key, OPTION_SERVICE_DESCRIPTION), command=settings.services.get(key, OPTION_SERVICE_COMMAND), terminal=settings.services.get_boolean( key, OPTION_SERVICE_TERMINAL), icon=settings.services.get(key, OPTION_SERVICE_ICON)) self.loadUI() self.model_hosts = ModelHosts(self.ui.store_hosts) self.model_groups = ModelGroups(self.ui.store_groups) # Prepare the debug dialog debug.debug = debug.UIDebug(self.ui.win_main, self.on_window_debug_delete_event) # Prepare the processes dialog processes.processes = processes.UIProcesses( self.ui.win_main, self.on_window_processes_delete_event) # Load the groups and hosts list self.hosts = {} self.reload_groups() # Sort the data in the models self.model_groups.model.set_sort_column_id( self.ui.column_group.get_sort_column_id(), Gtk.SortType.ASCENDING) self.model_hosts.model.set_sort_column_id( self.ui.column_name.get_sort_column_id(), Gtk.SortType.ASCENDING) # Automatically select the first host if any self.ui.tvw_groups.set_cursor(0) if self.model_hosts.count() > 0: self.ui.tvw_connections.set_cursor(0) # Restore the saved size and position settings.positions.restore_window_position(self.ui.win_main, SECTION_WINDOW_NAME)
class UIMain(object): def __init__(self, application): self.application = application # Load settings settings.settings = settings.Settings(FILE_SETTINGS, False) settings.positions = settings.Settings(FILE_WINDOWS_POSITION, False) settings.services = settings.Settings(FILE_SERVICES, False) preferences.preferences = preferences.Preferences() # Load services for key in settings.services.get_sections(): model_services.services[key] = ServiceInfo( name=key, description=settings.services.get(key, OPTION_SERVICE_DESCRIPTION), command=settings.services.get(key, OPTION_SERVICE_COMMAND), terminal=settings.services.get_boolean( key, OPTION_SERVICE_TERMINAL), icon=settings.services.get(key, OPTION_SERVICE_ICON)) self.loadUI() self.model_hosts = ModelHosts(self.ui.store_hosts) self.model_groups = ModelGroups(self.ui.store_groups) # Prepare the debug dialog debug.debug = debug.UIDebug(self.ui.win_main, self.on_window_debug_delete_event) # Prepare the processes dialog processes.processes = processes.UIProcesses( self.ui.win_main, self.on_window_processes_delete_event) # Load the groups and hosts list self.hosts = {} self.reload_groups() # Sort the data in the models self.model_groups.model.set_sort_column_id( self.ui.column_group.get_sort_column_id(), Gtk.SortType.ASCENDING) self.model_hosts.model.set_sort_column_id( self.ui.column_name.get_sort_column_id(), Gtk.SortType.ASCENDING) # Automatically select the first host if any self.ui.tvw_groups.set_cursor(0) if self.model_hosts.count() > 0: self.ui.tvw_connections.set_cursor(0) # Restore the saved size and position settings.positions.restore_window_position(self.ui.win_main, SECTION_WINDOW_NAME) def loadUI(self): """Load the interface UI""" self.ui = GtkBuilderLoader(get_ui_file('main.glade')) self.ui.win_main.set_application(self.application) self.ui.win_main.set_title(APP_NAME) # Initialize actions for widget in self.ui.get_objects_by_type(Gtk.Action): # Connect the actions accelerators widget.connect_accelerator() # Set labels widget.set_label(text(widget.get_label())) # Initialize tooltips for widget in self.ui.get_objects_by_type(Gtk.ToolButton): action = widget.get_related_action() if action: widget.set_tooltip_text(action.get_label().replace('_', '')) # Initialize column headers for widget in self.ui.get_objects_by_type(Gtk.TreeViewColumn): widget.set_title(text(widget.get_title())) # Set list items row height icon_size = preferences.ICON_SIZE self.ui.cell_name.props.height = preferences.get(icon_size) self.ui.cell_group_name.props.height = preferences.get(icon_size) # Set groups visibility self.ui.scroll_groups.set_visible( preferences.get(preferences.GROUPS_SHOW)) # Add a Gtk.Headerbar, only for GTK+ 3.10.0 and higher if (not Gtk.check_version(3, 10, 0) and not preferences.get(preferences.HEADERBARS_DISABLE)): self.load_ui_headerbar() if preferences.get(preferences.HEADERBARS_REMOVE_TOOLBAR): # This is only for development, it should always be True # Remove the redundant toolbar self.ui.toolbar_main.destroy() # Flatten the Gtk.ScrolledWindows self.ui.scroll_groups.set_shadow_type(Gtk.ShadowType.NONE) self.ui.scroll_connections.set_shadow_type(Gtk.ShadowType.NONE) # Connect signals from the glade file to the module functions self.ui.connect_signals(self) def load_ui_headerbar(self): """Add a Gtk.HeaderBar to the window with buttons""" def create_button_from_action(action): """Create a new Gtk.Button from a Gtk.Action""" if isinstance(action, Gtk.ToggleAction): new_button = Gtk.ToggleButton() else: new_button = Gtk.Button() new_button.set_use_action_appearance(False) new_button.set_related_action(action) # Use icon from the action icon_name = action.get_icon_name() if preferences.get(preferences.HEADERBARS_SYMBOLIC_ICONS): icon_name += '-symbolic' # Get desired icon size icon_size = (Gtk.IconSize.BUTTON if preferences.get(preferences.HEADERBARS_SMALL_ICONS) else Gtk.IconSize.LARGE_TOOLBAR) new_button.set_image( Gtk.Image.new_from_icon_name(icon_name, icon_size)) # Set the tooltip from the action label new_button.set_tooltip_text(action.get_label().replace('_', '')) return new_button # Add the Gtk.HeaderBar header_bar = Gtk.HeaderBar() header_bar.props.title = self.ui.win_main.get_title() header_bar.set_show_close_button(True) self.ui.win_main.set_titlebar(header_bar) # Add buttons to the left side for action in (self.ui.action_new, self.ui.action_edit, self.ui.action_copy, self.ui.action_connect, self.ui.action_delete): header_bar.pack_start(create_button_from_action(action)) # Add buttons to the right side (in reverse order) for action in reversed((self.ui.action_services, self.ui.action_groups, self.ui.action_debug, self.ui.action_processes, self.ui.action_about)): header_bar.pack_end(create_button_from_action(action)) def run(self): """Show the UI""" self.ui.win_main.show_all() def on_win_main_delete_event(self, widget, event): """Save the settings and close the application""" debug.debug.destroy() processes.processes.destroy() settings.positions.save_window_position(self.ui.win_main, SECTION_WINDOW_NAME) settings.positions.save() settings.services.save() settings.settings.save() self.application.quit() def on_action_about_activate(self, action): """Show the about dialog""" dialog = UIAbout(self.ui.win_main) dialog.show() dialog.destroy() def on_action_quit_activate(self, action): """Close the application by closing the main window""" event = Gdk.Event() event.key.type = Gdk.EventType.DELETE self.ui.win_main.event(event) def on_action_services_activate(self, action): """Edit services""" selected_row = get_treeview_selected_row(self.ui.tvw_connections) if selected_row: iter_parent = self.ui.store_hosts.iter_parent(selected_row) selected_path = self.model_hosts.model[selected_row].path # Get the path of the host if iter_parent is None: tree_path = self.model_hosts.model[selected_row].path else: tree_path = self.model_hosts.model[iter_parent].path expanded = self.ui.tvw_connections.row_expanded(tree_path) dialog_services = UIServices(parent=self.ui.win_main) # Load services list dialog_services.model.load(model_services.services) dialog_services.show() # Get the new services list, clear and store the list again model_services.services = dialog_services.model.dump() dialog_services.destroy() settings.services.clear() for key in model_services.services.iterkeys(): settings.services.set( section=key, option=OPTION_SERVICE_DESCRIPTION, value=model_services.services[key].description) settings.services.set(section=key, option=OPTION_SERVICE_COMMAND, value=model_services.services[key].command) settings.services.set_boolean( section=key, option=OPTION_SERVICE_TERMINAL, value=model_services.services[key].terminal) settings.services.set(section=key, option=OPTION_SERVICE_ICON, value=model_services.services[key].icon) self.reload_hosts() if selected_row: # Automatically expand the row if it was expanded before if expanded: self.ui.tvw_connections.expand_row(tree_path, True) # Automatically select again the previously selected row self.ui.tvw_connections.set_cursor(path=selected_path, column=None, start_editing=False) def reload_hosts(self): """Load hosts from the settings files""" self.model_hosts.clear() self.hosts.clear() hosts_path = self.get_current_group_path() # Fix bug where the groups model isn't yet emptied, resulting in # being still used after a clear, then an invalid path if not os.path.isdir(hosts_path): return for filename in os.listdir(hosts_path): # Skip folders, used for groups if os.path.isdir(os.path.join(hosts_path, filename)): continue debug.add_info('Loading host %s' % os.path.join(hosts_path, filename)) settings_host = settings.Settings(filename=os.path.join( hosts_path, filename), case_sensitive=True) name = settings_host.get(SECTION_HOST, OPTION_HOST_NAME) description = settings_host.get(SECTION_HOST, OPTION_HOST_DESCRIPTION) host = HostInfo(name=name, description=description) destinations = {} # Load host destinations if SECTION_DESTINATIONS in settings_host.get_sections(): for option in settings_host.get_options(SECTION_DESTINATIONS): value = settings_host.get(SECTION_DESTINATIONS, option) destinations[option] = DestinationInfo(name=option, value=value) # Load associations association_index = 1 associations_count = settings_host.get_int( section=SECTION_HOST, option=OPTION_HOST_ASSOCIATIONS) while association_index <= associations_count: section = '%s %d' % (SECTION_ASSOCIATION, association_index) host.add_association( description=settings_host.get( section=section, option=OPTION_ASSOCIATION_DESCRIPTION), destination_name=settings_host.get( section=section, option=OPTION_ASSOCIATION_DESTINATION), service_name=settings_host.get( section=section, option=OPTION_ASSOCIATION_SERVICE), arguments=json.loads( settings_host.get( section=section, option=OPTION_ASSOCIATION_ARGUMENTS))) association_index += 1 self.add_host(host, destinations, False) def add_host(self, host, destinations, update_settings): """Add a new host along as with its destinations""" # Add the host to the data and to the model self.hosts[host.name] = host treeiter = self.model_hosts.add_data(host) # Add the destinations to the data for destination_name in destinations: destination = destinations[destination_name] host.add_destination(item=destination) # Add service associations to the model for association in host.associations: description = association.description service_name = association.service_name service_arguments = json.dumps(association.service_arguments) destination = destinations[association.destination_name] if service_name in model_services.services: service = model_services.services[service_name] self.model_hosts.add_association(treeiter=treeiter, description=description, destination=destination, service=service, arguments=service_arguments) else: debug.add_warning('service %s not found' % service_name) # Update settings file if requested if update_settings: hosts_path = self.get_current_group_path() settings_host = settings.Settings(filename=os.path.join( hosts_path, '%s.conf' % host.name), case_sensitive=True) # Add host information settings_host.set(SECTION_HOST, OPTION_HOST_NAME, host.name) settings_host.set(SECTION_HOST, OPTION_HOST_DESCRIPTION, host.description) # Add destinations for key in host.destinations: destination = host.destinations[key] settings_host.set(section=SECTION_DESTINATIONS, option=destination.name, value=destination.value) association_index = 0 for association in host.associations: arguments = json.dumps(association.service_arguments) # Add associations to the settings association_index += 1 section = '%s %d' % (SECTION_ASSOCIATION, association_index) settings_host.set(section=section, option=OPTION_ASSOCIATION_DESCRIPTION, value=association.description) settings_host.set(section=section, option=OPTION_ASSOCIATION_DESTINATION, value=association.destination_name) settings_host.set(section=section, option=OPTION_ASSOCIATION_SERVICE, value=association.service_name) settings_host.set(section=section, option=OPTION_ASSOCIATION_ARGUMENTS, value=arguments) settings_host.set_int(section=SECTION_HOST, option=OPTION_HOST_ASSOCIATIONS, value=association_index) # Save the settings to the file settings_host.save() def remove_host(self, name): """Remove a host by its name""" hosts_path = self.get_current_group_path() filename = os.path.join(hosts_path, '%s.conf' % name) if os.path.isfile(filename): os.unlink(filename) self.hosts.pop(name) self.model_hosts.remove(self.model_hosts.get_iter(name)) def reload_groups(self): """Load groups from hosts folder""" self.model_groups.clear() # Always add a default group self.model_groups.add_data(GroupInfo('', _('Default group'))) for filename in os.listdir(DIR_HOSTS): if os.path.isdir(os.path.join(DIR_HOSTS, filename)): # For each folder add a new group self.model_groups.add_data(GroupInfo(filename, filename)) def on_action_new_activate(self, action): """Define a new host""" dialog = UIHost(parent=self.ui.win_main, hosts=self.model_hosts) response = dialog.show(default_name='', default_description='', title=_('Add a new host'), treeiter=None) if response == Gtk.ResponseType.OK: destinations = dialog.model_destinations.dump() associations = dialog.model_associations.dump() host = HostInfo(dialog.name, dialog.description) # Set the associations for values in associations: (destination_name, description, service_name, service_arguments) = associations[values] destination = destinations[destination_name] arguments = json.loads(service_arguments) host.add_association(description=description, destination_name=destination_name, service_name=service_name, arguments=arguments) self.add_host(host=host, destinations=destinations, update_settings=True) # Automatically select the newly added host self.ui.tvw_connections.set_cursor( path=self.model_hosts.get_path_by_name(dialog.name), column=None, start_editing=False) dialog.destroy() def on_action_edit_activate(self, action): """Define a new host""" selected_row = get_treeview_selected_row(self.ui.tvw_connections) if selected_row: if self.is_selected_row_host(): # First level (host) name = self.model_hosts.get_key(selected_row) description = self.model_hosts.get_description(selected_row) selected_iter = self.model_hosts.get_iter(name) expanded = self.ui.tvw_connections.row_expanded( self.model_hosts.get_path(selected_iter)) dialog = UIHost(parent=self.ui.win_main, hosts=self.model_hosts) # Restore the destinations for the selected host destinations = self.hosts[name].destinations for destination_name in destinations: destination = destinations[destination_name] dialog.model_destinations.add_data(destination) # Restore the associations for the selected host for association in self.hosts[name].associations: service_name = association.service_name if service_name in model_services.services: dialog.model_associations.add_data( index=dialog.model_associations.count(), name=association.destination_name, description=association.description, service=model_services.services[service_name], arguments=association.service_arguments) else: debug.add_warning('service %s not found' % service_name) # Show the edit host dialog response = dialog.show(default_name=name, default_description=description, title=_('Edit host'), treeiter=selected_iter) if response == Gtk.ResponseType.OK: # Remove older host and add the newer destinations = dialog.model_destinations.dump() associations = dialog.model_associations.dump() host = HostInfo(dialog.name, dialog.description) # Set the associations for values in associations: (destination_name, description, service_name, service_arguments) = associations[values] destination = destinations[destination_name] arguments = json.loads(service_arguments) host.add_association(description=description, destination_name=destination_name, service_name=service_name, arguments=arguments) self.remove_host(name) self.add_host(host=host, destinations=destinations, update_settings=True) # Get the path of the host tree_path = self.model_hosts.get_path_by_name(dialog.name) # Automatically select again the previously selected host self.ui.tvw_connections.set_cursor(path=tree_path, column=None, start_editing=False) # Automatically expand the row if it was expanded before if expanded: self.ui.tvw_connections.expand_row(tree_path, False) def on_tvw_connections_row_activated(self, widget, treepath, column): """Edit the selected row on activation""" selected_row = get_treeview_selected_row(self.ui.tvw_connections) if selected_row and self.is_selected_row_host(): # Start host edit self.ui.action_edit.activate() else: # Connect to the destination self.ui.action_connect.activate() def on_action_delete_activate(self, action): """Remove the selected host""" selected_row = get_treeview_selected_row(self.ui.tvw_connections) if selected_row and show_message_dialog( class_=UIMessageDialogNoYes, parent=self.ui.win_main, message_type=Gtk.MessageType.QUESTION, title=None, msg1=_("Remove host"), msg2=_("Remove the selected host?"), is_response_id=Gtk.ResponseType.YES): self.remove_host(self.model_hosts.get_key(selected_row)) def on_action_copy_activate(self, action): """Copy the selected host to another""" selected_row = get_treeview_selected_row(self.ui.tvw_connections) if selected_row: if self.is_selected_row_host(): # First level (host) name = self.model_hosts.get_key(selected_row) description = self.model_hosts.get_description(selected_row) selected_iter = self.model_hosts.get_iter(name) expanded = self.ui.tvw_connections.row_expanded( self.model_hosts.get_path(selected_iter)) dialog = UIHost(parent=self.ui.win_main, hosts=self.model_hosts) # Restore the destinations for the selected host destinations = self.hosts[name].destinations for destination_name in destinations: destination = destinations[destination_name] dialog.model_destinations.add_data(destination) # Restore the associations for the selected host for association in self.hosts[name].associations: service_name = association.service_name if service_name in model_services.services: dialog.model_associations.add_data( index=dialog.model_associations.count(), name=association.destination_name, description=association.description, service=model_services.services[service_name], arguments=association.service_arguments) else: debug.add_warning('service %s not found' % service_name) # Show the edit host dialog response = dialog.show(default_name=_('Copy of %s') % name, default_description='', title=_('Copy host'), treeiter=None) if response == Gtk.ResponseType.OK: destinations = dialog.model_destinations.dump() associations = dialog.model_associations.dump() host = HostInfo(dialog.name, dialog.description) # Set the associations for values in associations: (destination_name, description, service_name, service_arguments) = associations[values] destination = destinations[destination_name] arguments = json.loads(service_arguments) host.add_association(description=description, destination_name=destination_name, service_name=service_name, arguments=arguments) self.add_host(host=host, destinations=destinations, update_settings=True) # Get the path of the host tree_path = self.model_hosts.get_path_by_name(dialog.name) # Automatically select again the previously selected host self.ui.tvw_connections.set_cursor(path=tree_path, column=None, start_editing=False) # Automatically expand the row if it was expanded before if expanded: self.ui.tvw_connections.expand_row(tree_path, False) # Collapse the duplicated row self.ui.tvw_connections.collapse_row( self.model_hosts.get_path(selected_iter)) def on_tvw_connections_cursor_changed(self, widget): """Set actions sensitiveness for host and connection""" if get_treeview_selected_row(self.ui.tvw_connections): self.ui.actions_connection.set_sensitive( not self.is_selected_row_host()) self.ui.actions_host.set_sensitive(self.is_selected_row_host()) def on_action_debug_toggled(self, action): """Show and hide the debug window""" if self.ui.action_debug.get_active(): debug.debug.show() else: debug.debug.hide() def on_window_debug_delete_event(self, widget, event): """Catch the delete_event in the debug window to hide the window""" self.ui.action_debug.set_active(False) return True def on_action_processes_toggled(self, action): """Show and hide the processes window""" if self.ui.action_processes.get_active(): processes.processes.show() else: processes.processes.hide() def on_window_processes_delete_event(self, widget, event): """Catch the delete_event in the processes window to hide the window""" self.ui.action_processes.set_active(False) return True def on_tvw_connections_key_press_event(self, widget, event): """Expand and collapse nodes with keyboard arrows""" if event.keyval in (Gdk.KEY_Left, Gdk.KEY_Right): # Collapse or expand the selected row using <Left> and <Right> selected_row = get_treeview_selected_row(self.ui.tvw_connections) if (selected_row and self.is_selected_row_host()): if event.keyval == Gdk.KEY_Left: self.ui.action_host_collapse.activate() elif event.keyval == Gdk.KEY_Right: self.ui.action_host_expand.activate() return True def on_action_connect_activate(self, action): """Establish the connection for the destination""" selected_row = get_treeview_selected_row(self.ui.tvw_connections) if selected_row and not self.is_selected_row_host(): host = self.hosts[self.model_hosts.get_key( self.ui.store_hosts.iter_parent(selected_row))] destination_name = self.model_hosts.get_key(selected_row) destination = host.destinations[destination_name] description = self.model_hosts.get_association(selected_row) service_name = self.model_hosts.get_service(selected_row) service_arguments = self.model_hosts.get_arguments(selected_row) arguments = json.loads(service_arguments) association = host.find_association(description=description, destination=destination.name, service=service_name, arguments=arguments) if service_name in model_services.services: service = model_services.services[service_name] command = service.command # Prepares the arguments arguments_map = {} arguments_map['address'] = destination.value for key in association.service_arguments: arguments_map[key] = association.service_arguments[key] # Execute command try: command = command.format(**arguments_map) processes.processes.add_process(host, destination, service, command) except KeyError as error: # An error occurred processing the command error_msg1 = _('Connection open failed') error_msg2 = _('An error occurred processing the ' 'service command.') show_message_dialog(class_=UIMessageDialogClose, parent=self.ui.win_main, message_type=Gtk.MessageType.ERROR, title=None, msg1=error_msg1, msg2=error_msg2, is_response_id=None) debug.add_error(error_msg2) debug.add_error('Host: "%s"' % host.name) debug.add_error('Destination name: "%s"' % destination.name) debug.add_error('Destination value: "%s"' % destination.value) debug.add_error('Service: %s' % service.name), debug.add_error('Command: "%s"' % command) else: debug.add_warning('service %s not found' % service_name) def is_selected_row_host(self): """Return if the currently selected row is an host""" return self.ui.store_hosts.iter_parent( get_treeview_selected_row(self.ui.tvw_connections)) is None def get_current_group_path(self): """Return the path of the currently selected group""" selected_row = get_treeview_selected_row(self.ui.tvw_groups) group_name = self.model_groups.get_key(selected_row) if selected_row \ else '' return os.path.join(DIR_HOSTS, group_name) if group_name else DIR_HOSTS def on_tvw_groups_cursor_changed(self, widget): """Set actions sensitiveness for host and connection""" if get_treeview_selected_row(self.ui.tvw_groups): self.reload_hosts() # Automatically select the first host for the group self.ui.tvw_connections.set_cursor(0) def on_action_groups_activate(self, widget): """Edit groups""" dialog_groups = UIGroups(parent=self.ui.win_main) dialog_groups.model = self.model_groups dialog_groups.ui.tvw_groups.set_model(self.model_groups.model) dialog_groups.show() dialog_groups.destroy() def on_tvw_groups_button_release_event(self, widget, event): """Show groups popup menu on right click""" if event.button == Gdk.BUTTON_SECONDARY: show_popup_menu(self.ui.menu_groups, event.button) def on_tvw_connections_button_release_event(self, widget, event): """Show connections popup menu on right click""" if event.button == Gdk.BUTTON_SECONDARY: show_popup_menu(self.ui.menu_connections, event.button) def on_action_group_previous_activate(self, action): """Move to the previous group""" selected_row = get_treeview_selected_row(self.ui.tvw_groups) new_iter = self.model_groups.model.iter_previous(selected_row) if new_iter: # Select the newly selected row in the groups list new_path = self.model_groups.get_path(new_iter) self.ui.tvw_groups.set_cursor(new_path) def on_action_group_next_activate(self, action): """Move to the next group""" selected_row = get_treeview_selected_row(self.ui.tvw_groups) new_iter = self.model_groups.model.iter_next(selected_row) if new_iter: # Select the newly selected row in the groups list new_path = self.model_groups.get_path(new_iter) self.ui.tvw_groups.set_cursor(new_path) def on_action_host_collapse_activate(self, action): """Collapse the selected host and hide the associations""" selected_row = get_treeview_selected_row(self.ui.tvw_connections) if (selected_row and self.is_selected_row_host()): tree_path = self.model_hosts.get_path(selected_row) if self.ui.tvw_connections.row_expanded(tree_path): self.ui.tvw_connections.collapse_row(tree_path) def on_action_host_expand_activate(self, action): """Expand the selected host and show the associations""" selected_row = get_treeview_selected_row(self.ui.tvw_connections) if (selected_row and self.is_selected_row_host()): tree_path = self.model_hosts.get_path(selected_row) if not self.ui.tvw_connections.row_expanded(tree_path): self.ui.tvw_connections.expand_row(tree_path, False)
class UIGroups(object): def __init__(self, parent): """Prepare the groups dialog""" # Load the user interface self.ui = GtkBuilderLoader(get_ui_file('groups.glade')) if not preferences.get(preferences.DETACHED_WINDOWS): self.ui.dialog_groups.set_transient_for(parent) # Restore the saved size and position settings.positions.restore_window_position(self.ui.dialog_groups, SECTION_WINDOW_NAME) # Initialize actions for widget in self.ui.get_objects_by_type(Gtk.Action): # Connect the actions accelerators widget.connect_accelerator() # Set labels widget.set_label(text(widget.get_label())) # Initialize tooltips for widget in self.ui.get_objects_by_type(Gtk.Button): action = widget.get_related_action() if action: widget.set_tooltip_text(action.get_label().replace('_', '')) # Initialize column headers for widget in self.ui.get_objects_by_type(Gtk.TreeViewColumn): widget.set_title(text(widget.get_title())) # Load the groups self.model = ModelGroups(self.ui.store_groups) self.selected_iter = None # Sort the data in the models self.model.model.set_sort_column_id( self.ui.column_name.get_sort_column_id(), Gtk.SortType.ASCENDING) # Connect signals from the glade file to the module functions self.ui.connect_signals(self) def show(self): """Show the Groups dialog""" self.ui.dialog_groups.run() self.ui.dialog_groups.hide() def destroy(self): """Destroy the Groups dialog""" settings.positions.save_window_position(self.ui.dialog_groups, SECTION_WINDOW_NAME) self.ui.dialog_groups.destroy() self.ui.dialog_groups = None def on_action_add_activate(self, action): """Add a new group""" dialog = UIGroupDetail(self.ui.dialog_groups, self.model) if dialog.show(default_name='', title=_('Add new group'), treeiter=None) == Gtk.ResponseType.OK: os.mkdir(os.path.join(DIR_HOSTS, dialog.name)) self.model.add_data( GroupInfo(name=dialog.name, description=dialog.name)) debug.add_info(_('Added a new group "%s"') % dialog.name) dialog.destroy() def on_action_remove_activate(self, action): """Remove the selected group""" selected_row = get_treeview_selected_row(self.ui.tvw_groups) group_name = self.model.get_key(selected_row) if selected_row else '' if selected_row and group_name and show_message_dialog( class_=UIMessageDialogNoYes, parent=self.ui.dialog_groups, message_type=Gtk.MessageType.WARNING, title=None, msg1=_('Remove the group'), msg2=_('Remove the group «%s»?') % group_name, is_response_id=Gtk.ResponseType.YES): group_path = os.path.join(DIR_HOSTS, group_name) # Check for directory not empty if len(os.listdir(group_path)) and not show_message_dialog( class_=UIMessageDialogNoYes, parent=self.ui.dialog_groups, message_type=Gtk.MessageType.WARNING, title=None, msg1=_('The group is not empty'), msg2='%s\n%s\n\n%s' % ( text('If you delete an item, it will ' 'be permanently lost.'), _('All the hosts defined for the group will be lost.'), _('Are you sure you want to delete the ' 'group «%s»?') % group_name, ), is_response_id=Gtk.ResponseType.YES): # Exit immediately without deleting the group return # Delete all the contained files and the directory for the group for filename in os.listdir(group_path): os.remove(os.path.join(group_path, filename)) os.rmdir(group_path) debug.add_info(_('Removed the group "%s"') % group_name) self.model.remove(selected_row)
class UIGroups(object): def __init__(self, parent): """Prepare the groups dialog""" # Load the user interface self.ui = GtkBuilderLoader(get_ui_file('groups.glade')) if not preferences.get(preferences.DETACHED_WINDOWS): self.ui.dialog_groups.set_transient_for(parent) # Restore the saved size and position settings.positions.restore_window_position( self.ui.dialog_groups, SECTION_WINDOW_NAME) # Initialize actions for widget in self.ui.get_objects_by_type(Gtk.Action): # Connect the actions accelerators widget.connect_accelerator() # Set labels widget.set_label(text(widget.get_label())) # Initialize tooltips for widget in self.ui.get_objects_by_type(Gtk.Button): action = widget.get_related_action() if action: widget.set_tooltip_text(action.get_label().replace('_', '')) # Initialize column headers for widget in self.ui.get_objects_by_type(Gtk.TreeViewColumn): widget.set_title(text(widget.get_title())) # Load the groups self.model = ModelGroups(self.ui.store_groups) self.selected_iter = None # Sort the data in the models self.model.model.set_sort_column_id( self.ui.column_name.get_sort_column_id(), Gtk.SortType.ASCENDING) # Connect signals from the glade file to the module functions self.ui.connect_signals(self) def show(self): """Show the Groups dialog""" self.ui.dialog_groups.run() self.ui.dialog_groups.hide() def destroy(self): """Destroy the Groups dialog""" settings.positions.save_window_position( self.ui.dialog_groups, SECTION_WINDOW_NAME) self.ui.dialog_groups.destroy() self.ui.dialog_groups = None def on_action_add_activate(self, action): """Add a new group""" dialog = UIGroupDetail(self.ui.dialog_groups, self.model) if dialog.show(default_name='', title=_('Add new group'), treeiter=None) == Gtk.ResponseType.OK: os.mkdir(os.path.join(DIR_HOSTS, dialog.name)) self.model.add_data(GroupInfo(name=dialog.name, description=dialog.name)) debug.add_info(_('Added a new group "%s"') % dialog.name) dialog.destroy() def on_action_remove_activate(self, action): """Remove the selected group""" selected_row = get_treeview_selected_row(self.ui.tvw_groups) group_name = self.model.get_key(selected_row) if selected_row else '' if selected_row and group_name and show_message_dialog( class_=UIMessageDialogNoYes, parent=self.ui.dialog_groups, message_type=Gtk.MessageType.WARNING, title=None, msg1=_('Remove the group'), msg2=_('Remove the group «%s»?') % group_name, is_response_id=Gtk.ResponseType.YES): group_path = os.path.join(DIR_HOSTS, group_name) # Check for directory not empty if len(os.listdir(group_path)) and not show_message_dialog( class_=UIMessageDialogNoYes, parent=self.ui.dialog_groups, message_type=Gtk.MessageType.WARNING, title=None, msg1=_('The group is not empty'), msg2='%s\n%s\n\n%s' % ( text('If you delete an item, it will ' 'be permanently lost.'), _('All the hosts defined for the group will be lost.'), _('Are you sure you want to delete the ' 'group «%s»?') % group_name, ), is_response_id=Gtk.ResponseType.YES): # Exit immediately without deleting the group return # Delete all the contained files and the directory for the group for filename in os.listdir(group_path): os.remove(os.path.join(group_path, filename)) os.rmdir(group_path) debug.add_info(_('Removed the group "%s"') % group_name) self.model.remove(selected_row)