fix: misc improvements

src/antares_web_installer/app.py

@@ -5,10 +5,11 @@
 import textwrap
 import time
 import webbrowser
-
+from contextlib import suppress
 from difflib import SequenceMatcher
 from pathlib import Path
 from shutil import copy2, copytree
+from typing import List
 
 import httpx
 
@@ -54,7 +55,7 @@ class App:
     def __post_init__(self):
         # Prepare the path to the executable which is located in the target directory
         server_name = SERVER_NAMES[os.name]
-        self.server_path = self.target_dir.joinpath("AntaresWeb", server_name)
+        self.server_path = self.target_dir / "AntaresWeb" / server_name
 
         # Set all progress variables needed to compute current progress of the installation
         self.nb_steps = 2  # kill, install steps
@@ -95,35 +96,43 @@ def kill_running_server(self) -> None:
         Check whether Antares service is up.
         Kill the process if so.
         """
-        processes_list = list(psutil.process_iter(["pid", "name"]))
-        processes_list_length = len(processes_list)
+        server_processes = self._get_server_processes()
+        if len(server_processes) > 0:
+            logger.info("Attempt to stop running Antares server ...")
+            for p in server_processes:
+                try:
+                    p.kill()
+                except psutil.NoSuchProcess:
+                    logger.debug(f"The process '{p.pid}' was stopped before being killed.")
+                    continue
+            gone, alive = psutil.wait_procs(server_processes, timeout=5)
+            alive_count = len(alive)
+            if alive_count > 0:
+                raise InstallError(
+                    "Could not to stop Antares server. Please stop it before launching again the installation."
+                )
+            else:
+                logger.info("Antares server successfully stopped...")
+        else:
+            logger.info("No running server found, resuming installation.")
+        self.update_progress(100)
 
-        for index, proc in enumerate(processes_list):
-            # evaluate matching between query process name and existing process name
+    def _get_server_processes(self) -> List[psutil.Process]:
+        res = []
+        for process in psutil.process_iter(["pid", "name"]):
             try:
-                matching_ratio = SequenceMatcher(None, "antareswebserver", proc.name().lower()).ratio()
+                # evaluate matching between query process name and existing process name
+                matching_ratio = SequenceMatcher(None, "antareswebserver", process.name().lower()).ratio()
             except FileNotFoundError:
-                logger.warning("The process '{}' does not exist anymore. Skipping its analysis".format(proc.name()))
+                logger.warning("The process '{}' does not exist anymore. Skipping its analysis".format(process.name()))
                 continue
             except psutil.NoSuchProcess:
-                logger.warning("The process '{}' was stopped before being analyzed. Skipping.".format(proc.name()))
+                logger.warning("The process '{}' was stopped before being analyzed. Skipping.".format(process.name()))
                 continue
             if matching_ratio > 0.8:
-                logger.info("Running server found. Attempt to stop it ...")
-                logger.debug(f"Server process:{proc.name()} -  process id: {proc.pid}")
-                running_app = psutil.Process(pid=proc.pid)
-                running_app.kill()
-
-                try:
-                    running_app.wait(5)
-                except psutil.TimeoutExpired as e:
-                    raise InstallError(
-                        "Impossible to kill the server. Please kill it manually before relaunching the installer."
-                    ) from e
-                else:
-                    logger.info("The application was successfully stopped.")
-            self.update_progress((index + 1) * 100 / processes_list_length)
-        logger.info("No other processes found.")
+                res.append(process)
+                logger.debug(f"Running server found: {process.name()} -  process id: {process.pid}")
+        return res
 
     def install_files(self):
         """ """
@@ -254,7 +263,7 @@ def create_shortcuts(self):
                 description="Launch Antares Web Server in background",
             )
         except com_error as e:
-            raise InstallError("Impossible to create a new shortcut: {}\nSkip shortcut creation".format(e))
+            raise InstallError(f"Impossible to create a new shortcut: {e}\nSkipping shortcut creation") from e
         else:
             assert shortcut_path in list(desktop_path.iterdir())
             logger.info(f"Server shortcut {shortcut_path} was successfully created.")
@@ -288,17 +297,16 @@ def start_server(self):
             raise InstallError(msg)
 
         nb_attempts = 0
-        max_attempts = 10
+        max_attempts = 30
 
         while nb_attempts < max_attempts:
-            logger.info(f"Attempt #{nb_attempts}...")
-            try:
+            logger.info(f"Waiting for server start (attempt #{nb_attempts})...")
+            with suppress(httpx.RequestError):
                 res = httpx.get(SERVER_ADDRESS + "/health", timeout=1)
                 if res.status_code == 200:
                     logger.info("The server is now running.")
                     break
-            except httpx.RequestError:
-                time.sleep(1)
+            time.sleep(1)
             nb_attempts += 1
         else:
             stdout, stderr = server_process.communicate()

src/antares_web_installer/config/__init__.py

@@ -29,5 +29,8 @@ def update_config(source_path: Path, target_path: Path, version: str) -> None:
             with target_path.open("r") as f:
                 config = yaml.safe_load(f)
 
+    if version_info < (2, 18):
+        update_to_2_15(config)
+
     with source_path.open(mode="w") as f:
         yaml.dump(config, f)

src/antares_web_installer/config/config_2_18.py

@@ -0,0 +1,10 @@
+import typing as t
+
+
+def update_to_2_18(config: t.MutableMapping[str, t.Any]) -> None:
+    """
+    Update the configuration file to version 2.18 in-place:
+    we need to ensure root_path is / and api_prefix is /api
+    """
+    del config["root_path"]
+    config["api_prefix"] = "/api"

src/antares_web_installer/gui/controller.py

@@ -6,6 +6,7 @@
 import typing
 from pathlib import Path
 from threading import Thread
+from typing import Optional
 
 from antares_web_installer.gui.mvc import Controller, ControllerError
 from antares_web_installer.gui.model import WizardModel
@@ -16,17 +17,11 @@
 from antares_web_installer.gui.logger import ConsoleHandler, ProgressHandler, LogFileHandler
 
 
-class InstallationThread(Thread):
-    def __init__(self, group=None, target=None, name=None, args=(), kwargs=None, *, daemon=None):
-        super().__init__(group, target, name, args, kwargs, daemon=daemon)
-
-    def run(self):
-        try:
-            super().run()
-        except OSError as e:
-            raise e
-        except InstallError as e:
-            raise e
+def run_installation(app: App) -> None:
+    try:
+        app.run()
+    except Exception as e:
+        logger.exception(f"An error occurred during installation: {e}")
 
 
 class WizardController(Controller):
@@ -39,8 +34,8 @@ class WizardController(Controller):
     def __init__(self):
         super().__init__()
         self.app = None
-        self.log_dir = None
-        self.log_file = None
+        self.log_dir: Optional[Path] = None
+        self.log_file: Optional[Path] = None
 
         # init loggers
         self.logger = logger
@@ -64,17 +59,18 @@ def init_view(self) -> "WizardView":
         return WizardView(self)
 
     def init_file_handler(self):
-        self.log_dir = self.model.target_dir.joinpath("logs/")
+        self.log_dir: Path = self.model.target_dir / "logs"
         tmp_file_name = "wizard.log"
 
         if not self.log_dir.exists():
-            self.log_dir = self.model.source_dir.joinpath("logs/")  # use the source directory as tmp dir for logs
+            self.log_dir = self.model.source_dir / "logs"  # use the source directory as tmp dir for logs
             self.logger.debug(
                 "No log directory found with path '{}'. Attempt to generate the path.".format(self.log_dir)
             )
+            self.log_dir.mkdir(parents=True, exist_ok=True)
             self.logger.info("Path '{}' was successfully created.".format(self.log_dir))
 
-        self.log_file = self.log_dir.joinpath(tmp_file_name)
+        self.log_file = self.log_dir / tmp_file_name
 
         # check if file exists
         if self.log_file not in list(self.log_dir.iterdir()):
@@ -140,22 +136,13 @@ def install(self, callback: typing.Callable):
             logger.warning("Impossible to create a new shortcut. Skip this step.")
             logger.debug(e)
 
-        thread = InstallationThread(target=self.app.run, args=())
+        self.thread = Thread(target=lambda: run_installation(self.app), args=())
 
         try:
-            thread.run()
+            self.thread.start()
         except InstallError as e:
             self.view.raise_error(e)
 
-    def installation_over(self) -> None:
-        """
-        This method makes sure the thread terminated. If not, it waits for it to terminate.
-        """
-        if self.thread:
-            while self.thread.join():
-                if not self.thread.is_alive():
-                    break
-
     def get_target_dir(self) -> Path:
         return self.model.target_dir
 

src/antares_web_installer/gui/logger.py

@@ -2,8 +2,11 @@
 import typing
 
 
+MessageConsumer = typing.Callable[[str], None]
+
+
 class ConsoleHandler(logging.Handler):
-    def __init__(self, callback: typing.Callable):
+    def __init__(self, callback: MessageConsumer):
         logging.Handler.__init__(self)
         self.setLevel(logging.INFO)
         formatter = logging.Formatter("[%(asctime)-15s] %(message)s")
@@ -15,7 +18,7 @@ def emit(self, logs: logging.LogRecord):
 
 
 class ProgressHandler(logging.Handler):
-    def __init__(self, callback: typing.Callable):
+    def __init__(self, callback: MessageConsumer):
         """
         This logging handler intercept all logs that are progression values
         @param progress_var: tkinter.StringVar

src/antares_web_installer/gui/mvc.py

@@ -6,6 +6,8 @@
 from __future__ import annotations
 
 import tkinter as tk
+import traceback
+from tkinter import messagebox
 
 
 class Model:
@@ -35,6 +37,11 @@ class View(tk.Tk):
     def __init__(self, controller: Controller):
         super().__init__()
         self.controller = controller
+        self.report_callback_exception = self.show_error
+
+    def show_error(self, *args):
+        err = traceback.format_exception(*args)
+        messagebox.showerror("Exception", "".join(err))
 
 
 class ControllerError(Exception):

src/antares_web_installer/gui/view.py

@@ -188,5 +188,4 @@ def run_installation(self, callback):
         self.controller.install(callback)
 
     def installation_over(self):
-        self.controller.installation_over()
         self.frames["progress_frame"].installation_over()

src/antares_web_installer/gui/widgets/frame.py

@@ -280,5 +280,5 @@ def __init__(self, master: tk.Misc, window: "WizardView", *args, **kwargs):
             style="Description.TLabel",
         ).pack(side="top", fill="x")
 
-        self.control_btn = ControlFrame(parent=self, window=window, finish_btn=True)
+        self.control_btn = ControlFrame(parent=self, window=window, cancel_btn=False, finish_btn=True)
         self.control_btn.pack(side="bottom", fill="x")

src/antares_web_installer/shortcuts/_win32_shell.py

@@ -11,13 +11,12 @@
 
 import os
 import typing as t
+from contextlib import contextmanager
 
+import pythoncom
 import win32com.client
 from win32com.shell import shell, shellcon
 
-_WSHELL = win32com.client.Dispatch("Wscript.Shell")
-
-
 # Windows Special Folders
 # see: https://docs.microsoft.com/en-us/windows/win32/shell/csidl
 
@@ -42,6 +41,15 @@ def get_start_menu() -> str:
     return shell.SHGetFolderPath(0, shellcon.CSIDL_PROGRAMS, None, 0)  # type: ignore
 
 
+@contextmanager
+def initialize_com():
+    try:
+        pythoncom.CoInitialize()
+        yield
+    finally:
+        pythoncom.CoUninitialize()
+
+
 def create_shortcut(
     target: t.Union[str, os.PathLike],
     exe_path: t.Union[str, os.PathLike],
@@ -56,13 +64,15 @@ def create_shortcut(
     if isinstance(arguments, str):
         arguments = [arguments] if arguments else []
 
-    wscript = _WSHELL.CreateShortCut(str(target))
-    wscript.TargetPath = str(exe_path)
-    wscript.Arguments = " ".join(arguments)
-    wscript.WorkingDirectory = str(working_dir)
-    wscript.WindowStyle = 0
-    if description:
-        wscript.Description = description
-    if icon_path:
-        wscript.IconLocation = str(icon_path)
-    wscript.save()
+    with initialize_com():
+        _WSHELL = win32com.client.Dispatch("Wscript.Shell")
+        wscript = _WSHELL.CreateShortCut(str(target))
+        wscript.TargetPath = str(exe_path)
+        wscript.Arguments = " ".join(arguments)
+        wscript.WorkingDirectory = str(working_dir)
+        wscript.WindowStyle = 0
+        if description:
+            wscript.Description = description
+        if icon_path:
+            wscript.IconLocation = str(icon_path)
+        wscript.save()

tests/config/test_config_2_18.py

@@ -0,0 +1,12 @@
+from pathlib import Path
+
+import yaml
+
+from antares_web_installer.config.config_2_18 import update_to_2_18
+
+
+def test_update_to_2_18(datadir: Path) -> None:
+    config = yaml.safe_load(datadir.joinpath("application-2.17.yaml").read_text())
+    expected = yaml.safe_load(datadir.joinpath("application-2.18.yaml").read_text())
+    update_to_2_18(config)
+    assert config == expected

tests/config/test_config_2_18/application-2.17.yaml

@@ -0,0 +1,42 @@
+security:
+  disabled: false
+  jwt:
+    key: super-secret
+  login:
+    admin:
+      pwd: admin
+
+db:
+  url: "sqlite:////home/john/antares_data/database.db"
+
+storage:
+  tmp_dir: /tmp
+  matrixstore: /home/john/antares_data/matrices
+  archive_dir: /home/john/antares_data/archives
+  allow_deletion: false
+  workspaces:
+    default:
+      path: /home/john/antares_data/internal_studies/
+    studies:
+      path: /home/john/antares_data/studies/
+
+launcher:
+  default: local
+  local:
+    binaries:
+      850: /home/john/opt/antares-8.5.0-Ubuntu-20.04/antares-solver
+      860: /home/john/opt/antares-8.6.0-Ubuntu-20.04/antares-8.6-solver
+    enable_nb_cores_detection: True
+
+debug: false
+
+root_path: "/api"
+
+server:
+  worker_threadpool_size: 12
+  services:
+    - watcher
+    - matrix_gc
+
+logging:
+  level: INFO