add install menu

add install menu

Author: JE-Chen

.idea/misc.xml

@@ -1,13 +1,14 @@
 import sys
 
 from PySide6.QtWidgets import QApplication
-from je_editor import EditorMain
+from je_editor import EditorMain, shell_manager
 from qt_material import apply_stylesheet
 
 from automation_editor.automation_editor_ui. \
     menu.api_testka_menu.build_api_testka_menu import set_apitestka_menu
 from automation_editor.automation_editor_ui. \
     menu.auto_control_menu.build_autocontrol_menu import set_autocontrol_menu
+from automation_editor.automation_editor_ui.menu.install_menu.build_install_menu import set_install_menu
 from automation_editor.automation_editor_ui.menu. \
     load_density_menu.build_load_density_menu import set_load_density_menu
 from automation_editor.automation_editor_ui \
@@ -16,7 +17,7 @@
     syntax_extend_package
 
 
-class ITE(EditorMain):
+class AutomationEditor(EditorMain):
 
     def __init__(self):
         super().__init__()
@@ -26,13 +27,20 @@ def __init__(self):
         set_apitestka_menu(self)
         set_load_density_menu(self)
         set_web_runner_menu(self)
+        set_install_menu(self)
         syntax_extend_package(self)
+        shell_manager.main_window = self
+        shell_manager.later_init()
         self.setWindowTitle("Automation Editor")
 
 
 def start_editor():
     new_editor = QApplication(sys.argv)
-    window = ITE()
+    window = AutomationEditor()
     apply_stylesheet(new_editor, theme='dark_amber.xml')
-    window.show()
+    window.showMaximized()
+    try:
+        window.startup_setting()
+    except Exception as error:
+        print(repr(error))
     sys.exit(new_editor.exec())

automation_editor/automation_editor_ui/editor_main/main_ui.py

@@ -91,3 +91,4 @@ def create_project():
     package = package_manager.installed_package_dict.get("je_api_testka", None)
     if package is not None:
         package.create_project_dir()
+

automation_editor/automation_editor_ui/menu/api_testka_menu/build_api_testka_menu.py

@@ -0,0 +1,61 @@
+from PySide6.QtGui import QAction
+from PySide6.QtWidgets import QMainWindow
+
+
+def set_install_menu(ui_we_want_to_set: QMainWindow):
+    ui_we_want_to_set.install_menu = ui_we_want_to_set.menu.addMenu("Install")
+    # Try to install Build Tools
+    ui_we_want_to_set.install_tool_action = QAction("Install Build Tools")
+    ui_we_want_to_set.install_tool_action.triggered.connect(
+        install_build_tools
+    )
+    ui_we_want_to_set.install_menu.addAction(ui_we_want_to_set.install_tool_action)
+    # Try to install AutoControl
+    ui_we_want_to_set.install_autocontrol_action = QAction("Install AutoControl")
+    ui_we_want_to_set.install_autocontrol_action.triggered.connect(
+        install_autocontrol
+    )
+    ui_we_want_to_set.install_menu.addAction(ui_we_want_to_set.install_autocontrol_action)
+    # Try to install APITestka
+    ui_we_want_to_set.install_api_testka = QAction("Install APITestka")
+    ui_we_want_to_set.install_api_testka.triggered.connect(
+        install_api_testka
+    )
+    ui_we_want_to_set.install_menu.addAction(ui_we_want_to_set.install_api_testka)
+    # Try to install LoadDensity
+    ui_we_want_to_set.install_load_density_action = QAction("Install LoadDensity")
+    ui_we_want_to_set.install_load_density_action.triggered.connect(
+        install_load_density
+    )
+    ui_we_want_to_set.install_menu.addAction(ui_we_want_to_set.install_load_density_action)
+    # Try to install WebRunner
+    ui_we_want_to_set.install_web_runner_action = QAction("Install WebRunner")
+    ui_we_want_to_set.install_web_runner_action.triggered.connect(
+        install_web_runner
+    )
+    ui_we_want_to_set.install_menu.addAction(ui_we_want_to_set.install_web_runner_action)
+
+
+def install_build_tools():
+    from je_editor import shell_manager
+    shell_manager.exec_shell("pip install --user --upgrade pip setuptools wheel")
+
+
+def install_autocontrol():
+    from je_editor import shell_manager
+    shell_manager.exec_shell("pip install --user --upgrade je_auto_control")
+
+
+def install_api_testka():
+    from je_editor import shell_manager
+    shell_manager.exec_shell("pip install --user --upgrade je_api_testka")
+
+
+def install_load_density():
+    from je_editor import shell_manager
+    shell_manager.exec_shell("pip install --user --upgrade je_load_density")
+
+
+def install_web_runner():
+    from je_editor import shell_manager
+    shell_manager.exec_shell("pip install --user --upgrade je_web_runner")

automation_editor/automation_editor_ui/menu/install_menu/__init__.py

@@ -91,4 +91,4 @@ def open_web_browser(url: str):
 def create_project():
     package = package_manager.installed_package_dict.get("je_load_density", None)
     if package is not None:
-        package.create_project_dir()
+        package.create_project_dir()

automation_editor/automation_editor_ui/menu/install_menu/build_install_menu.py

@@ -1,7 +1,4 @@
 import os
-import sys
-from importlib import import_module
-from importlib.util import find_spec
 
 
 class PackageManager(object):
@@ -21,32 +18,6 @@ def __init__(self):
             "je_api_testka",
             "je_web_runner"
         ]
-        try:
-            import je_load_density
-            import je_web_runner
-            import je_auto_control
-            import je_api_testka
-            self.installed_package_dict.update(
-                {
-                    "je_api_testka": je_api_testka,
-                    "je_web_runner": je_web_runner,
-                    "je_auto_control": je_auto_control,
-                    "je_load_density": je_load_density,
-                }
-            )
-        except ImportError:
-            pass
-
-    # This will make subprocess deadlock so emm we don't use this
-    def check_package(self):
-        for package, value in self.installed_package_dict.items():
-            found_spec = find_spec(package)
-            if found_spec is not None and value is None:
-                try:
-                    installed_package = import_module(found_spec.name)
-                    self.installed_package_dict.update({found_spec.name: installed_package})
-                except ModuleNotFoundError as error:
-                    print(repr(error), file=sys.stderr)
 
 
 package_manager = PackageManager()

automation_editor/automation_editor_ui/menu/load_density_menu/build_load_density_menu.py

@@ -91,10 +91,11 @@ def pull_text(self):
         except queue.Empty:
             pass
         if self.process.returncode == 0:
+            self.timer.stop()
             self.exit_program()
         elif self.process.returncode is not None:
-            self.exit_program()
             self.timer.stop()
+            self.exit_program()
         if self.still_run_program:
             # poll return code
             self.process.poll()
@@ -122,6 +123,7 @@ def print_and_clear_queue(self):
                 if std_err:
                     self.code_result.append(std_err)
             self.code_result.setTextColor(output_color)
+            print(f"Task exit with code {self.process.returncode}")
         except queue.Empty:
             pass
         self.run_output_queue = queue.Queue()

automation_editor/utils/manager/package_manager/package_manager_class.py

@@ -0,0 +1,2 @@
+import sys
+print(sys.prefix)