Update __main__.py

Author: AAriam

.manager/src/proman/__main__.py

@@ -1,5 +1,7 @@
 import argparse
+import json
 
+import actionman
 from loggerman import logger
 
 from proman import script
@@ -18,14 +20,96 @@ def get_recursive(parts, current_object):
         return get_recursive(parts, script)
 
     initialize_logger()
+    args = _read_args()
+    logger.debug("Input Arguments", args)
+    endpoint = get_endpoint(args.endpoint)
+    endpoint(args)
+    return
+
+
+def _read_args() -> argparse.Namespace:
+    """Read inputs and return them as a namespace object."""
+    args = _parse_args()
+    _set_github_context(args)
+    _set_github_token(args)
+    _set_github_admin_token(args)
+    _set_zenodo_token(args)
+    _set_zenodo_sandbox_token(args)
+    return args
+
+
+def _set_github_context(args: argparse.Namespace) -> None:
+    args.github_context = json.loads(
+        args.github_context
+    ) if "github_context" in args else actionman.env_var.read("GITHUB_CONTEXT", typ=dict, remove=args.remove_tokens)
+    if args.github_context:
+        github_token = args.github_context.pop("token")
+        if github_token:
+            args.github_token = github_token
+    return
+
+
+def _set_github_token(args: argparse.Namespace) -> None:
+    if "github_token" in args:
+        return
+    for env_var_name in ("GITHUB_TOKEN", "GITHUB_ADMIN_TOKEN"):
+        token = actionman.env_var.read(env_var_name, typ=str, remove=args.remove_tokens)
+        if token:
+            args.github_token = token
+            return
+    token = _pyshellman.run(
+        command=["gh", "auth", "token"],
+        logger=logger,
+        raise_execution=False,
+        raise_exit_code=False,
+        raise_stderr=False,
+    ).out
+    if token:
+        args.github_token = token
+    return
+
+
+def _set_github_admin_token(args: argparse.Namespace) -> None:
+    if "github_admin_token" in args:
+        return
+    for env_var_name in ("GITHUB_ADMIN_TOKEN", ):
+        token = actionman.env_var.read(env_var_name, typ=str, remove=args.remove_tokens)
+        if token:
+            args.github_admin_token = token
+            return
+    if "github_token" in args:
+        args.github_admin_token = args.github_token
+    return
+
+
+def _set_zenodo_token(args: argparse.Namespace) -> None:
+    if "zenodo_token" in args:
+        return
+    args.zenodo_token = actionman.env_var.read("ZENODO_TOKEN", typ=str, remove=args.remove_tokens)
+    return
+
+
+def _set_zenodo_sandbox_token(args: argparse.Namespace) -> None:
+    if "zenodo_sandbox_token" in args:
+        return
+    args.zenodo_sandbox_token = actionman.env_var.read("ZENODO_SANDBOX_TOKEN", typ=str, remove=args.remove_tokens)
+    return
 
+
+def _parse_args() -> argparse.Namespace:
+    """Generate the command line interface for the project manager and parse input arguments.
+
+    Notes
+    -----
+    The code for this function is automatically generated by the control center.
+    """
     # begin auto-generated parser
     parser = argparse.ArgumentParser(description="Project Manager CLI")
     parser.add_argument("--repo", type=str, help="Local path to the repository root directory.", default="./")
     # Sub-parsers for parser
     subparsers_main = parser.add_subparsers(dest="command", required=True)
     subparser_cca = subparsers_main.add_parser("cca", help="Run Continuous Configuration Automation on the repository.")
-    subparser_cca.add_argument("-t", "--token", type=str, help="GitHub token for accessing the repository.")
+    subparser_cca.add_argument("-t", "--github-token", type=str, help="GitHub token for accessing the repository.")
     subparser_cca.add_argument("-b", "--branch-version", help="Branch-name to version mappings (e.g., -b main=0.0.0 dev=1.0.0a1) to use instead of git tags.", type=str, nargs="*", metavar="BRNACH=VERSION")
     subparser_cca.add_argument("-c", "--control-center", help="Path to the control center directory containing configuration files.", type=str)
     subparser_cca.add_argument("-d", "--dry-run", help="Perform a dry run without making any changes.", action="store_true")
@@ -89,11 +173,7 @@ def get_recursive(parts, current_object):
         if (args.from_ref and not args.to_ref) or (args.to_ref and not args.from_ref):
             parser.error("Both --from-ref and --to-ref must be provided together.")
     # end auto-generated parser
-
-    logger.debug("Input Arguments", args)
-    endpoint = get_endpoint(args.endpoint)
-    endpoint(args)
-    return
+    return args
 
 
 if __name__ == "__main__":