Add function to install DTU HAWCStab2 license

wetb/utils/installer.py

@@ -131,10 +131,27 @@ def install_wind_tool(
 
 def install_hawc2_dtu_license():
     """Function to install the DTU HAWC2 license. In order to install the license, you must be logged in to the DTU network."""
+    install_dtu_license("hawc2")
+
+
+def install_hawcstab2_dtu_license():
+    """Function to install the DTU HAWCStab2 license. In order to install the license, you must be logged in to the DTU network."""
+    install_dtu_license("hawcstab2")
+
+
+def install_ellipsys_dtu_license():
+    """Function to install the DTU HAWCStab2 license. In order to install the license, you must be logged in to the DTU network."""
+    install_dtu_license("ellipsys")
+
+
+def install_dtu_license(software : str):
+    """Function to install the DTU online license for HAWC2, HAWCStab2 and Ellipsys. In order to install the license, you must be logged in to the DTU network."""
+    software = software.lower()
+    assert software in ["hawc2", "hawcstab2", "ellipsys"], "Argument 'software' must be one of ['hawc2', 'hawcstab2,' 'ellipsys']"
     if sys.platform.lower() == "win32":
-        f = Path(os.getenv("APPDATA")) / "DTU Wind Energy/hawc2/license.cfg"
+        f = Path(os.getenv("APPDATA")) / f"DTU Wind Energy/{software}/license.cfg"
     else:
-        f = Path.home() / ".config/hawc2/license.cfg"
+        f = Path.home() / f".config/{software}/license.cfg"
     if not f.exists():
         f.parent.mkdir(parents=True, exist_ok=True)
         r = urlopen("http://license-internal.windenergy.dtu.dk:34523").read()
@@ -148,6 +165,8 @@ def install_hawc2_dtu_license():
             )
 
 
+
+
 def install_keygen_license(software: str, cfg_file: str, force: bool = False):
     """Install license file for HAWC2, HAWCStab2 or Ellipsys on your machine
 

wetb/utils/tests/test_installer.py

@@ -5,7 +5,7 @@ import os
 
 import pytest
 
-from wetb.utils.installer import install_wind_tool, install_hawc2_dtu_license, install_keygen_license
+from wetb.utils.installer import install_wind_tool, install_hawc2_dtu_license, install_keygen_license, install_hawcstab2_dtu_license, install_ellipsys_dtu_license
 import shutil
 
 DESTINATION="/tmp/hawc2"
@@ -82,15 +82,21 @@ def test_version_not_available():
         shutil.rmtree(DESTINATION)
 
 
-def test_install_hawc2_dtu_license():
+@pytest.mark.parametrize("software", ["HAwC2","HAWCStab2", "ellipsys"])
+def test_install_dtu_license(software):
+    software = software.lower()
     license_path = local_license_dir(USER_PLATFORM, "HAWC2")
-    try:
+
+    if software == "hawc2":
         install_hawc2_dtu_license()
-        assert os.path.exists(f"{local_license_dir(USER_PLATFORM, 'hawc2')}/{local_license_file('hawc2')}")
-    except:
-        raise
-    finally:
-        shutil.rmtree(license_path, ignore_errors=True)
+    elif software == "hawcstab2":
+        install_hawcstab2_dtu_license()
+    elif software == "ellipsys":
+        install_ellipsys_dtu_license()
+        
+    assert os.path.exists(f"{local_license_dir(USER_PLATFORM, software)}/{local_license_file(software)}")
+
+    shutil.rmtree(license_path, ignore_errors=True)