from Pcap2CsvWidgetUI import *
from PyQt5.QtWidgets import QApplication, QMainWindow
from PyQt5.QtWidgets import QFileDialog
from PyQt5.QtCore import QRegExp
from PyQt5.QtGui import QIntValidator, QRegExpValidator
import os


class MainWidget(QMainWindow, Ui_MainWindow):
    def __init__(self, parent=None):
        super(MainWidget, self).__init__(parent)
        self.setupUi(self)
        self.fileNames = []  # 打开的文件的名称
        self.fileNamePaths = []  # 打开文件的路径

        # 信号注册
        self.chooseFileButton.clicked.connect(self.Choose_Files)
        self.addItemButton.clicked.connect(self.Add_Field_Item)
        self.removeItemButton.clicked.connect(self.Remove_Field_Item)
        self.generateScriptButton.clicked.connect(self.Generate_Script)

    # 生成脚本数据
    def Generate_Script(self):
        print(self.fieldsListWidget.count())

    def Add_Field_Item(self):
        self.fieldsListWidget.addItem("(Default)")

    def Remove_Field_Item(self, index):
        selectedIndex = index.row()

    def Load_Config(self):
        if self.Load_User_Config() is not True:
            pass
        if self.Load_Field_Config() is not True:
            pass

    def Load_User_Config(self):
        pass

    def Load_Field_Config(self):
        pass


    # 选择文件(多个)
    def Choose_Files(self):
        fetch = lambda x: x.split('/')[-1]  # 文件名绝对路径提取文件名

        self.fileNamePaths, _ = QFileDialog.getOpenFileNames(self, "选择文件", "./", "All Files (*);;Text Files (*.txt)")

        result = map(fetch, self.fileNamePaths[0])  # TODO 后期换成多个文件处理

        self.fileNames.clear()
        for file in result:
            self.fileNames.append(file)

        self.Clear_Files_Msg()
        self.Show_Files()

    # 打印文件名和文件数量
    def Show_Files(self):
        self.fileNumber.setText(str(len(self.fileNames)))
        self.fileNameShow.setText("; ".join(self.fileNames))

    # 清空文件数据
    def Clear_Files_Msg(self):
        self.fileNumber.clear()
        self.fileNameShow.clear()