def onAccept(self): if len(self.editID.text()) and " " not in list(self.editID.text()): item = Item() item.name = self.editName.text() item.idItem = self.editID.text() data = Database() try: data.insertItem(item) except sqlite3.IntegrityError: msg = QtWidgets.QMessageBox() msg.setIcon(QtWidgets.QMessageBox.Warning) msg.setText("ID này đã tồn tại !!") msg.setInformativeText("Xin hãy chọn một ID khác và thử lại !!") msg.setStandardButtons(QtWidgets.QMessageBox.Ok) msg.exec() self.isAccept = -1 return self.isAccept = 1 else: msg = QtWidgets.QMessageBox() msg.setIcon(QtWidgets.QMessageBox.Warning) msg.setText("ID này đã tồn tại !!") msg.setInformativeText("Xin hãy chọn một ID khác và thử lại !!") msg.setStandardButton(QtWidgets.QMessageBox.Ok) exec() self.isAccept = -1
def onAddItem(self): date = datetime.datetime.now() newId = "IT-" + date.strftime("%d%m%Y%H%M%S") newItem = Item(idItem=newId) data = Database() while True: try: data.insertItem(newItem) except sqlite3.IntegrityError: date = date + datetime.timedelta(0, 1) newId = "IT-" + date.strftime("%d%m%Y%H%M%S") newItem.idItem = newId else: break data.closeDatabase() self.addItemIntoTable(newItem)