popup multi-usage

Author: Okaneeee

assets/checkmark.png

@@ -16,19 +16,27 @@
 
 # classes
 class popup(QWidget):
-    def __init__(self) -> None:
+    def __init__(self, popupText: str) -> None:
         # default
         super().__init__()
-        self.setWindowTitle("Files Renamer") ; self.setWindowIcon(QIcon("assets/warning.png"))
         self.resize(160, 80) ; self.setFixedSize(self.size())
 
         # main layout
         self.mainLayout: QVBoxLayout = QVBoxLayout() ; self.setLayout(self.mainLayout)
 
         # widgets
         # --- warning message ---
-        self.warning: QLabel = QLabel("Something is wrong !") ; self.warning.setAlignment(Qt.AlignmentFlag.AlignCenter)
-        self.mainLayout.addWidget(self.warning)
+        if (popupText == "warning"):
+            self.setWindowTitle("Warning")
+            self.setWindowIcon(QIcon("assets/warning.png"))
+            self.pText: QLabel = QLabel("Something is wrong!")
+        else:
+            self.setWindowTitle("Notice")
+            self.setWindowIcon(QIcon("assets/checkmark.png"))
+            self.pText: QLabel = QLabel("Done!")
+
+        self.pText.setAlignment(Qt.AlignmentFlag.AlignCenter)
+        self.mainLayout.addWidget(self.pText)
 
         # --- close button ---
         self.closeButton : QPushButton = QPushButton("Close")
@@ -48,7 +56,6 @@ def __init__(self) -> None:
         self.setWindowTitle("Files Renamer") ; self.setWindowIcon(QIcon("assets/icon.png"))
         self.resize(500, 220) ; self.setFixedSize(self.size())
         self.fr: FileRenamer = FileRenamer()
-        self.popup: popup = popup()
 
         # main layout
         self.topLayout: QVBoxLayout = QVBoxLayout()
@@ -121,8 +128,9 @@ def rename(self) -> None:
 
         if(selectedFolderCheck and startFileCheck and startNumberCheck and renameFormatCheck and fileExtCheck):
             self.fr.rename(folder, fileExt, startFile, int(startNb), renameFrmt)
+            self.popUp: popup = popup("done") ; self.popUp.show()
         else:
-            self.popup.show()
+            self.popUp: popup = popup("warning") ; self.popUp.show()
 
 # test
 if __name__ == "__main__":