189 8069 5689
from PyQt5.QtWidgets import QWidget,QLabel,QLineEdit,QGroupBox,QGridLayout,QVBoxLayout,QPushButton,QMessageBox
import pickle,pymssql,os,threading,time
#定义数据库窗口
class db_window(QWidget):
def __init__(self):
super().__init__()
self.db_server = QLineEdit()
self.db_port = QLineEdit()
self.db_user = QLineEdit()
self.db_password = QLineEdit()
self.dir_db = {}
#初始化数据库链接信息
self.db_test_flag = -1
self.dir_test_db = {}
self.initUI()
self.initDB()
def initUI(self):
#初始化页面
db_server_label = QLabel('数据库地址:')
db_port_label = QLabel('端口:')
db_user_label = QLabel('数据库账号:')
db_password_label = QLabel('数据库密码:')
self.db_password.setEchoMode(QLineEdit.Password)
db_test_button = QPushButton('测试数据库链接')
db_input_button = QPushButton('保存数据库链接')
db_test_button.clicked.connect(self.test_db)
db_input_button.clicked.connect(self.save_db)
db_dbinfo_group = QGroupBox('MSSQL数据库')
db_mssql_grid = QGridLayout()
db_mssql_grid.addWidget(db_server_label,0,0)
db_mssql_grid.addWidget(db_port_label,0,1)
db_mssql_grid.addWidget(db_user_label,0,2)
db_mssql_grid.addWidget(db_password_label,0,3)
db_mssql_grid.addWidget(db_test_button,0,4)
db_mssql_grid.addWidget(self.db_server,1,0)
db_mssql_grid.addWidget(self.db_port,1,1)
db_mssql_grid.addWidget(self.db_user,1,2)
db_mssql_grid.addWidget(self.db_password,1,3)
db_mssql_grid.addWidget(db_input_button,1,4)
db_dbinfo_group.setLayout(db_mssql_grid)
vbox = QVBoxLayout()
vbox.addWidget(db_dbinfo_group)
self.test_label = QLabel('')
vbox.addWidget(self.test_label)
vbox.setStretchFactor(db_dbinfo_group,3)
vbox.setStretchFactor(self.test_label,7)
self.setLayout(vbox)
#初始化数据库链接信息
def initDB(self):
try:
if os.path.exists('.//DATA//DBinfo.pik'):
with open('.//DATA//DBinfo.pik','rb') as db_file:
self.dir_db = pickle.load(db_file)
for line in self.dir_db.keys():
if line == 'server':
self.db_server.setText(self.dir_db['server'])
elif line == 'port':
self.db_port.setText(str(self.dir_db['port']))
elif line == 'user':
self.db_user.setText(self.dir_db['user'])
elif line == 'password':
self.db_password.setText(self.dir_db['password'])
except Exception as error:
return
#数据库测试按钮
def test_db(self):
self.dir_test_db.clear()
if self.db_server.text().strip() != '':
self.dir_test_db['server'] = self.db_server.text().strip()
if self.db_port.text().strip() != '':
try:
port = int(self.db_port.text().strip())
except Exception as error:
QMessageBox.warning(self,'端口错误','服务器端口是数字,您输入错误.')
return
self.dir_test_db['port'] = port
if self.db_user.text().strip() != '':
self.dir_test_db['user'] = self.db_user.text().strip()
if self.db_password.text().strip() != '':
self.dir_test_db['password'] = self.db_password.text().strip()
try:
conn = pymssql.connect(**self.dir_test_db)
except Exception as error:
QMessageBox.warning(self,'错误','无法链接数据库.')
return
self.test_label.setText('')
QMessageBox.about(self,'链接成功','成功链接数据库.')
self.db_test_flag = 1
conn.close()
#数据库保存按钮
def save_db(self):
if self.db_test_flag < 0:
QMessageBox.warning(self,'提示','保存数据库信息之前,需要先测试数据库链接.\n数据库链接正常之后,才能保存信息.')
return
self.dir_db = self.dir_test_db.copy()
with open('.//DATA//DBinfo.pik','wb') as db_file:
pickle.dump(self.dir_db, db_file)
self.db_test_flag = -1
QMessageBox.about(self,'提示','保存信息成功.')