ece4-exp 1.0.0__py3-none-any.whl

ece4_exp/cli.py

@@ -0,0 +1,289 @@
+#!/usr/bin/env python3
+"""
+CLI entry point for ece4-exp.
+
+This replaces the bash wrapper with a proper Python console_scripts entry point.
+"""
+
+import argparse
+import os
+import sys
+from pathlib import Path
+
+from . import paths
+from .yaml_util import log_info, log_warn, log_error, COLOR_CYAN, COLOR_NC
+
+
+def cmd_list(args):
+    """List available recipes."""
+    print(f"{COLOR_CYAN}Available Recipes:{COLOR_NC}\n")
+    print("Main Recipes:")
+
+    if paths.RECIPES_DIR.exists():
+        recipes = sorted(paths.RECIPES_DIR.glob("*.yml")) + sorted(paths.RECIPES_DIR.glob("*.yaml"))
+        for recipe in recipes:
+            print(f"  - {recipe.name}")
+    else:
+        log_warn(f"Recipes directory not found: {paths.RECIPES_DIR}")
+
+    weekly_tests = paths.RECIPES_DIR / "weekly_tests"
+    if weekly_tests.exists():
+        print("\nWeekly Tests:")
+        tests = sorted(weekly_tests.glob("*.yml")) + sorted(weekly_tests.glob("*.yaml"))
+        for test in tests:
+            print(f"  - {test.name}")
+
+    print()
+
+
+def cmd_info(args):
+    """Show current configuration info."""
+    from .yaml_util import set_quiet_mode
+
+    if args.quiet:
+        set_quiet_mode(True)
+
+    # Get EXPID from environment or guess
+    expid = os.environ.get("EXPID", "unknown")
+    if expid == "unknown":
+        # Try to guess from path structure
+        cwd_parts = Path.cwd().parts
+        if len(cwd_parts) >= 3:
+            guess = cwd_parts[-3]
+            if len(guess) == 4 and guess.isalnum():
+                expid = guess
+
+    conf_path = os.environ.get("CONF_PATH", f"/esarchive/autosubmit/{expid}/conf")
+    expdef_file = Path(conf_path) / f"expdef_{expid}.yml"
+    jobs_file = Path(conf_path) / f"jobs_{expid}.yml"
+
+    # Import and run generate with --info flag
+    import importlib
+    gec = importlib.import_module("ece4_exp.generate-experiment-config")
+
+    sys.argv = [
+        "ece4-exp",
+        "--expdef", str(expdef_file),
+        "--jobs", str(jobs_file),
+        "--info"
+    ]
+
+    if args.quiet:
+        sys.argv.append("--quiet")
+
+    gec.main()
+
+
+def cmd_init_user(args):
+    """Initialize user configuration."""
+    log_info("Initializing user configuration in ~/.config/ece4-exp")
+
+    from . import init_config
+    init_config.main()
+
+
+def cmd_generate(args):
+    """Generate experiment configuration."""
+    from .yaml_util import set_quiet_mode
+
+    if args.quiet:
+        set_quiet_mode(True)
+        os.environ["COLOR_NC"] = ""
+        os.environ["COLOR_GREEN"] = ""
+        os.environ["COLOR_CYAN"] = ""
+        os.environ["COLOR_YELLOW"] = ""
+        os.environ["COLOR_RED"] = ""
+
+    # Build sys.argv for the generate module
+    import importlib
+    gec = importlib.import_module("ece4_exp.generate-experiment-config")
+
+    # Get EXPID for default paths
+    expid = args.expid if hasattr(args, 'expid') and args.expid else os.environ.get("EXPID", "unknown")
+    conf_path = os.environ.get("CONF_PATH", f"/esarchive/autosubmit/{expid}/conf")
+
+    sys.argv = ["ece4-exp"]
+
+    # Add autosubmit files if they exist (backward compatibility)
+    expdef_file = Path(conf_path) / f"expdef_{expid}.yml"
+    jobs_file = Path(conf_path) / f"jobs_{expid}.yml"
+
+    if expdef_file.exists() and jobs_file.exists():
+        sys.argv.extend(["--expdef", str(expdef_file)])
+        sys.argv.extend(["--jobs", str(jobs_file)])
+
+    # Forward all args to generate module
+    if args.recipe:
+        sys.argv.extend(["--recipe", args.recipe])
+    if args.sim_procs:
+        sys.argv.extend(["--sim-procs", str(args.sim_procs)])
+    if args.expid:
+        sys.argv.extend(["--expid", args.expid])
+    if args.platform:
+        sys.argv.extend(["--platform", args.platform])
+    if args.launcher:
+        sys.argv.extend(["--launcher", args.launcher])
+    if args.kind:
+        sys.argv.extend(["--kind", args.kind])
+    if args.account:
+        sys.argv.extend(["--account", args.account])
+    if args.walltime:
+        sys.argv.extend(["--walltime", str(args.walltime)])
+    if args.description:
+        sys.argv.extend(["--description", args.description])
+    if args.repo_owner:
+        sys.argv.extend(["--repo-owner", args.repo_owner])
+    if args.repo_branch:
+        sys.argv.extend(["--repo-branch", args.repo_branch])
+    if args.output:
+        sys.argv.extend(["--output", args.output])
+    if args.dry_run:
+        sys.argv.append("--dry-run")
+    if args.quiet:
+        sys.argv.append("--quiet")
+
+    gec.main()
+
+
+def cmd_validate(args):
+    """Validate experiment configuration."""
+    import importlib
+    vec = importlib.import_module("ece4_exp.validate-experiment-config")
+
+    config_file = args.config_file
+    if not config_file:
+        # Default to EXPID_experiment.yml
+        expid = os.environ.get("EXPID", "unknown")
+        config_file = f"{expid}_experiment.yml"
+
+    if not Path(config_file).exists():
+        log_error(f"Configuration file not found: {config_file}")
+        sys.exit(1)
+
+    sys.argv = ["ece4-exp", config_file]
+    vec.main()
+
+
+def cmd_save(args):
+    """Save changes as a recipe."""
+    import importlib
+    srd = importlib.import_module("ece4_exp.save_recipe_from_diff")
+
+    # Get EXPID
+    expid = args.expid if args.expid else os.environ.get("EXPID", "unknown")
+    conf_path = os.environ.get("CONF_PATH", f"/esarchive/autosubmit/{expid}/conf")
+    expdef_file = Path(conf_path) / f"expdef_{expid}.yml"
+
+    # Output file
+    output_file = args.output if args.output else f"{expid}.yml"
+
+    log_info(f"Saving recipe: {COLOR_CYAN}{output_file}{COLOR_NC} (Expid: {expid})")
+
+    sys.argv = ["ece4-exp", "--expid", expid]
+
+    if expdef_file.exists():
+        sys.argv.extend(["--expdef", str(expdef_file)])
+
+    if args.recipe:
+        sys.argv.extend(["--recipe", args.recipe])
+
+    sys.argv.extend(["--output", output_file])
+
+    srd.main()
+
+
+def main():
+    """Main CLI entry point."""
+    parser = argparse.ArgumentParser(
+        prog="ece4-exp",
+        description="EC-Earth4 experiment configuration tool",
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog="""
+Examples:
+  ece4-exp list
+  ece4-exp generate --recipe gcm-sr.yml --sim-procs 1120 --expid test001
+  ece4-exp validate myexp.yml
+  ece4-exp save --expid myexp --recipe gcm-sr.yml -o my-recipe.yml
+
+For more help: ece4-exp <command> --help
+        """
+    )
+
+    subparsers = parser.add_subparsers(dest="command", help="Available commands")
+
+    # --- list ---
+    parser_list = subparsers.add_parser("list", help="List available recipes")
+    parser_list.set_defaults(func=cmd_list)
+
+    # --- info ---
+    parser_info = subparsers.add_parser("info", help="Show current configuration info")
+    parser_info.add_argument("--quiet", action="store_true", help="Suppress colored output")
+    parser_info.set_defaults(func=cmd_info)
+
+    # --- init-user ---
+    parser_init = subparsers.add_parser("init-user", help="Initialize user configuration")
+    parser_init.set_defaults(func=cmd_init_user)
+
+    # --- generate ---
+    parser_gen = subparsers.add_parser("generate", help="Generate experiment configuration")
+
+    # Core parameters
+    parser_gen.add_argument("--recipe", help="Recipe name (e.g. gcm-sr.yml)")
+    parser_gen.add_argument("--sim-procs", type=int, dest="sim_procs", help="Number of processors for SIM job")
+    parser_gen.add_argument("--expid", help="Experiment ID")
+
+    # Platform parameters
+    parser_gen.add_argument("--platform", help="HPC Platform (e.g. bsc-marenostrum5)")
+    parser_gen.add_argument("--launcher", help="Launcher type (e.g. slurm-wrapper-taskset)")
+    parser_gen.add_argument("--kind", help="Launcher kind (e.g. CPLD-SR, auto)")
+
+    # User parameters
+    parser_gen.add_argument("--account", help="HPC account/project")
+    parser_gen.add_argument("--walltime", type=int, help="Walltime in hours")
+    parser_gen.add_argument("--description", help="Experiment description")
+
+    # Repository parameters
+    parser_gen.add_argument("--repo-owner", dest="repo_owner", help="ECE4 repository owner")
+    parser_gen.add_argument("--repo-branch", dest="repo_branch", help="ECE4 repository branch")
+
+    # Output control
+    parser_gen.add_argument("-o", "--output", help="Output file name")
+    parser_gen.add_argument("--dry-run", action="store_true", help="Preview without writing")
+    parser_gen.add_argument("--quiet", action="store_true", help="Suppress colored output")
+
+    parser_gen.set_defaults(func=cmd_generate)
+
+    # --- validate ---
+    parser_val = subparsers.add_parser("validate", help="Validate experiment configuration")
+    parser_val.add_argument("config_file", nargs="?", help="Path to configuration file")
+    parser_val.set_defaults(func=cmd_validate)
+
+    # --- save ---
+    parser_save = subparsers.add_parser("save", help="Save changes as a recipe")
+    parser_save.add_argument("--expid", help="Experiment ID")
+    parser_save.add_argument("--recipe", help="Current user recipe name")
+    parser_save.add_argument("-o", "--output", help="Recipe file name")
+    parser_save.set_defaults(func=cmd_save)
+
+    # Parse args
+    args = parser.parse_args()
+
+    if not args.command:
+        parser.print_help()
+        sys.exit(1)
+
+    # Run command
+    try:
+        args.func(args)
+    except KeyboardInterrupt:
+        print("\nInterrupted by user")
+        sys.exit(130)
+    except Exception as e:
+        log_error(f"Command failed: {e}")
+        if "--debug" in sys.argv or os.environ.get("DEBUG"):
+            raise
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()