-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain2.py
206 lines (175 loc) · 7.68 KB
/
main2.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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
import sys, os
if hasattr(sys, 'frozen'):
os.environ['PATH'] = sys._MEIPASS + ";" + os.environ['PATH']
from PyQt5 import QtCore
import PyQt5
from PyQt5 import QtWidgets,QtGui
from QTgeneration.MainWindow import Ui_MainWindow
from QTgeneration.AddWindow import Ui_Form
from db_utils.migrate import MyConnection
"""
对demo的UI进行优化，去除冗余元素
"""
class AddWindow(QtWidgets.QWidget):
"""
mainwindow点击add按钮后，弹出add窗口，用于增加记录
"""
def __init__(self, father_MainWindow):
super().__init__()
# Setting
self.Ui_Form = Ui_Form()
self.Ui_Form.setupUi(self)
self.setWindowTitle('Add record')
self.father_MainWindow = father_MainWindow
# SQL
self.my_conn = MyConnection(path='./db_utils/db.db')
# binding
self.Ui_Form.pushButton.clicked.connect(self.on_push_Button_clicked)
self.Ui_Form.pushButton_2.clicked.connect(self.on_push_Button2_clicked)
def on_push_Button_clicked(self):
"""
增加记录，依次传入username，password，place
:return:
"""
self.my_conn.add_record(self.Ui_Form.lineEdit.text(), self.Ui_Form.lineEdit_2.text(),
self.Ui_Form.lineEdit_3.text())
self.father_MainWindow.refresh_record()
def on_push_Button2_clicked(self):
"""
清空输入
:return:
"""
self.Ui_Form.lineEdit.setText(None)
self.Ui_Form.lineEdit_2.setText(None)
self.Ui_Form.lineEdit_3.setText(None)
class MainWindow(QtWidgets.QMainWindow):
def __init__(self):
super().__init__()
# Setting
self.Ui_MainWindow = Ui_MainWindow()
self.Ui_MainWindow.setupUi(self)
self.add_window = AddWindow(self) # add window for adding record
self.Ui_MainWindow.listWidget.hide()
# SQL
self.current_id = -1 # currrent id
self.my_conn = MyConnection(path='./db_utils/db.db')
# binding
self.Ui_MainWindow.pushButton.clicked.connect(self.on_push_button_clicked)
self.Ui_MainWindow.pushButton_2.clicked.connect(self.on_push_button2_clicked)
self.Ui_MainWindow.pushButton_3.clicked.connect(self.on_push_button3_clicked)
self.Ui_MainWindow.listWidget.doubleClicked.connect(self.on_list_Widget_double_clicked)
self.Ui_MainWindow.tableWidget.doubleClicked.connect(self.on_table_Widget_double_clicked)
self.Ui_MainWindow.pushButton_4.clicked.connect(self.on_push_button4_clicked)
# call
self.table_widget_init()
self.show_all()
self.Ui_MainWindow.lineEdit_4.setPlaceholderText("query all if nothing")
def table_widget_init(self):
self.Ui_MainWindow.tableWidget.setColumnCount(4) # 设置table的列数
self.Ui_MainWindow.tableWidget.verticalHeader().setSectionsClickable(False) # 表头不可点击
self.Ui_MainWindow.tableWidget.setHorizontalHeaderLabels(["id", "username", "password", "place"]) # 表头内容
self.Ui_MainWindow.tableWidget.setColumnHidden(0,True) # 隐藏第0列
self.Ui_MainWindow.tableWidget.horizontalHeader().setSectionResizeMode(QtWidgets.QHeaderView.Stretch) # 表宽自动适应
self.Ui_MainWindow.tableWidget.horizontalHeader().setFont(QtGui.QFont("黑体", 12, QtGui.QFont.Bold)) # 表头字体
# self.Ui_MainWindow.tableWidget.item(0,0).setBackground(QtGui.QColor(250,0,0))
# self.Ui_MainWindow.tableWidget.horizontalHeader().setBackgroundRole()
self.Ui_MainWindow.tableWidget.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows) # 一次选一行
self.Ui_MainWindow.tableWidget.setEditTriggers(QtWidgets.QAbstractItemView.NoEditTriggers)
def show_all(self):
"""
将所有记录show在listView，在打开主面板的时候调用
:return:
"""
self.refresh_record()
# self.Ui_MainWindow.listWidget.addItem("asdhasjkhf")
def remove_all_tableW_row(self):
current_row = self.Ui_MainWindow.tableWidget.rowCount()
for r in range(0, current_row)[::-1]: # 倒序是关键
self.Ui_MainWindow.tableWidget.removeRow(r)
def refresh_record(self, place=None):
"""
刷新listView，（可传入条件place）
:param place:
:return:
"""
result = self.my_conn.query_records(place)
self.Ui_MainWindow.listWidget.clear()
self.remove_all_tableW_row()
for item in result:
self.Ui_MainWindow.listWidget.addItem(str(item))
# table widget
current_row = self.Ui_MainWindow.tableWidget.rowCount()
self.Ui_MainWindow.tableWidget.insertRow(current_row)
self.Ui_MainWindow.tableWidget.setItem(current_row, 0, QtWidgets.QTableWidgetItem(str(item[0])))
self.Ui_MainWindow.tableWidget.setItem(current_row, 1, QtWidgets.QTableWidgetItem(item[1]))
self.Ui_MainWindow.tableWidget.setItem(current_row, 2, QtWidgets.QTableWidgetItem(item[2]))
self.Ui_MainWindow.tableWidget.setItem(current_row, 3, QtWidgets.QTableWidgetItem(item[3]))
def on_list_Widget_double_clicked(self, e):
"""
双击listView事件，读取该条记录的信息，并显示在面板上
:param e:
:return:
"""
# print("hello pyqt")
# QtCore.QModelIndex
_data = eval(e.data())
self.current_id = _data[0]
self.Ui_MainWindow.lineEdit.setText(_data[1])
self.Ui_MainWindow.lineEdit_2.setText(_data[2])
self.Ui_MainWindow.lineEdit_3.setText(_data[3])
def on_table_Widget_double_clicked(self, e):
"""
双击listView事件，读取该条记录的信息，并显示在面板上
:param e:
:return:
"""
# print(e.data()) # 点击事件以item为单位（can not row）
selected_row = e.row() # 获取当前点击item所在的行数
# _data = eval(e.data()[1])
self.current_id = self.Ui_MainWindow.tableWidget.item(selected_row, 0).text()
self.Ui_MainWindow.lineEdit.setText(self.Ui_MainWindow.tableWidget.item(selected_row, 1).text())
self.Ui_MainWindow.lineEdit_2.setText(self.Ui_MainWindow.tableWidget.item(selected_row, 2).text())
self.Ui_MainWindow.lineEdit_3.setText(self.Ui_MainWindow.tableWidget.item(selected_row, 3).text())
def on_push_button_clicked(self):
"""
修改当前所显示的信息（先双击，再修改）
:return:
"""
username = self.Ui_MainWindow.lineEdit.text()
password = self.Ui_MainWindow.lineEdit_2.text()
place = self.Ui_MainWindow.lineEdit_3.text()
self.my_conn.modify_record(username, password, place, self.current_id)
self.refresh_record()
def on_push_button3_clicked(self):
"""
查询按钮，可传入条件place
:param place:
:return:
"""
place = self.Ui_MainWindow.lineEdit_4.text()
if place == '':
self.refresh_record(None)
else:
self.refresh_record(place)
# print(result)
def on_push_button2_clicked(self):
"""
删除按钮（先双击，再删除）
:return:
"""
self.my_conn.delete_record(self.current_id)
self.refresh_record()
def on_push_button4_clicked(self):
"""
增加新纪录，弹出增加窗口
:return:
"""
self.add_window.show()
if __name__ == '__main__':
app = QtWidgets.QApplication(sys.argv)
main_window = MainWindow()
# add_window = AddWindow()
# add_window.show()
# main_window.on_push_button3_clicked()
main_window.show()
sys.exit(app.exec_())