GuiShell/ui/AddFile.py

import os
import sys
from decimal import *
import pathlib
from PySide2.QtWidgets import *
from PySide2.QtCore import *
from PySide2.QtGui import *
from ui.ui_AddPath import Ui_AddPathWindow
from ui.addressBar.addressbar import BreadcrumbsAddressBar
from ui.file_browser import file_list
from ui.file_browser.PathObject import PathObject
from ui.file_browser.filewidget import FileWidget
from ui.file_browser.FileObject import FileObject

class Category:
    def __init__(self, text, simple='') -> None:
        self.text = text
        self.simple = simple
        if not self.simple:
            self.simple = text

        
class FolderCategory:
    Search_Path = Category("Search Path", "SearchPath")


class FileCategory:
    Filelist = Category('Filelist', 'Filelist')
    Netlist = Category('Netlist', 'Netlist')
    Verilog_2001 = Category('Verilog 2001', 'V2001')
    Verilog_2005 = Category("Verilog 2005", 'V2005')
    SystemVerilog_2005 = Category("SystemVerilog 2005", "SV2005")
    SystemVerilog_2009 = Category("SystemVerilog 2009", "SV2009")
    SystemVerilog_2012 = Category("SystemVerilog 2012", "SV2012")
    all_category = [Filelist, Netlist, Verilog_2001, Verilog_2005, SystemVerilog_2005, SystemVerilog_2009, SystemVerilog_2012]
    
    def get_category(text):
        for category in FileCategory.all_category:
            if category.text == text:
                return category
    
    def get_all_text()->list:
        return [category.text for category in FileCategory.all_category]
            
    
class AddPathWindow(Ui_AddPathWindow, QDialog):
    add_file_sig = Signal(str,str)
    def __init__(self, parent, path) -> None:
        super().__init__(parent)
        self.setupUi(self)
        self.file_suffix = ["All Directories(*.*)"]
        self.init(parent, path)
    
    def init(self, parent, path):
        self.setWindowTitle("Add Path")
        self.Combobox_FileType.addItems(self.file_suffix)
        self.Combobox_FileType.currentIndexChanged.connect(lambda index:self.show_file_list())
        self.path_obj = PathObject(path)
        self.bnt_Apply.clicked.connect(self.apply)
        self.bnt_OK.clicked.connect(self.click_ok)
        self.init_file_browser(path)
        self.fileTable.clearSelection()
        self.fileTable.setCurrentIndex(QModelIndex())
        self.move(parent.x()+(parent.width()-self.width())/2 + self.width()*0.8,parent.y()+(parent.height()-self.height())/2)
        self.show()
    
    def init_file_browser(self, path):
        # 表格随窗口适应大小
        self.fileTable.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
        self.fileTable.cellDoubleClicked.connect(self.double_click)
        self.fileTable.mousePressEvent = self.click_table_widget
        self.address_bar = BreadcrumbsAddressBar(self)
        self.address_bar.path_selected.connect(self.open_folder)
        self.address_bar.path_error.connect(lambda path: self.address_error_connect("No such file or directory", path.name))
        self.address_bar.listdir_error.connect(lambda path: self.address_error_connect("Permission denied", path.name))
        self.address_bar.set_path(path)
        self.verticalLayout_3.insertWidget(0, self.address_bar)

    def open_folder(self, dir):
        self.path_obj.set(dir)
        self.show_file_list()

    def show_file_list(self):
        _, self.dirs = file_list.get_files_dirs(self.path_obj)
        self.fileTable.setRowCount(0)

        height = self.fileTable.verticalHeader().defaultSectionSize()
        for dir in self.dirs:
            row = self.fileTable.rowCount()
            self.fileTable.insertRow(row)
            fwidget = FileWidget(dir, "Folder", height)
            self.fileTable.setCellWidget(row, 0, fwidget)

    def address_error_connect(self, error, path):
        self.path_error_popup(error, path)
        self.address_bar._show_address_field(True)
        self.address_bar.line_address.deselect()
        self.address_bar.line_address.insert('/')

    def path_error_popup(self, error='', path=''):
        msg = QMessageBox(self)
        msg.setWindowTitle("Error")
        msg.setText(error)
        msg.setIcon(QMessageBox.Critical)
        msg.move(self.x()+(self.width()-msg.width())/2,self.y()+(self.height()-msg.height())/2)
        msg.exec_()

    # 点击空白处取消选择
    def click_table_widget(self, event:QMouseEvent):
        p = event.pos()
        index = self.fileTable.indexAt(p)
        if index.row() == -1:
            self.fileTable.clearSelection()
            self.fileTable.setCurrentIndex(QModelIndex())
        else:
            QTableWidget.mousePressEvent(self.fileTable, event)

    def double_click(self, row, column):
        if column != 0:
            return
        widg:FileWidget = self.fileTable.cellWidget(row, column)
        name = widg.get_text()
        abs_path = os.path.join(self.path_obj.pathtext, name)
        if os.path.isfile(abs_path):
            self.open_file(abs_path)
        # 这里不能用 elif os.path.isdir(abs_path) ，因为假如目录没有权限访问，即便它是个目录也会返回 False
        else:
            self.address_bar.set_path(abs_path)

    def current_select_path(self):
        row = self.fileTable.currentRow()
        if row < 0:
            return str(self.address_bar.path_)
        widg:FileWidget = self.fileTable.cellWidget(row, 0)
        filename = widg.get_text()
        abs_path = os.path.join(self.path_obj.pathtext, filename)
        return abs_path
    
        
    def apply(self):
        abs_path = self.current_select_path()
        if not abs_path:
            return
        if not os.path.isdir(abs_path):
            return 
        self.add_file_sig.emit(FolderCategory.Search_Path.text, abs_path)
        self.fileTable.clearSelection()
        self.fileTable.setCurrentIndex(QModelIndex())
        
    def click_ok(self):
        self.apply()
        self.close()


class AddFileWindow(AddPathWindow):
    def __init__(self, parent, path) -> None:
        super().__init__(parent, path)
    
    def init(self,parent, path):
        self.setWindowTitle("Add File")
        self.file_type = [".v", ".sv", ".f", ".lst", ".list"]
        self.file_suffix = ["Supported Files(%s)" % " ".join(['*'+str(s) for s in self.file_type]), "All Files(*.*)"]
        super().init(parent, path)
        self.verticalLayout_2 = QVBoxLayout()
        self.verticalLayout_2.setObjectName(u"verticalLayout_2")
        self.verticalLayout_2.setContentsMargins(0, 0, 0, 0)
        self.verticalSpacer_2 = QSpacerItem(20, 31, QSizePolicy.Minimum, QSizePolicy.Fixed)
        self.verticalLayout_2.addItem(self.verticalSpacer_2)
        file_type = FileCategory.get_all_text()
        for t in file_type:
            radio = QRadioButton(self)
            radio.setObjectName(t)
            radio.setText(t)
            self.verticalLayout_2.addWidget(radio)
            if t == FileCategory.Verilog_2005.text:
                radio.setChecked(True)
        self.verticalSpacer = QSpacerItem(20, 40, QSizePolicy.Minimum, QSizePolicy.Expanding)
        self.verticalLayout_2.addItem(self.verticalSpacer)
        self.horizontalLayout.insertLayout(0, self.verticalLayout_2)
        

    def path_select(self, path):
        self.path_obj.set(path)
        self.open_folder(path)

        
    def files_filter(self, files):
        if self.Combobox_FileType.currentText() == "All Files(*.*)":
            return files
        filt_res = []
        for i in range(len(files)):
            _,type =os.path.splitext(files[i])
            if type in self.file_type:
                filt_res.append(files[i])
        return filt_res
    
    def show_file_list(self):
        all_files, self.dirs = file_list.get_files_dirs(self.path_obj)
        self.files = self.files_filter(all_files)
        self.fileTable.setRowCount(0)

        height = self.fileTable.verticalHeader().defaultSectionSize()
        for dir in self.dirs:
            row = self.fileTable.rowCount()
            self.fileTable.insertRow(row)
            fwidget = FileWidget(dir, "Folder", height)
            self.fileTable.setCellWidget(row, 0, fwidget)

        for f in self.files:
            row = self.fileTable.rowCount()
            self.fileTable.insertRow(row)
            fwidget = FileWidget(f, "File", height)
            self.fileTable.setCellWidget(row, 0, fwidget)

    def current_radio_text(self):
        file_type = None
        for children in self.findChildren(QRadioButton):
            children:QRadioButton
            if children.isChecked():
                file_type = children.text()
        return file_type

    def apply(self):
        abs_path = self.current_select_path()
        if not abs_path:
            return
        if not os.path.isfile(abs_path):
            return 
        text = self.current_radio_text()
        self.add_file_sig.emit(text, abs_path)
        self.fileTable.clearSelection()
        self.fileTable.setCurrentIndex(QModelIndex())

    def click_ok(self):
        self.apply()
        self.close()
    
    def open_file(self, file):
        self.apply()
        self.close()