#導入包
import pymysql
from functools import partial
from PyQt5.Qt import QWidget
from PyQt5 import QtGui,QtWidgets
from PyQt5.QtCore import Qt
from PyQt5.QtWidgets import (QFrame,QApplication,QDialog, QDialogButtonBox,
 QMessageBox,QVBoxLayout, QLineEdit,QTableWidgetItem,QTableWidget,QHBoxLayout)

#建立界面類
class creat_view(QDialog):
 def __init__(self,parent = None):
 super(creat_view,self).__init__(parent)

 #設置界面大小、名稱、背景
 self.resize(1000,800)
 self.setWindowTitle('Database')
 self.setStyleSheet("background-image:url(tubiao_meitu.jpg)")

 #窗體屬性
 self.setWindowFlags(Qt.Widget)


 #連接數(shù)據(jù)庫
 db = pymysql.connect("localhost", "root", "password", "mysql",charset='utf8')
 #獲取游標、數(shù)據(jù)
 cur = db.cursor()
 cur.execute("SELECT * FROM pm_25")
 data = cur.fetchall()

 #數(shù)據(jù)列名
 col_lst = [tup[0] for tup in cur.description]

 #數(shù)據(jù)的大小
 row = len(data)
 vol = len(data[0])


 #插入表格
 self.MyTable = QTableWidget(row,vol)
 font = QtGui.QFont('微軟雅黑',10)

 #設置字體、表頭
 self.MyTable.horizontalHeader().setFont(font)
 self.MyTable.setHorizontalHeaderLabels(col_lst)
 #設置豎直方向表頭不可見
 self.MyTable.verticalHeader().setVisible(False)
 self.MyTable.setFrameShape(QFrame.NoFrame)
#設置表格顏色 self.MyTable.horizontalHeader().setStyleSheet('QHeaderView::section{background:skyblue}')

 #構建表格插入數(shù)據(jù)
 for i in range(row):
 for j in range(vol):
 temp_data = data[i][j] # 臨時記錄，不能直接插入表格
 data1 = QTableWidgetItem(str(temp_data)) # 轉換后可插入表格
 self.MyTable.setItem(i, j, data1)


 #編輯按鈕
 self.qle = QLineEdit()
 buttonBox = QDialogButtonBox()
 #增刪查改四個按鈕
 addButton = buttonBox.addButton("&ADD",QDialogButtonBox.ActionRole)
 okButton = buttonBox.addButton("&OK",QDialogButtonBox.ActionRole)
 deleteButton = buttonBox.addButton("&DELETE",QDialogButtonBox.ActionRole)
 inquireButton = buttonBox.addButton("&QUERY",QDialogButtonBox.ActionRole)

 #設置按鈕內(nèi)字體樣式
 addButton.setFont(font)
 okButton.setFont(font)
 deleteButton.setFont(font)
 inquireButton.setFont(font)

 #垂直布局
 layout = QVBoxLayout()
 layout.addWidget(self.qle)
 layout.addWidget(buttonBox)
 layout.addWidget(self.MyTable)
 self.setLayout(layout)

 addButton.clicked.connect(partial(self.add_data,cur,db))#插入實現(xiàn)
 okButton.clicked.connect(partial(self.up_data, cur, db,col_lst))#插入實現(xiàn)
 deleteButton.clicked.connect(partial(self.del_data,cur,db))#刪除實現(xiàn)
 inquireButton.clicked.connect(partial(self.inq_data,db))#查詢實現(xiàn)

 #添加空表格
 def add_data(self,cur,db):
 #獲取行數(shù)
 row = self.MyTable.rowCount()
 #在末尾插入一空行
 self.MyTable.insertRow(row)


 #插入數(shù)據(jù)
 def up_data(self,cur,db,col_lst):
 row_1 = self.MyTable.rowCount()

 value_lst = []
 for i in range(len(col_lst)):
 if(len(self.MyTable.item(row_1-1,i).text())==0):
 value_lst.append(None)
 else:
 value_lst.append(self.MyTable.item(row_1-1,i).text())

 tup_va_lst = []
 for cl,va in zip(col_lst,value_lst):
 tup_va_lst.append((cl,va))

 #插入語句
 cur.execute(
 "INSERT INTO pm_25 VALUES (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)",value_lst)



 #刪除
 def del_data(self,cur,db):
 #是否刪除的對話框
 reply = QMessageBox.question(self, 'Message', 'Are you sure to delete it ?', QMessageBox.Yes | QMessageBox.No,
   QMessageBox.No)
 if reply == QMessageBox.Yes:
 #當前行
 row_2 = self.MyTable.currentRow()
 del_d = self.MyTable.item(row_2, 0).text()

 #在數(shù)據(jù)庫刪除數(shù)據(jù)
 cur.execute("DELETE FROM pm_25 WHERE f_id = '"+del_d+"'")
 db.commit()

 #刪除表格
 self.MyTable.removeRow(row_2)

 #查詢
 def inq_data(self,db):
 txt = self.qle.text()

 #模糊查詢
 if len(txt) != 0:
 cur.execute("SELECT * FROM pm25_fn WHERE f_area LIKE '%"+txt+"%' or f_place LIKE '%"+txt+"%'")# CONCAT('f_id','f_area','f_place','f_AQI','f_AQItype','f_PM25per1h'),concat(concat('%','#txt'),'%')

 data_x = cur.fetchall()

 self.MyTable.clearContents()

 row_4 = len(data_x)
 vol_1 = len(cur.description)

 #查詢到的更新帶表格當中
 for i_x in range(row_4):
 for j_y in range(vol_1):
  temp_data_1 = data_x[i_x][j_y] # 臨時記錄，不能直接插入表格
  data_1 = QTableWidgetItem(str(temp_data_1)) # 轉換后可插入表格
  self.MyTable.setItem(i_x, j_y, data_1)

 #空輸入返回原先數(shù)據(jù)表格
 else:
 self.MyTable.clearContents()
 cur.execute("SELECT * FROM pm_25")
 data_y = cur.fetchall()

 row_5 = len(data_y)
 vol_1 = len(cur.description)

 for i_x_1 in range(row_5):
 for j_y_1 in range(vol_1):
  temp_data_2 = data_y[i_x_1][j_y_1] # 臨時記錄，不能直接插入表格
  data_2 = QTableWidgetItem(str(temp_data_2)) # 轉換后可插入表格
  self.MyTable.setItem(i_x_1, j_y_1, data_2)

def main():
 #顯示
 app = QApplication(sys.argv)

 c = creat_view()
 c.show()

 sys.exit(app.exec_())

main()