GuiShell/ui/AddFile.py

import os
import sys
from decimal import *
from PySide2.QtWidgets import *
from PySide2.QtCore import *
from PySide2.QtGui import *
from ui.ui_AddFile import Ui_AddFileWindow
from ui.addressBar.addressbar import BreadcrumbsAddressBar
from ui.file_browser.file_list import get_files_dirs
from ui.file_browser.PathObject import PathObject
from ui.file_browser.filewidget import FileWidget


class AddFileWindow(Ui_AddFileWindow, QDialog):
    def __init__(self, parent, path) -> None:
        super().__init__()
        self.file_suffix = ["Supported Files（*.v *.sv *.f *.lst *.list）", "All files(*.*)"]
        self.setupUi(self)
        self.init_file_browser(path)
        
        self.move(parent.x()+(parent.width()-self.width())/2,parent.y()+(parent.height()-self.height())/2)
        self.show()
        self.fileTable.cellDoubleClicked.connect(self.open_item)
        self.path_obj = PathObject(path)
        self.dir_stack = []
        self.show_file_list()
        self.comboBox.addItems(self.file_suffix)
        
    def init_file_browser(self, path):
        # 表格随窗口适应大小
        self.fileTable.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
        self.address_bar = BreadcrumbsAddressBar(self)
        self.address_bar.path_selected.connect(self.path_select)
        self.verticalLayout_3.insertWidget(0, self.address_bar)

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


    def update_path(self):
        path_text = self.pathBar.text()
        if os.path.isdir(path_text):
            self.dir_stack.clear()
            self.path_obj.set(path_text)
            self.show_file_list()
        else:
            self.path_error_popup()
            self.pathBar.setText(str(self.path_obj))

    def path_error_popup(self):
        msg = QMessageBox()
        msg.setWindowTitle("Error")
        msg.setText("Invalid path!")
        msg.setIcon(QMessageBox.Critical)
        x = msg.exec_()

    def show_file_list(self):
        self.files, self.dirs = get_files_dirs(self.path_obj)
        self.update_counts()
        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.name, dir.type, height)
            self.fileTable.setCellWidget(row, 0, fwidget)

        for f in self.files:
            row = self.fileTable.rowCount()
            self.fileTable.insertRow(row)
            fwidget = FileWidget(f.name, f.type, height)
            self.fileTable.setCellWidget(row, 0, fwidget)
        # self.fileTable.resizeRowsToContents()
        # self.fileTable
        # self.show_count()

    def format_size(self, size):
        if size < 999:
            return f"{size} B"
        if size < 999999:
            num = Decimal(size / 1000)
            return f"{round(num, 2)} KB"
        if size < 999999999:
            num = Decimal(size / 1000000)
            return f"{round(num, 2)} MB"
        else:
            num = Decimal(size / 1000000000)
            return f"{round(num, 2)} GB"

    def update_counts(self):
        self.file_count = len(self.files)
        self.dir_count = len(self.dirs)

    def open_item(self, row, column):
        if column != 0:
            return
        widg = self.fileTable.cellWidget(row, column)
        name = widg.get_text()
        self.dir_stack.clear()
        abs_path = os.path.join(self.path_obj.pathtext, name)
        if os.path.isfile(os.path.join(self.path_obj.pathtext, name)):
            self.open_file(abs_path)
        elif os.path.isdir(abs_path):
            self.address_bar.set_path(abs_path)
    
    
    def open_folder(self, dir):
        if self.path_obj.is_root():
            self.dir_stack.clear()
        self.path_obj.set(dir)
        self.show_file_list()
        
    def open_file(self, file):
        path = PathObject(str(self.path_obj))
        path.set(file)
        print(file)

    def back(self):
        if self.path_obj.is_root():
            return
        dir = self.path_obj.back()
        self.dir_stack.append(dir)
        self.show_file_list()

    def forward(self):
        if len(self.dir_stack) == 0:
            return
        dir = self.dir_stack[-1]
        self.dir_stack.pop()
        self.path_obj.add(dir)
        self.show_file_list()

    def path_error_popup(self):
        msg = QMessageBox()
        msg.setWindowTitle("Error")
        msg.setText("Invalid path!")
        msg.setIcon(QMessageBox.Critical)
        msg.exec_()

    def show_count(self):
        count_text = ""
        count = self.file_count + self.dir_count

        if count == 1:
            count_text += "1 item ("
        else:
            count_text += f"{count} items ("

        if self.dir_count == 1:
            count_text += "1 folder, "
        else:
            count_text += f"{self.dir_count} dirs, "

        if self.file_count == 1:
            count_text += "1 file)"
        else:
            count_text += f"{self.file_count} files)"

        self.status_bar.showMessage(count_text)