Refactor script to capture widget gallery img

Author: 5yutan5

.github/workflows/test.yml

@@ -47,12 +47,11 @@ jobs:
       - name: Capture WidgetGallery img
         run: >-
           poetry run python ./tools/capture.py
-          -i ${{runner.os}}-py${{matrix.py}}-${{matrix.qt}}
-          -d dist
+          --name ${{runner.os}}-py${{matrix.py}}-${{matrix.qt}}
 
       - uses: actions/upload-artifact@v3
         with:
-          path: ./dist/*.png
+          path: ./*.png
 
   pytest-without-qt:
     strategy:

tools/capture.py

@@ -5,7 +5,6 @@
 import os
 import platform
 import sys
-from pathlib import Path
 
 import qdarktheme
 from qdarktheme.qtpy.QtCore import Qt, QTimer, Slot
@@ -16,27 +15,14 @@
 
 def _parse_args() -> argparse.Namespace:
     parser = argparse.ArgumentParser(description="This program capture images of WidgetGallery.")
-    parser.add_argument(
-        "-d",
-        "--dir",
-        help="Output dir path",
-    )
-    parser.add_argument(
-        "-i",
-        "--identifier",
-        help="Image identifier",
-    )
+    parser.add_argument("--name", help="Image name")
     return parser.parse_args()
 
 
 class _Application(QApplication):
-    def __init__(self, img_dir_path: str | None, img_identifier: str | None) -> None:
+    def __init__(self, img_name: str) -> None:
         super().__init__(sys.argv)
-        self._img_dir_path = Path("dist" if img_dir_path is None else img_dir_path)
-        self._img_dir_path.mkdir(exist_ok=True)
-        if img_identifier is not None:
-            img_identifier = img_identifier.replace("~=", "-")
-        self._img_identifier = img_identifier
+        self._img_name = img_name.replace("~=", "-")
 
         if hasattr(Qt.ApplicationAttribute, "AA_UseHighDpiPixmaps"):
             self.setAttribute(Qt.ApplicationAttribute.AA_UseHighDpiPixmaps)  # type: ignore
@@ -48,21 +34,14 @@ def __init__(self, img_dir_path: str | None, img_identifier: str | None) -> None
     def _capture_window_img(self) -> None:
         for theme in qdarktheme.get_themes():
             self.setStyleSheet(qdarktheme.load_stylesheet(theme))
-            save_file_name = (
-                f"{self._img_identifier}-{theme}.png" if self._img_identifier else f"{theme}.png"
-            )
-
             self._gallery.setGeometry(QGuiApplication.primaryScreen().geometry())
-            img_path = self._img_dir_path / save_file_name
-            self._gallery.grab().save(img_path.as_posix())
+            self._gallery.grab().save(f"{self._img_name}-{theme}.png")
         self.exit()
 
 
 if __name__ == "__main__":
     if platform.system() == "Linux":
         os.environ["QT_QPA_PLATFORM"] = "offscreen"
-    args = _parse_args()
-
-    app = _Application(args.dir, args.identifier)
+    app = _Application(_parse_args().name)
     QTimer.singleShot(10, app._capture_window_img)
     app.exec()