from PyQt5.QtCore import Qt
from PyQt5.QtGui import QPalette, QBrush, QPixmap
from PyQt5.QtWidgets import QWidget, QLabel, QPushButton, QVBoxLayout, QApplication, QHBoxLayout, \
     QFormLayout, QLineEdit, QRadioButton, QFileDialog
from PyQt5 import QtGui
import sys
from local_calculation_probability import localfile


class local_confirm(QWidget):
    def __init__(self):
        super(local_confirm, self).__init__()
        self.resize(1240, 732)
        self.iniUI()
        palette = QPalette()
        pix = QPixmap("background.jpg")
        pix = pix.scaled(1920, 1080)
        palette.setBrush(QPalette.Background, QBrush(pix))
        self.setPalette(palette)
        self.setAutoFillBackground(True)

    def iniUI(self):
        font = QtGui.QFont()
        font.setFamily('微软雅黑')
        font.setBold(True)
        formlagout = QFormLayout()
        self.edt = QLineEdit()
        self.edt.setFixedSize(500, 40)
        self.edt.setObjectName("editline")
        self.edt.setText("")

        self.local_edt = QLineEdit()
        self.local_edt.setFixedSize(500, 40)
        self.local_edt.setObjectName("editline")
        hbotton = QHBoxLayout()
        self.submit_btn = QPushButton("确定")
        self.submit_btn.setObjectName("mitcancel")
        self.cancel_btn = QPushButton("取消")
        self.cancel_btn.setObjectName("mitcancel")
        self.submit_btn.setFixedSize(200, 50)
        self.cancel_btn.setFixedSize(200, 50)
        hbotton.addWidget(self.submit_btn)
        hbotton.addWidget(self.cancel_btn)
        hbotton.setAlignment(Qt.AlignCenter)
        # 添加到表单布局里
        v_box = QHBoxLayout()
        self.allset = QRadioButton("文件夹")
        self.allset.toggled.connect(self.changeradio)
        self.allset.setChecked(True)
        self.image = QRadioButton("文件")
        self.image.toggled.connect(self.changeradio)

        self.allset.setObjectName('radiobutton')
        self.image.setObjectName('radiobutton')
        d = QLabel("上传文件类型:")
        d.setObjectName("textsize")
        v_box.addStretch()
        v_box.addWidget(d)
        v_box.addWidget(self.allset)
        v_box.addWidget(self.image)
        v_box.addStretch()
        ab = QVBoxLayout(self)


        formlagout.setLabelAlignment(Qt.AlignCenter)
        self.choice2 = QPushButton("选择")
        self.choice2.clicked.connect(self.mfd)

        self.choice2.setFixedSize(100, 40)
        self.choice2.setObjectName("choice")
        formlagout1 = QFormLayout()

        bbb = QHBoxLayout()
        local_lable = QLabel("  本地数据集:")
        local_lable.setObjectName("textsize")
        formlagout1.addRow(local_lable, self.local_edt)
        bbb.addStretch(1)
        bbb.addLayout(formlagout1)
        bbb.addWidget(self.choice2)
        bbb.setAlignment(Qt.AlignCenter)
        bbb.addStretch(1)

        ab.addLayout(v_box)
        ab.addLayout(bbb)
        ab.addLayout(hbotton)
        ab.setAlignment(Qt.AlignCenter)
        self.submit_btn.clicked.connect(self.operation)
        self.alc = localfile()

    def operation(self):
        self.resualt = self.alc.infomation(self.local_edt.text())
        return self.resualt

    def changeradio(self):
        try:
            if self.allset.isChecked() == True:
                self.choice2.disconnect()
                self.choice2.clicked.connect(self.mfd)
            elif self.image.isChecked() == True:
                self.choice2.disconnect()
                self.choice2.clicked.connect(self.msg)
        except:
            pass

    def mfd(self):
        directory1 = QFileDialog.getExistingDirectory(self,
                                                      "选取文件夹",
                                                      "./")  # 起始路径
        self.local_edt.setText(directory1)

    def msg(self):
        fileName1, filetype = QFileDialog.getOpenFileName(self,
                                                          "选取文件",
                                                          "./",
                                                          "All Files (*);;Text Files (*.txt)")  # 设置文件扩展名过滤,注意用双分号间隔
        self.local_edt.setText(fileName1)


if __name__ == "__main__":
    app = QApplication(sys.argv)
    app.setStyleSheet(open("./UnFrameStyle.qss").read())
    windous = local_confirm()
    windous.show()
    sys.exit(app.exec_())