def __init__(self):
QWidget.__init__(self)
# Set up the user interface from Designer.
self.ui = Ui_AccessWidget()
self.ui.setupUi(self)
header = self.ui.treeWidget.header()
header.resizeSection(0, 60)
header.resizeSection(3, 130)
self.ui.treeWidget.doubleClicked.connect(self.edit)
self.ui.addButton.clicked.connect(self.new)
self.ui.editButton.clicked.connect(self.edit)
self.ui.deleteButton.clicked.connect(self.destroy)
self.formDialog = QDialog(self)
self.formDialog.ui = Ui_FormDialog()
self.formDialog.ui.setupUi(self.formDialog)
self.formDialog.ui.buttonBox.accepted.connect(self.commit)
self.formDialog.resize(250, 200)
self.userRoleMapper = {
u"用户": "staff",
u"审计员": "audit",
u"管理员": "admin",
}
self.userRoleReverseMapper = {}
for key, value in self.userRoleMapper.iteritems():
self.userRoleReverseMapper[value] = key
self.userFormOptions = [
{'title': u"用户名", 'type': String},
{'title': u"密码", 'type': String},
{'title': u"确认密码", 'type': String},
{'title': u"用户角色", 'type': List, 'list': self.userRoleMapper.keys()},
]
generateForm(self.formDialog.ui.formLayout, self.userFormOptions)
class AccessController(QWidget):
def __init__(self):
QWidget.__init__(self)
# Set up the user interface from Designer.
self.ui = Ui_AccessWidget()
self.ui.setupUi(self)
header = self.ui.treeWidget.header()
header.resizeSection(0, 60)
header.resizeSection(3, 130)
self.ui.treeWidget.doubleClicked.connect(self.edit)
self.ui.addButton.clicked.connect(self.new)
self.ui.editButton.clicked.connect(self.edit)
self.ui.deleteButton.clicked.connect(self.destroy)
self.formDialog = QDialog(self)
self.formDialog.ui = Ui_FormDialog()
self.formDialog.ui.setupUi(self.formDialog)
self.formDialog.ui.buttonBox.accepted.connect(self.commit)
self.formDialog.resize(250, 200)
self.userRoleMapper = {
u"用户": "staff",
u"审计员": "audit",
u"管理员": "admin",
}
self.userRoleReverseMapper = {}
for key, value in self.userRoleMapper.iteritems():
self.userRoleReverseMapper[value] = key
self.userFormOptions = [
{'title': u"用户名", 'type': String},
{'title': u"密码", 'type': String},
{'title': u"确认密码", 'type': String},
{'title': u"用户角色", 'type': List, 'list': self.userRoleMapper.keys()},
]
generateForm(self.formDialog.ui.formLayout, self.userFormOptions)
def show(self):
self.updateUserList()
QWidget.show(self)
def new(self):
setFormValue(self.userFormOptions, {})
self.user = None
self.formDialog.setWindowTitle(u"创建用户")
self.formDialog.show()
def edit(self):
if not self.currentSelectUser(): return
formValue = {u"用户名": self.user.username, u"用户角色": self.user.roles}
setFormValue(self.userFormOptions, formValue)
self.formDialog.setWindowTitle(u"编辑用户")
self.formDialog.show()
def commit(self):
if self.user:
self.update()
else:
self.user = User()
self.create()
def create(self):
self.parseFormValue()
# save return True on success, False on failed
if self.user.save():
self.updateUserList()
QDialog.accept(self.formDialog)
else:
QMessageBox.information(self, u"创建用户失败", self.user.full_error_messages())
return
def update(self):
self.parseFormValue()
# save return True on success, False on failed
if self.user.save():
self.updateUserList()
QDialog.accept(self.formDialog)
else:
QMessageBox.information(self, u"编辑用户失败", self.user.full_error_messages())
return
def destroy(self):
if not self.currentSelectUser(): return
if QMessageBox.question(self, u"删除用户", u"您确定要删除用户%s？" % self.user.username,
QMessageBox.Ok | QMessageBox.Cancel) == QMessageBox.Ok:
self.user.destroy()
self.updateUserList()
def parseFormValue(self):
formValue = getFormValue(self.userFormOptions)
# set user fields
self.user.username = formValue[u"用户名"]
if formValue[u"密码"] or formValue[u"确认密码"]:
self.user.password = formValue[u"密码"]
self.user.password_confirm = formValue[u"确认密码"]
self.user.roles = self.userRoleMapper[formValue[u"用户角色"]]
def updateUserList(self):
self.user = None
self.allItemMapper = {}
self.ui.treeWidget.clear()
for user in User.all():
stringList = QStringList([
unicode(user.id),
unicode(user.username),
", ".join([self.userRoleReverseMapper[role] for role in user.roles]),
user.created_at.strftime('%Y-%m-%d %H:%M'),
user.updated_at.strftime('%Y-%m-%d %H:%M'),
])
item = QTreeWidgetItem(self.ui.treeWidget, stringList)
self.allItemMapper[item] = user
def currentSelectUser(self):
item = self.ui.treeWidget.currentItem()
if not item:
QMessageBox.information(self, u"未选中用户", u"请先选中一个用户！")
return None
self.user = self.allItemMapper[item]
return self.user