PyQt TableView从最后一列的前两行插入计算

时间:2019-01-27 11:09:18

标签: python sqlite pyqt pyside

我有一个简单的Qtableview,用户可以在下面的sqlite数据库中插入日期,我在第3列和第4列使用自定义委托直接插入日期时间。我想在第5列上动态插入,即总时间增量。在下面的图片中我手动插入了01:00小时,但我希望它是自动的。此答案将给出一种方法: Virtual column in QTableView?,并且我之前使用过它,但这会创建一个虚拟列,并且不会将其插入数据库中。 因此,我应该使用QSqlQuery然后将其插入到total列中还是使用tableview本身吗? enter image description here

到目前为止的代码:

from PyQt5.QtWidgets import QMainWindow, QApplication, QStyledItemDelegate, QDateTimeEdit, QHeaderView
from PyQt5.QtCore import pyqtSlot, QDateTime, Qt, QAbstractItemModel,QModelIndex, QDateTime
from datetime import datetime,timedelta

import sys


import essai_find
from essai_find_db import *


class MainWindow(QMainWindow, essai_find.Ui_MainWindow):
    def __init__(self, parent=None):
        super(MainWindow, self).__init__(parent)
        self.setupUi(self)
        self.DB = essaiFindDb()
        self.db_model = QSqlRelationalTableModel()
        self.db_model.setTable('Pilots_exp')
        self.db_model.setEditStrategy(QSqlRelationalTableModel.OnFieldChange)
        self.db_model.setRelation(2, QSqlRelation('Aircraft', 'immatriculation', 'immatriculation'))
        self.db_model.select()
        self.tableView.setModel(self.db_model)
        self.tableView.setColumnHidden(0, True)

        self.tableView.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
        self.custom_delegate = customDelegate()
        self.tableView.setItemDelegateForColumn(3, self.custom_delegate)
        self.tableView.setItemDelegateForColumn(4, self.custom_delegate)

        self.label.setText(str(self.hours_minutes()))

def get_tot_hours(self):
    """select dates from database """
    query1 = QSqlQuery("SELECT date_time1,date_time2 FROM Pilots_exp")
    liste = []
    while query1.next():
        date1 = query1.value(0)
        date2 = query1.value(1)
        essai = datetime.strptime(date2, "%Y/%m/%d %H:%M") - datetime.strptime(date1, "%Y/%m/%d %H:%M")
        liste.append(essai)
    total = sum(liste, timedelta())
    return total

def hours_minutes(self):
    """conversion of time delta get_tot_hours to hours"""
    td = self.get_tot_hours()
    print(td)
    resultat = td.days * 24 + td.seconds // 3600
    print(resultat)
    return resultat

def get_hours_diff(self):
    query1 = QSqlQuery("SELECT date_time1,date_time2 FROM Pilots_exp")
    result = []
    while query1.next():
        date1 = query1.value(0)
        date2 = query1.value(1)
        diff = datetime.strptime(date2, "%Y/%m/%d %H:%M") - datetime.strptime(date1, "%Y/%m/%d %H:%M")
        result.append(str(diff))
    return result


@pyqtSlot()
def on_pushButton_clicked(self):
    self.add_record()

def add_record(self):
    row = self.db_model.rowCount()
    self.db_model.insertRow(row)


class customDelegate(QStyledItemDelegate):
    """DELEGATE INSERT CUSTOM DATEEDIT IN CELL """

def __init__(self, parent=None):
    super(customDelegate, self).__init__(parent)

def createEditor(self, parent, option, index):
    date_edit = QDateTimeEdit(parent)
    date_edit.setDisplayFormat("yyyy/MM/dd HH:mm")
    date_edit.setDateTime(QDateTime.currentDateTime())

    date_edit.setCalendarPopup(True)
    return date_edit

def setModelData(self, editor, model, index):
    value = editor.dateTime().toString("yyyy/MM/dd HH:mm")
    model.setData(index, value)

def setEditorData(self, editor, index):
    value = index.model().data(index, Qt.EditRole)
    qdate = QDateTime().fromString(value, "yyyy/MM/dd HH:mm")
    editor.setDateTime(qdate)


if __name__ == '__main__':
    try:
        app = QApplication(sys.argv)
        form = MainWindow()
        # app.setStyleSheet(qdarkstyle.load_stylesheet_pyqt5())
        form.show()
        app.exec_()
        sys.exit(0)
    except NameError:
        print("Name Error: ", sys.exc_info()[1])
    except SystemExit:
        print("Closing Window....")
    except Exception:
        print(sys.exc_info()[1])

数据库很简单:

from PyQt5.QtSql import *


class essaiFindDb():
def __init__(self):
    self.db = QSqlDatabase.addDatabase("QSQLITE")
    self.db.setDatabaseName("essai_find_database.db")

    self.db.open()

    query = QSqlQuery()
    query.exec_('''CREATE TABLE Pilots_exp(id INTEGER PRIMARY KEY UNIQUE , pilot_1 TEXT,aircraft TEXT,
     date_time1 TEXT, date_time2 TEXT, total TEXT)''')

    query.exec_(
        '''CREATE TABLE Aircraft(id INTEGER PRIMARY KEY, immatriculation TEXT)''')

    query.exec_()

    self.db.commit()
    self.db.close()


if __name__ == '__main__':
    essaiFindDb()

0 个答案:

没有答案