Add support for allowing unreleased branches in get_available_package_sets function

src/mpsd_software_manager/cmds/available.py

@@ -10,7 +10,7 @@ from mpsd_software_manager import config_vars
 from mpsd_software_manager.utils.filesystem_utils import clone_repo
 
 
-def get_available_package_sets(mpsd_release: str) -> List[str]:
+def get_available_package_sets(mpsd_release: str, allow_unreleased_branch: bool = False) -> List[str]:
     """Given a release, return the available package_sets.
 
     This is based on the spack-environment's repository [1]. For this function
@@ -48,10 +48,15 @@ def get_available_package_sets(mpsd_release: str) -> List[str]:
     tmp_dir_path = Path(tmp_dir.name)
 
     # find package_sets by cloning repository and checking out right branch
+    if allow_unreleased_branch:
+        spe_branch = mpsd_release
+    else:
+        spe_branch = f"releases/{mpsd_release}"
+
     clone_repo(
         tmp_dir_path,
         config_vars.spack_environments_repo,
-        branch=f"releases/{mpsd_release}",
+        branch=spe_branch,
     )
     # look for directories defining the package_sets
 

src/mpsd_software_manager/mpsd_software.py

@@ -99,6 +99,13 @@ def main():
 
     parser.add_argument("--version", action="version", version=__version__)
 
+    parser.add_argument(
+        "--develop",
+        action="store_true",
+        default=False,
+        help="Use the develop release version",
+    )
+
     subparsers = parser.add_subparsers(
         dest="action",
         title="actions",
@@ -249,7 +256,7 @@ def main():
         prepare_environment(args.release, root_dir)
     elif args.action == "available":
         if args.release:
-            get_available_package_sets(args.release)
+            get_available_package_sets(args.release, args.develop)
         else:
             get_available_releases(print_result=True)
             sys.exit(0)