-- coding: utf-8 --

"""
Module implementing MainWindow.
"""

from PyQt5.QtCore import pyqtSlot
from PyQt5.QtWidgets import QMainWindow, QApplication

from PyQt5 import QtCore, QtGui, QtWidgets

class Ui_MainWindow(object):
def setupUi(self, MainWindow):
MainWindow.setObjectName("MainWindow")
MainWindow.resize(800, 600)
self.centralWidget = QtWidgets.QWidget(MainWindow)
self.centralWidget.setObjectName("centralWidget")
self.pushButton = QtWidgets.QPushButton(self.centralWidget)
self.pushButton.setGeometry(QtCore.QRect(680, 60, 75, 23))
self.pushButton.setObjectName("pushButton")
self.pushButton_2 = QtWidgets.QPushButton(self.centralWidget)
self.pushButton_2.setGeometry(QtCore.QRect(680, 170, 75, 23))
self.pushButton_2.setObjectName("pushButton_2")
self.pandastablewidget = DataTableWidget(self.centralWidget)
self.pandastablewidget.setGeometry(QtCore.QRect(10, 30, 591, 331))
self.pandastablewidget.setStyleSheet("")
self.pandastablewidget.setObjectName("pandastablewidget")
MainWindow.setCentralWidget(self.centralWidget)

    self.retranslateUi(MainWindow)
    QtCore.QMetaObject.connectSlotsByName(MainWindow)

def retranslateUi(self, MainWindow):
    _translate = QtCore.QCoreApplication.translate
    MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))
    self.pushButton.setText(_translate("MainWindow", "数据初始化"))
    self.pushButton_2.setText(_translate("MainWindow", "保存数据"))

from qtpandas.views.DataTableView import DataTableWidget

from qtpandas.models.DataFrameModel import DataFrameModel
import pandas as pd

class MainWindow(QMainWindow, Ui_MainWindow):
"""
Class documentation goes here.
"""

def __init__(self, parent=None):
    """
    Constructor

    @param parent reference to the parent widget
    @type QWidget
    """
    super(MainWindow, self).__init__(parent)
    self.setupUi(self)

    '''初始化pandasqt'''
    widget = self.pandastablewidget
    widget.resize(600, 500)  # 如果对部件尺寸大小不满意可以在这里设置

    self.model = DataFrameModel()  # 设置新的模型
    widget.setViewModel(self.model)

    self.df = pd.read_excel(r'./data/fund_data.xlsx', encoding='gbk')
    self.df_original = self.df.copy()  # 备份原始数据
    self.model.setDataFrame(self.df)

@pyqtSlot()
def on_pushButton_clicked(self):
    """
    初始化pandas
    """
    self.model.setDataFrame(self.df_original)

@pyqtSlot()
def on_pushButton_2_clicked(self):
    """
    保存数据
    """
    self.df.to_excel(r'./data/fund_data_new.xlsx')

if name == "main":
import sys

app = QApplication(sys.argv)
ui = MainWindow()
ui.show()
sys.exit(app.exec_())