PyBreeze 1.0.7__tar.gz → 1.0.8__tar.gz

{pybreeze-1.0.7 → pybreeze-1.0.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pybreeze
-Version: 1.0.7
+Version: 1.0.8
 Summary: IDE for multi automation
 Author-email: JE-Chen <jechenmailman@gmail.com>
 Project-URL: Homepage, https://github.com/Intergration-Automation-Testing/AutomationEditor
@@ -21,9 +21,10 @@ Requires-Dist: je_load_density
 Requires-Dist: je_api_testka
 Requires-Dist: je-mail-thunder
 Requires-Dist: automation-file
-Requires-Dist: PySide6==6.10.2
+Requires-Dist: PySide6==6.11.0
 Requires-Dist: test_pioneer
 Requires-Dist: paramiko
+Requires-Dist: jupyterlab
 Dynamic: license-file
 
 # PyBreeze: The Automation-First IDE

{pybreeze-1.0.7 → pybreeze-1.0.8}/pybreeze/__init__.py

@@ -1,7 +1,7 @@
-from pybreeze.pybreeze_ui.editor_main.main_ui import AutomationEditor
+from pybreeze.pybreeze_ui.editor_main.main_ui import PyBreezeMainWindow
 from pybreeze.pybreeze_ui.editor_main.main_ui import EDITOR_EXTEND_TAB
 from pybreeze.pybreeze_ui.editor_main.main_ui import start_editor
 
 __all__ = [
-    "start_editor", "AutomationEditor", "EDITOR_EXTEND_TAB"
+    "start_editor", "PyBreezeMainWindow", "EDITOR_EXTEND_TAB"
 ]

{pybreeze-1.0.7 → pybreeze-1.0.8}/pybreeze/extend/process_executor/api_testka/api_testka_process.py

@@ -5,14 +5,14 @@ from typing import TYPE_CHECKING, Union
 from pybreeze.extend.process_executor.process_executor_utils import build_process
 
 if TYPE_CHECKING:
-    from pybreeze.pybreeze_ui.editor_main.main_ui import AutomationEditor
+    from pybreeze.pybreeze_ui.editor_main.main_ui import PyBreezeMainWindow
 import sys
 
 from pybreeze.utils.file_process.get_dir_file_list import ask_and_get_dir_files_as_list
 
 
 def call_api_testka(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         exec_str: Union[str, None] = None,
         program_buffer: int = 1024000
 ):
@@ -20,7 +20,7 @@ def call_api_testka(
 
 
 def call_api_testka_with_send(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         exec_str: Union[str, None] = None,
         program_buffer: int = 1024000
 ):
@@ -28,7 +28,7 @@ def call_api_testka_with_send(
 
 
 def call_api_testka_multi_file(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         program_buffer: int = 1024000
 ):
     try:
@@ -47,7 +47,7 @@ def call_api_testka_multi_file(
 
 
 def call_api_testka_multi_file_and_send(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         program_buffer: int = 1024000
 ):
     try:

{pybreeze-1.0.7 → pybreeze-1.0.8}/pybreeze/extend/process_executor/auto_control/auto_control_process.py

@@ -5,14 +5,14 @@ from typing import TYPE_CHECKING, Union
 from pybreeze.extend.process_executor.process_executor_utils import build_process
 
 if TYPE_CHECKING:
-    from pybreeze.pybreeze_ui.editor_main.main_ui import AutomationEditor
+    from pybreeze.pybreeze_ui.editor_main.main_ui import PyBreezeMainWindow
 import sys
 
 from pybreeze.utils.file_process.get_dir_file_list import ask_and_get_dir_files_as_list
 
 
 def call_auto_control(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         exec_str: Union[str, None] = None,
         program_buffer: int = 1024000
 ):
@@ -20,7 +20,7 @@ def call_auto_control(
 
 
 def call_auto_control_with_send(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         exec_str: Union[str, None] = None,
         program_buffer: int = 1024000
 ):
@@ -28,7 +28,7 @@ def call_auto_control_with_send(
 
 
 def call_auto_control_multi_file(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         program_buffer: int = 1024000
 ):
     need_to_execute_list: list = ask_and_get_dir_files_as_list(main_window)
@@ -44,7 +44,7 @@ def call_auto_control_multi_file(
 
 
 def call_auto_control_multi_file_and_send(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         program_buffer: int = 1024000
 ):
     try:

{pybreeze-1.0.7 → pybreeze-1.0.8}/pybreeze/extend/process_executor/file_automation/file_automation_process.py

@@ -5,14 +5,14 @@ from typing import TYPE_CHECKING, Union
 from pybreeze.extend.process_executor.process_executor_utils import build_process
 
 if TYPE_CHECKING:
-    from pybreeze.pybreeze_ui.editor_main.main_ui import AutomationEditor
+    from pybreeze.pybreeze_ui.editor_main.main_ui import PyBreezeMainWindow
 import sys
 
 from pybreeze.utils.file_process.get_dir_file_list import ask_and_get_dir_files_as_list
 
 
 def call_file_automation_test(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         exec_str: Union[str, None] = None,
         program_buffer: int = 1024000
 ):
@@ -20,7 +20,7 @@ def call_file_automation_test(
 
 
 def call_file_automation_test_with_send(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         exec_str: Union[str, None] = None,
         program_buffer: int = 1024000
 ):
@@ -28,7 +28,7 @@ def call_file_automation_test_with_send(
 
 
 def call_file_automation_test_multi_file(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         program_buffer: int = 1024000
 ):
     try:
@@ -47,7 +47,7 @@ def call_file_automation_test_multi_file(
 
 
 def call_file_automation_test_multi_file_and_send(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         program_buffer: int = 1024000
 ):
     try:

{pybreeze-1.0.7 → pybreeze-1.0.8}/pybreeze/extend/process_executor/load_density/load_density_process.py

@@ -5,14 +5,14 @@ from typing import TYPE_CHECKING, Union
 from pybreeze.extend.process_executor.process_executor_utils import build_process
 
 if TYPE_CHECKING:
-    from pybreeze.pybreeze_ui.editor_main.main_ui import AutomationEditor
+    from pybreeze.pybreeze_ui.editor_main.main_ui import PyBreezeMainWindow
 import sys
 
 from pybreeze.utils.file_process.get_dir_file_list import ask_and_get_dir_files_as_list
 
 
 def call_load_density(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         exec_str: Union[str, None] = None,
         program_buffer: int = 1024000
 ):
@@ -20,7 +20,7 @@ def call_load_density(
 
 
 def call_load_density_with_send(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         exec_str: Union[str, None] = None,
         program_buffer: int = 1024000
 ):
@@ -28,7 +28,7 @@ def call_load_density_with_send(
 
 
 def call_load_density_multi_file(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         program_buffer: int = 1024000
 ):
     try:
@@ -47,7 +47,7 @@ def call_load_density_multi_file(
 
 
 def call_load_density_multi_file_and_send(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         program_buffer: int = 1024000
 ):
     try:

{pybreeze-1.0.7 → pybreeze-1.0.8}/pybreeze/extend/process_executor/mail_thunder/mail_thunder_process.py

@@ -5,11 +5,11 @@ from typing import TYPE_CHECKING, Union
 from pybreeze.extend.process_executor.process_executor_utils import build_process
 
 if TYPE_CHECKING:
-    from pybreeze.pybreeze_ui.editor_main.main_ui import AutomationEditor
+    from pybreeze.pybreeze_ui.editor_main.main_ui import PyBreezeMainWindow
 
 
 def call_mail_thunder(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         exec_str: Union[str, None] = None,
         program_buffer: int = 1024000
 ):

{pybreeze-1.0.7 → pybreeze-1.0.8}/pybreeze/extend/process_executor/process_executor_utils.py

@@ -13,11 +13,11 @@ from pybreeze.utils.exception.exception_tags import wrong_test_data_format_excep
 from pybreeze.utils.exception.exceptions import ITETestExecutorException
 
 if TYPE_CHECKING:
-    from pybreeze.pybreeze_ui.editor_main.main_ui import AutomationEditor
+    from pybreeze.pybreeze_ui.editor_main.main_ui import PyBreezeMainWindow
 
 
 def build_process(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         package: str,
         exec_str: Union[str, None] = None,
         send_mail: bool = False,
@@ -42,7 +42,7 @@ def build_process(
 
 
 def start_process(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         package: str,
         test_format_code: str,
         send_mail: bool = False,

{pybreeze-1.0.7 → pybreeze-1.0.8}/pybreeze/extend/process_executor/test_pioneer/test_pioneer_process_manager.py

@@ -17,19 +17,19 @@ from je_editor.utils.venv_check.check_venv import check_and_choose_venv
 from pybreeze.pybreeze_ui.show_code_window.code_window import CodeWindow
 
 if TYPE_CHECKING:
-    from pybreeze.pybreeze_ui.editor_main.main_ui import AutomationEditor
+    from pybreeze.pybreeze_ui.editor_main.main_ui import PyBreezeMainWindow
 
 
 class TestPioneerProcess(object):
 
     def __init__(
             self,
-            main_window: AutomationEditor,
+            main_window: PyBreezeMainWindow,
             executable_path: str,
             program_buffer: int = 1024000,
             encoding: str = "utf-8",
     ):
-        self._main_window: AutomationEditor = main_window
+        self._main_window: PyBreezeMainWindow = main_window
         self._widget: QWidget = main_window.tab_widget.currentWidget()
         # Code window init
         self._code_window = CodeWindow()
@@ -175,7 +175,7 @@ class TestPioneerProcess(object):
         self._timer.start()
 
 
-def init_and_start_test_pioneer_process(ui_we_want_to_set: AutomationEditor, file_path: str):
+def init_and_start_test_pioneer_process(ui_we_want_to_set: PyBreezeMainWindow, file_path: str):
     test_pioneer_process_manager = TestPioneerProcess(
         main_window=ui_we_want_to_set, executable_path=file_path)
     test_pioneer_process_manager.start_test_pioneer_process()

{pybreeze-1.0.7 → pybreeze-1.0.8}/pybreeze/extend/process_executor/web_runner/web_runner_process.py

@@ -5,14 +5,14 @@ from typing import TYPE_CHECKING, Union
 from pybreeze.extend.process_executor.process_executor_utils import build_process
 
 if TYPE_CHECKING:
-    from pybreeze.pybreeze_ui.editor_main.main_ui import AutomationEditor
+    from pybreeze.pybreeze_ui.editor_main.main_ui import PyBreezeMainWindow
 import sys
 
 from pybreeze.utils.file_process.get_dir_file_list import ask_and_get_dir_files_as_list
 
 
 def call_web_runner_test(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         exec_str: Union[str, None] = None,
         program_buffer: int = 1024000
 ):
@@ -20,7 +20,7 @@ def call_web_runner_test(
 
 
 def call_web_runner_test_with_send(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         exec_str: Union[str, None] = None,
         program_buffer: int = 1024000
 ):
@@ -28,7 +28,7 @@ def call_web_runner_test_with_send(
 
 
 def call_web_runner_test_multi_file(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         program_buffer: int = 1024000
 ):
     try:
@@ -47,7 +47,7 @@ def call_web_runner_test_multi_file(
 
 
 def call_web_runner_test_multi_file_and_send(
-        main_window: AutomationEditor,
+        main_window: PyBreezeMainWindow,
         program_buffer: int = 1024000
 ):
     try:

{pybreeze-1.0.7 → pybreeze-1.0.8}/pybreeze/pybreeze_ui/editor_main/main_ui.py

@@ -4,7 +4,7 @@ from os import environ
 from pathlib import Path
 from typing import List, Dict, Type
 
-from pybreeze.utils.logging.logger import automation_ide_logger
+from pybreeze.utils.logging.logger import pybreeze_logger
 
 environ["LOCUST_SKIP_MONKEY_PATCH"] = "1"
 
@@ -24,7 +24,7 @@ EDITOR_EXTEND_TAB: Dict[str, Type[QWidget]] = {
 }
 
 
-class AutomationEditor(EditorMain):
+class PyBreezeMainWindow(EditorMain):
 
     def __init__(self, debug_mode: bool = False, show_system_tray_ray: bool = False, extend: bool = False) -> None:
         super().__init__(debug_mode, show_system_tray_ray, extend=True)
@@ -94,7 +94,7 @@ def start_editor(debug_mode: bool = False, **kwargs) -> None:
     new_ide = QCoreApplication.instance()
     if new_ide is None:
         new_ide = QApplication(sys.argv)
-    window = AutomationEditor(debug_mode=debug_mode, **kwargs)
+    window = PyBreezeMainWindow(debug_mode=debug_mode, **kwargs)
     apply_stylesheet(new_ide, theme="dark_amber.xml")
     window.showMaximized()
     try:

{pybreeze-1.0.7 → pybreeze-1.0.8}/pybreeze/pybreeze_ui/extend_multi_language/extend_english.py

@@ -13,6 +13,8 @@ def update_english_word_dict():
             "run_label": "Run",
             "help_label": "HELP",
             "project_label": "Project",
+            # Tab tools menu
+            "tab_menu_jupyterlab_tab_name": "JupyterLab",
             # APITestka Menu
             "apitestka_menu_label": "APITestka",
             "apitestka_run_script_label": "Run APITestka Script",

{pybreeze-1.0.7 → pybreeze-1.0.8}/pybreeze/pybreeze_ui/extend_multi_language/extend_traditional_chinese.py

@@ -13,6 +13,8 @@ def update_traditional_chinese_word_dict():
             "run_label": "運行",
             "help_label": "幫助",
             "project_label": "專案",
+            # Tab tools menu
+            "tab_menu_jupyterlab_tab_name": "JupyterLab",
             # APITestka Menu
             "apitestka_menu_label": "APITestka",
             "apitestka_run_script_label": "運行 APITestka 腳本",

pybreeze-1.0.8/pybreeze/pybreeze_ui/jupyter_lab_gui/jupyer_lab_thread.py

@@ -0,0 +1,54 @@
+import socket
+import subprocess
+import sys
+from PySide6.QtCore import QThread, Signal
+import time
+
+
+def find_free_port():
+    s = socket.socket()
+    s.bind(("", 0))
+    port = s.getsockname()[1]
+    s.close()
+    return port
+
+
+class JupyterServerThread(QThread):
+    server_ready = Signal(str)
+
+    def __init__(self):
+        super().__init__()
+        self.process = None
+
+    def run(self):
+        port = find_free_port()
+
+        cmd = [
+            sys.executable,
+            "-m",
+            "jupyterlab",
+            "--no-browser",
+            f"--ServerApp.port={port}",
+            "--ServerApp.token=",
+            "--ServerApp.password=",
+            "--ServerApp.allow_origin=*",
+            "--ServerApp.disable_check_xsrf=True",
+        ]
+
+        self.process = subprocess.Popen(cmd)
+
+        # 輪詢 port，直到可連
+        while True:
+            try:
+                s = socket.create_connection(("localhost", port), timeout=0.5)
+                s.close()
+                break
+            except OSError:
+                time.sleep(0.1)
+
+        # Server ready，發射 signal
+        self.server_ready.emit(f"http://localhost:{port}/lab")
+
+    def stop(self):
+        if self.process:
+            self.process.terminate()

pybreeze-1.0.8/pybreeze/pybreeze_ui/jupyter_lab_gui/jupyter_lab_widget.py

@@ -0,0 +1,58 @@
+import sys
+
+from PySide6.QtCore import QUrl
+from PySide6.QtWebEngineWidgets import QWebEngineView
+from PySide6.QtWidgets import QApplication, QVBoxLayout, QWidget
+
+from pybreeze.pybreeze_ui.jupyter_lab_gui.jupyer_lab_thread import JupyterServerThread
+
+
+class JupyterLabWidget(QWidget):
+
+    def __init__(self):
+        super().__init__()
+
+        layout = QVBoxLayout(self)
+
+        self.browser = QWebEngineView()
+        layout.addWidget(self.browser)
+
+        # 啟動 Jupyter server thread
+        self.thread = JupyterServerThread()
+        self.thread.server_ready.connect(self.load_lab)
+        self.thread.start()
+
+    def load_lab(self, url: str):
+        """Load JupyterLab URL into WebEngine"""
+
+        if not url:
+            print("Invalid JupyterLab URL")
+            return
+
+        print("JupyterLab running at:", url)
+
+        qurl = QUrl(url)
+
+        if not qurl.isValid():
+            print("Invalid QUrl:", url)
+            return
+
+        self.browser.setUrl(qurl)
+
+    def closeEvent(self, event):
+
+        if hasattr(self, "thread") and self.thread.isRunning():
+            self.thread.stop()
+            self.thread.quit()
+            self.thread.wait()
+
+        event.accept()
+
+
+if __name__ == "__main__":
+    app = QApplication(sys.argv)
+
+    win = JupyterLabWidget()
+    win.show()
+
+    sys.exit(app.exec())

{pybreeze-1.0.7 → pybreeze-1.0.8}/pybreeze/pybreeze_ui/menu/automation_menu/api_testka_menu/build_api_testka_menu.py

@@ -8,7 +8,7 @@ from je_editor import language_wrapper
 from pybreeze.pybreeze_ui.menu.menu_utils import open_web_browser
 
 if TYPE_CHECKING:
-    from pybreeze.pybreeze_ui.editor_main.main_ui import AutomationEditor
+    from pybreeze.pybreeze_ui.editor_main.main_ui import PyBreezeMainWindow
 import sys
 
 from PySide6.QtGui import QAction
@@ -17,7 +17,7 @@ from pybreeze.extend.process_executor.api_testka.api_testka_process import call_
     call_api_testka_with_send, call_api_testka_multi_file, call_api_testka_multi_file_and_send
 
 
-def set_apitestka_menu(ui_we_want_to_set: AutomationEditor):
+def set_apitestka_menu(ui_we_want_to_set: PyBreezeMainWindow):
     """
     Build menu include APITestka feature.
     :param ui_we_want_to_set: main window to add menu.
@@ -131,7 +131,7 @@ def create_project() -> None:
         print(repr(error), file=sys.stderr)
 
 
-def add_api_testka_gui(ui_we_want_to_set: AutomationEditor) -> None:
+def add_api_testka_gui(ui_we_want_to_set: PyBreezeMainWindow) -> None:
     ui_we_want_to_set.tab_widget.addTab(
         APITestkaWidget(), "APITestka GUI"
     )

{pybreeze-1.0.7 → pybreeze-1.0.8}/pybreeze/pybreeze_ui/menu/automation_menu/auto_control_menu/build_autocontrol_menu.py

@@ -9,7 +9,7 @@ from je_editor.pyside_ui.main_ui.save_settings.user_color_setting_file import ac
 from pybreeze.pybreeze_ui.menu.menu_utils import open_web_browser
 
 if TYPE_CHECKING:
-    from pybreeze.pybreeze_ui.editor_main.main_ui import AutomationEditor
+    from pybreeze.pybreeze_ui.editor_main.main_ui import PyBreezeMainWindow
 import sys
 
 import je_auto_control
@@ -20,7 +20,7 @@ from pybreeze.extend.process_executor.auto_control.auto_control_process import \
     call_auto_control_multi_file_and_send
 
 
-def set_autocontrol_menu(ui_we_want_to_set: AutomationEditor):
+def set_autocontrol_menu(ui_we_want_to_set: PyBreezeMainWindow):
     """
     Build menu include AutoControl feature.
     :param ui_we_want_to_set: main window to add menu.
@@ -153,7 +153,7 @@ def create_project() -> None:
         print(repr(error), file=sys.stderr)
 
 
-def stop_record(editor_instance: AutomationEditor):
+def stop_record(editor_instance: PyBreezeMainWindow):
     widget = editor_instance.tab_widget.currentWidget()
     if isinstance(widget, EditorWidget):
         text_cursor = widget.code_edit.textCursor()
@@ -163,7 +163,7 @@ def stop_record(editor_instance: AutomationEditor):
         text_cursor.insertBlock()
 
 
-def add_autocontrol_gui(ui_we_want_to_set: AutomationEditor) -> None:
+def add_autocontrol_gui(ui_we_want_to_set: PyBreezeMainWindow) -> None:
     ui_we_want_to_set.tab_widget.addTab(
         AutoControlGUIWidget(), "AutoControl GUI"
     )

{pybreeze-1.0.7 → pybreeze-1.0.8}/pybreeze/pybreeze_ui/menu/automation_menu/automation_file_menu/build_automation_file_menu.py

@@ -7,7 +7,7 @@ from je_editor import language_wrapper
 from pybreeze.pybreeze_ui.menu.menu_utils import open_web_browser
 
 if TYPE_CHECKING:
-    from pybreeze.pybreeze_ui.editor_main.main_ui import AutomationEditor
+    from pybreeze.pybreeze_ui.editor_main.main_ui import PyBreezeMainWindow
 import sys
 
 from PySide6.QtGui import QAction
@@ -17,7 +17,7 @@ from pybreeze.extend.process_executor.file_automation.file_automation_process im
     call_file_automation_test_multi_file_and_send
 
 
-def set_automation_file_menu(ui_we_want_to_set: AutomationEditor):
+def set_automation_file_menu(ui_we_want_to_set: PyBreezeMainWindow):
     """
     Build menu include WebRunner feature.
     :param ui_we_want_to_set: main window to add menu.

{pybreeze-1.0.7 → pybreeze-1.0.8}/pybreeze/pybreeze_ui/menu/automation_menu/load_density_menu/build_load_density_menu.py

@@ -8,7 +8,7 @@ from je_load_density.gui.main_widget import LoadDensityWidget
 from pybreeze.pybreeze_ui.menu.menu_utils import open_web_browser
 
 if TYPE_CHECKING:
-    from pybreeze.pybreeze_ui.editor_main.main_ui import AutomationEditor
+    from pybreeze.pybreeze_ui.editor_main.main_ui import PyBreezeMainWindow
 import sys
 
 from PySide6.QtGui import QAction
@@ -18,7 +18,7 @@ from pybreeze.extend.process_executor.load_density.load_density_process import \
     call_load_density_multi_file_and_send
 
 
-def set_load_density_menu(ui_we_want_to_set: AutomationEditor):
+def set_load_density_menu(ui_we_want_to_set: PyBreezeMainWindow):
     """
     Build menu include LoadDensity feature.
     :param ui_we_want_to_set: main window to add menu.
@@ -131,7 +131,7 @@ def create_project() -> None:
         print(repr(error), file=sys.stderr)
 
 
-def add_load_density_gui(ui_we_want_to_set: AutomationEditor) -> None:
+def add_load_density_gui(ui_we_want_to_set: PyBreezeMainWindow) -> None:
     ui_we_want_to_set.tab_widget.addTab(
         LoadDensityWidget(), "LoadDensity GUI"
     )

{pybreeze-1.0.7 → pybreeze-1.0.8}/pybreeze/pybreeze_ui/menu/automation_menu/mail_thunder_menu/build_mail_thunder_menu.py

@@ -7,7 +7,7 @@ from je_editor import language_wrapper
 from pybreeze.pybreeze_ui.menu.menu_utils import open_web_browser
 
 if TYPE_CHECKING:
-    from pybreeze.pybreeze_ui.editor_main.main_ui import AutomationEditor
+    from pybreeze.pybreeze_ui.editor_main.main_ui import PyBreezeMainWindow
 import sys
 
 from PySide6.QtGui import QAction
@@ -15,7 +15,7 @@ from PySide6.QtGui import QAction
 from pybreeze.extend.process_executor.mail_thunder.mail_thunder_process import call_mail_thunder
 
 
-def set_mail_thunder_menu(ui_we_want_to_set: AutomationEditor):
+def set_mail_thunder_menu(ui_we_want_to_set: PyBreezeMainWindow):
     """
     Build menu include LoadDensity feature.
     :param ui_we_want_to_set: main window to add menu.

{pybreeze-1.0.7 → pybreeze-1.0.8}/pybreeze/pybreeze_ui/menu/automation_menu/test_pioneer_menu/build_test_pioneer_menu.py

@@ -9,14 +9,14 @@ from pybreeze.extend.process_executor.test_pioneer.test_pioneer_process_manager
     init_and_start_test_pioneer_process
 
 if TYPE_CHECKING:
-    from pybreeze.pybreeze_ui.editor_main.main_ui import AutomationEditor
+    from pybreeze.pybreeze_ui.editor_main.main_ui import PyBreezeMainWindow
 
 from PySide6.QtGui import QAction
 from je_editor import language_wrapper
 from test_pioneer import create_template_dir
 
 
-def set_test_pioneer_menu(ui_we_want_to_set: AutomationEditor):
+def set_test_pioneer_menu(ui_we_want_to_set: PyBreezeMainWindow):
     """
     Build menu include Test Pioneer feature.
     :param ui_we_want_to_set: main window to add menu.
@@ -44,7 +44,7 @@ def set_test_pioneer_menu(ui_we_want_to_set: AutomationEditor):
     )
 
 
-def check_file(ui_we_want_to_set: AutomationEditor):
+def check_file(ui_we_want_to_set: PyBreezeMainWindow):
     dialog = QFileDialog(ui_we_want_to_set)
     dialog.setNameFilter("Yaml (*.yml)")
     file_tuple = dialog.getOpenFileName()

{pybreeze-1.0.7 → pybreeze-1.0.8}/pybreeze/pybreeze_ui/menu/automation_menu/web_runner_menu/build_webrunner_menu.py

@@ -7,7 +7,7 @@ from je_editor import language_wrapper
 from pybreeze.pybreeze_ui.menu.menu_utils import open_web_browser
 
 if TYPE_CHECKING:
-    from pybreeze.pybreeze_ui.editor_main.main_ui import AutomationEditor
+    from pybreeze.pybreeze_ui.editor_main.main_ui import PyBreezeMainWindow
 import sys
 
 from PySide6.QtGui import QAction
@@ -16,7 +16,7 @@ from pybreeze.extend.process_executor.web_runner.web_runner_process import call_
     call_web_runner_test_with_send, call_web_runner_test_multi_file, call_web_runner_test_multi_file_and_send
 
 
-def set_web_runner_menu(ui_we_want_to_set: AutomationEditor):
+def set_web_runner_menu(ui_we_want_to_set: PyBreezeMainWindow):
     """
     Build menu include WebRunner feature.
     :param ui_we_want_to_set: main window to add menu.