# -*- coding: utf-8 -*- """ *************************************************************************** dlg_field_properties.py --------------------- Date : April 2012 Copyright : (C) 2012 by Giuseppe Sucameli Email : brush dot tyler at gmail dot com *************************************************************************** * * * This program is free software; you can redistribute it and/or modify * * it under the terms of the GNU General Public License as published by * * the Free Software Foundation; either version 2 of the License, or * * (at your option) any later version. * * * *************************************************************************** """ __author__ = 'Giuseppe Sucameli' __date__ = 'April 2012' __copyright__ = '(C) 2012, Giuseppe Sucameli' # This will get replaced with a git SHA1 when you do a git archive __revision__ = '$Format:%H$' from PyQt4.QtCore import * from PyQt4.QtGui import * from .db_plugins.plugin import TableField from .ui.ui_DlgFieldProperties import Ui_DbManagerDlgFieldProperties as Ui_Dialog class DlgFieldProperties(QDialog, Ui_Dialog): def __init__(self, parent=None, fld=None, table=None, db=None): QDialog.__init__(self, parent) self.fld = fld self.table = self.fld.table() if self.fld and self.fld.table() else table self.db = self.table.database() if self.table and self.table.database() else db self.setupUi(self) for item in self.db.connector.fieldTypes(): self.cboType.addItem(item) self.setField(self.fld) self.connect(self.buttonBox, SIGNAL("accepted()"), self.onOK) def setField(self, fld): if fld == None: return self.editName.setText(fld.name) self.cboType.setEditText(fld.dataType) if fld.modifier: self.editLength.setText(unicode(fld.modifier)) self.chkNull.setChecked(not fld.notNull) if fld.hasDefault: self.editDefault.setText(fld.default) def getField(self, newCopy=False): fld = TableField(self.table) if not self.fld or newCopy else self.fld fld.name = self.editName.text() fld.dataType = self.cboType.currentText() fld.notNull = not self.chkNull.isChecked() fld.default = self.editDefault.text() fld.hasDefault = fld.default != "" try: modifier = int(self.editLength.text()) except ValueError: ok = False else: ok = True fld.modifier = modifier if ok else None return fld def onOK(self): """ first check whether everything's fine """ fld = self.getField(True) # don't change the original copy if fld.name == "": QMessageBox.critical(self, self.tr("Sorry"), self.tr("field name must not be empty")) return if fld.dataType == "": QMessageBox.critical(self, self.tr("Sorry"), self.tr("field type must not be empty")) return self.accept()