PyBreeze 1.0.7__tar.gz → 1.0.9__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pybreeze
-Version: 1.0.7
+Version: 1.0.9
 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.9}/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.9}/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.9}/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.9}/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.9}/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.9}/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.9}/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.9}/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.9}/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/pybreeze_ui → pybreeze-1.0.9/pybreeze}/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",
@@ -268,5 +270,9 @@ def update_english_word_dict():
             "skills_response_label": "Response:",
             "skills_missing_input": "Please enter API URL and Prompt",
             "skills_generating": "Generating...",
+            # JupyterLab GUI
+            "jupyterlab_init": "Initializing...",
+            "jupyterlab_downloading": "Downloading...",
+            "jupyterlab_loading": "Loading..."
         }
     )

{pybreeze-1.0.7/pybreeze/pybreeze_ui → pybreeze-1.0.9/pybreeze}/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 腳本",
@@ -268,5 +270,9 @@ def update_traditional_chinese_word_dict():
             "skills_response_label": "回傳結果:",
             "skills_missing_input": "請輸入 API URL 和 Prompt",
             "skills_generating": "產生中...",
+            # JupyterLab GUI
+            "jupyterlab_init": "初始化中...",
+            "jupyterlab_downloading": "下載中...",
+            "jupyterlab_loading": "載入中...",
         }
     )

pybreeze-1.0.9/pybreeze/extend_multi_language/update_language_dict.py

@@ -0,0 +1,8 @@
+from pybreeze.extend_multi_language.extend_english import update_english_word_dict
+from pybreeze.extend_multi_language.extend_traditional_chinese import \
+    update_traditional_chinese_word_dict
+
+
+def update_language_dict():
+    update_traditional_chinese_word_dict()
+    update_english_word_dict()

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

@@ -4,17 +4,15 @@ from os import environ
 from pathlib import Path
 from typing import List, Dict, Type
 
-from pybreeze.utils.logging.logger import automation_ide_logger
-
 environ["LOCUST_SKIP_MONKEY_PATCH"] = "1"
 
 from PySide6.QtCore import QTimer, QCoreApplication
 from PySide6.QtGui import QIcon
-from PySide6.QtWidgets import QApplication, QWidget, QSystemTrayIcon
+from PySide6.QtWidgets import QApplication, QWidget
 from je_editor import EditorMain, language_wrapper
 from qt_material import apply_stylesheet
 
-from pybreeze.pybreeze_ui.extend_multi_language.update_language_dict import update_language_dict
+from pybreeze.extend_multi_language import update_language_dict
 from pybreeze.pybreeze_ui.menu.build_menubar import add_menu_to_menubar
 from pybreeze.pybreeze_ui.syntax.syntax_extend import \
     syntax_extend_package
@@ -24,7 +22,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 +92,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.9/pybreeze/pybreeze_ui/jupyter_lab_gui/jupyer_lab_thread.py

@@ -0,0 +1,95 @@
+import os
+import socket
+import subprocess
+import sys
+from PySide6.QtCore import QThread, Signal
+import time
+
+from je_editor import language_wrapper
+
+
+def find_free_port():
+    s = socket.socket()
+    s.bind(("", 0))
+    port = s.getsockname()[1]
+    s.close()
+    return port
+
+def get_venv_python():
+    # 如果在 venv 中
+    if hasattr(sys, 'real_prefix') or (hasattr(sys, 'base_prefix') and sys.base_prefix != sys.prefix):
+        return sys.executable
+
+    # 嘗試從常見位置找 venv
+    possible_paths = [
+        os.path.join(os.getcwd(), "venv", "Scripts", "python.exe"),
+        os.path.join(os.getcwd(), ".venv", "Scripts", "python.exe"),
+    ]
+
+    for path in possible_paths:
+        if os.path.exists(path):
+            return path
+
+    raise RuntimeError("找不到 venv 的 python.exe")
+
+
+def is_jupyter_installed(python_exe):
+    result = subprocess.run(
+        [python_exe, "-m", "pip", "show", "jupyterlab"],
+        stdout=subprocess.PIPE,
+        stderr=subprocess.PIPE
+    )
+    return result.returncode == 0
+
+
+class JupyterLauncherThread(QThread):
+    server_ready = Signal(str)
+    status_update = Signal(str)
+
+    def __init__(self):
+        super().__init__()
+        self.process = None
+
+    def run(self):
+        python_exe = get_venv_python()
+
+        if not is_jupyter_installed(python_exe):
+            self.status_update.emit(language_wrapper.language_word_dict.get("jupyterlab_downloading"))
+            subprocess.run([
+                python_exe,
+                "-m",
+                "pip",
+                "install",
+                "jupyterlab",
+                "-U"
+            ])
+
+        self.status_update.emit(language_wrapper.language_word_dict.get("jupyterlab_loading"))
+
+        port = find_free_port()
+
+        self.process = subprocess.Popen([
+            python_exe,
+            "-m",
+            "jupyterlab",
+            "--no-browser",
+            f"--ServerApp.port={port}",
+            "--ServerApp.token=",
+            "--ServerApp.password=",
+            "--ServerApp.allow_origin=*",
+            "--ServerApp.disable_check_xsrf=True",
+        ])
+
+        while True:
+            try:
+                s = socket.create_connection(("localhost", port), timeout=0.5)
+                s.close()
+                break
+            except OSError:
+                time.sleep(0.2)
+
+        self.server_ready.emit(f"http://localhost:{port}/lab")
+
+    def stop(self):
+        if self.process:
+            self.process.terminate()

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

@@ -0,0 +1,56 @@
+import sys
+
+from PySide6.QtCore import QUrl
+from PySide6.QtWebEngineWidgets import QWebEngineView
+from PySide6.QtWidgets import QApplication, QVBoxLayout, QWidget, QLabel
+from je_editor import language_wrapper
+
+from pybreeze.pybreeze_ui.jupyter_lab_gui.jupyer_lab_thread import JupyterLauncherThread
+
+
+class JupyterLabWidget(QWidget):
+
+    def __init__(self):
+        super().__init__()
+
+        layout = QVBoxLayout(self)
+
+        self.status_label = QLabel(language_wrapper.language_word_dict.get("jupyterlab_init"))
+        layout.addWidget(self.status_label)
+
+        self.browser = QWebEngineView()
+        self.browser.hide()
+        layout.addWidget(self.browser)
+
+        self.thread = JupyterLauncherThread()
+        self.thread.status_update.connect(self.update_status)
+        self.thread.server_ready.connect(self.load_lab)
+        self.thread.start()
+
+    def update_status(self, text):
+        self.status_label.setText(text)
+
+    def load_lab(self, url):
+        if self.status_label:
+            self.status_label.setParent(None)
+            self.status_label.deleteLater()
+            self.status_label = None
+
+        self.browser.setUrl(QUrl(url))
+        self.browser.show()
+
+    def closeEvent(self, event):
+        if self.thread.isRunning():
+            self.thread.stop()
+            self.thread.quit()
+            self.thread.wait()
+        event.accept()
+
+
+if __name__ == "__main__":
+    app = QApplication(sys.argv)
+
+    win = JupyterLabWidget()
+    win.showMaximized()
+
+    sys.exit(app.exec())

{pybreeze-1.0.7 → pybreeze-1.0.9}/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.9}/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.9}/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.9}/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"
     )