/
hospital_display.py
58 lines (47 loc) · 1.9 KB
/
hospital_display.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
import sys
from PyQt5.QtSql import QSqlDatabase, QSqlQuery, QSqlTableModel, QSqlQueryModel
from PyQt5.QtCore import *
from PyQt5.QtWidgets import *
import re
class HospitalData(QWidget):
def __init__(self, parent=None):
super(HospitalData, self).__init__(parent)
# Declare Database Connections
self.db = None
# Layout Manager
self.layout = QVBoxLayout()
# Query Model
self.queryModel = QSqlQueryModel()
# Table View
self.tableView = QTableView()
self.tableView.setModel(self.queryModel)
self.initUI()
self.initializedModel()
def initUI(self):
self.tableView.horizontalHeader().setStretchLastSection(True)
self.tableView.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
self.layout.addWidget(self.tableView)
self.setLayout(self.layout)
self.setWindowTitle("HospitalData")
self.resize(1300, 600)
def initializedModel(self):
self.db = QSqlDatabase.addDatabase("QSQLITE")
self.db.setDatabaseName("project.db")
if not self.db.open():
return False
self.queryModel.setHeaderData(0, Qt.Horizontal, "Hospital Name")
self.queryModel.setHeaderData(1, Qt.Horizontal, "Address")
self.queryModel.setHeaderData(3, Qt.Horizontal, "City")
self.queryModel.setHeaderData(3, Qt.Horizontal, "State")
self.queryModel.setHeaderData(3, Qt.Horizontal, "Contact Details")
# Get all the records of the table
sql = "SELECT * FROM hospital"
self.queryModel.setQuery(sql, self.db)
self.totalRecordCount = self.queryModel.rowCount()
def closeEvent(self,event):
self.db.close()
if __name__ == "__main__":
app = QApplication(sys.argv)
window = HospitalData()
window.show()
sys.exit(app.exec_())