def on_add(self):
dialog = attrib_editor.create_dialog(self.manager.object_name())
if not dialog:
return False
if dialog.exec_() == QtGui.QDialog.Accepted:
data = dialog.get_api_data()
if self.model.add_object(data):
return True
else:
return False
def on_edit(self):
rows = self.selected_rows()
if not rows or len(rows) > 1:
QtGui.QMessageBox.about(None, u"选择错误", u"请选择一个")
return False
dialog = attrib_editor.create_dialog(self.manager.object_name(), self)
if not dialog:
return False
obj = self.model.get_object(rows[0])
obj.encode()
dialog.set_data(obj.data)
if dialog.exec_() == QtGui.QDialog.Accepted:
data = dialog.get_api_data()
self.model.update_one(rows[0], data)
return True
