def edit_button_press(self): if self.treeview.focus() == "": tkMessageBox.showwarning(None, "No element selected.") else: EditDialog(self, self, self.get_focused_element())