PyInstaller/PyInstallerCleanupDialog.py

# -*- coding: utf-8 -*-

# Copyright (c) 2018 - 2021 Detlev Offenbach <detlev@die-offenbachs.de>
#

"""
Module implementing a dialog to select the cleanup action.
"""

from PyQt5.QtWidgets import QDialog

from .Ui_PyInstallerCleanupDialog import Ui_PyInstallerCleanupDialog


class PyInstallerCleanupDialog(QDialog, Ui_PyInstallerCleanupDialog):
    """
    Class implementing a dialog to select the cleanup action.
    """
    BuildPath = "build"
    DistPath = "dist"
    
    def __init__(self, parent=None):
        """
        Constructor
        
        @param parent reference to the parent widget
        @type QWidget
        """
        super().__init__(parent)
        self.setupUi(self)
        
        msh = self.minimumSizeHint()
        self.resize(max(self.width(), msh.width()), msh.height())
    
    def getDirectories(self):
        """
        Public method to get the project relative directories to be cleaned.
        
        @return list of directories to be removed
        @rtype list of str
        """
        dirs = []
        
        if self.buildButton.isChecked() or self.bothButton.isChecked():
            dirs.append(PyInstallerCleanupDialog.BuildPath)
        if self.distButton.isChecked() or self.bothButton.isChecked():
            dirs.append(PyInstallerCleanupDialog.DistPath)
        
        return dirs