Add config option to be able to use non-default charon configuration

Author: ligangty

charon/cmd/cmd_cache.py

@@ -58,6 +58,14 @@
     format of CF defining too, and each path should be in a single line.
     """
 )
+@option(
+    "--config",
+    "-c",
+    help="""
+    The charon configuration yaml file path. Default is
+    $HOME/.charon/charon.yaml
+    """
+)
 @option(
     "--debug",
     "-D",
@@ -79,6 +87,7 @@ def invalidate(
     target: str,
     paths: List[str],
     path_file: str,
+    config: str = None,
     quiet: bool = False,
     debug: bool = False
 ):
@@ -110,7 +119,7 @@ def invalidate(
             break
 
     try:
-        (buckets, aws_profile) = _init_cmd(target)
+        (buckets, aws_profile) = _init_cmd(target, config)
 
         for b in buckets:
             cf_client = CFClient(aws_profile=aws_profile)
@@ -144,6 +153,14 @@ def invalidate(
     """,
     required=True
 )
+@option(
+    "--config",
+    "-c",
+    help="""
+    The charon configuration yaml file path. Default is
+    $HOME/.charon/charon.yaml
+    """
+)
 @option(
     "--debug",
     "-D",
@@ -164,6 +181,7 @@ def invalidate(
 def check(
     invalidation_id: str,
     target: str,
+    config: str = None,
     quiet: bool = False,
     debug: bool = False
 ):
@@ -175,7 +193,7 @@ def check(
         is_quiet=quiet, is_debug=debug, use_log_file=False
     )
     try:
-        (buckets, aws_profile) = _init_cmd(target)
+        (buckets, aws_profile) = _init_cmd(target, config)
         if not buckets:
             sys.exit(1)
 
@@ -203,8 +221,8 @@ def check(
         sys.exit(2)
 
 
-def _init_cmd(target: str) -> Tuple[List[Tuple[str, str, str, str, str]], str]:
-    conf = get_config()
+def _init_cmd(target: str, config: str) -> Tuple[List[Tuple[str, str, str, str, str]], str]:
+    conf = get_config(config)
     if not conf:
         sys.exit(1)
 

charon/cmd/cmd_checksum.py

@@ -191,11 +191,20 @@ def validate(
     """,
     required=True
 )
+@option(
+    "--config",
+    "-c",
+    help="""
+    The charon configuration yaml file path. Default is
+    $HOME/.charon/charon.yaml
+    """
+)
 @command()
 def refresh(
     target: str,
     paths: List[str],
     path_file: str,
+    config: str = None,
     quiet: bool = False,
     debug: bool = False
 ):
@@ -219,7 +228,7 @@ def refresh(
     if paths:
         work_paths.extend(paths)
 
-    conf = get_config()
+    conf = get_config(config)
     aws_profile = os.getenv("AWS_PROFILE") or conf.get_aws_profile()
     if not aws_profile:
         logger.error("No AWS profile specified!")

charon/cmd/cmd_delete.py

@@ -96,6 +96,14 @@
     be extracted, when needed.
     """,
 )
+@option(
+    "--config",
+    "-c",
+    help="""
+    The charon configuration yaml file path. Default is
+    $HOME/.charon/charon.yaml
+    """
+)
 @option(
     "--debug",
     "-D",
@@ -120,6 +128,7 @@ def delete(
     root_path="maven-repository",
     ignore_patterns: List[str] = None,
     work_dir: str = None,
+    config: str = None,
     debug=False,
     quiet=False,
     dryrun=False
@@ -136,7 +145,7 @@ def delete(
                         "no files will be deleted.")
         if not _validate_prod_key(product, version):
             return
-        conf = get_config()
+        conf = get_config(config)
         if not conf:
             sys.exit(1)
 

charon/cmd/cmd_index.py

@@ -42,6 +42,14 @@
     """,
     required=True
 )
+@option(
+    "--config",
+    "-c",
+    help="""
+    The charon configuration yaml file path. Default is
+    $HOME/.charon/charon.yaml
+    """
+)
 @option(
     "--debug",
     "-D",
@@ -61,6 +69,7 @@
 def index(
     path: str,
     target: str,
+    config: str = None,
     debug: bool = False,
     quiet: bool = False,
     dryrun: bool = False
@@ -73,7 +82,7 @@ def index(
         is_quiet=quiet, is_debug=debug, use_log_file=False
     )
     try:
-        conf = get_config()
+        conf = get_config(config)
         if not conf:
             sys.exit(1)
 

charon/cmd/cmd_upload.py

@@ -97,6 +97,14 @@
     be extracted, when needed.
     """,
 )
+@option(
+    "--config",
+    "-c",
+    help="""
+    The charon configuration yaml file path. Default is
+    $HOME/.charon/charon.yaml
+    """
+)
 @option(
     "--contain_signature",
     "-s",
@@ -137,6 +145,7 @@ def upload(
     root_path="maven-repository",
     ignore_patterns: List[str] = None,
     work_dir: str = None,
+    config: str = None,
     contain_signature: bool = False,
     sign_key: str = "redhatdevel",
     debug=False,
@@ -155,7 +164,7 @@ def upload(
                         "no files will be uploaded.")
         if not _validate_prod_key(product, version):
             return
-        conf = get_config()
+        conf = get_config(config)
         if not conf:
             sys.exit(1)
 
@@ -211,7 +220,8 @@ def upload(
                 cf_enable=conf.is_aws_cf_enable(),
                 key=sign_key,
                 dry_run=dryrun,
-                manifest_bucket_name=manifest_bucket_name
+                manifest_bucket_name=manifest_bucket_name,
+                config=config
             )
             if not succeeded:
                 sys.exit(1)

charon/config.py

@@ -68,8 +68,10 @@ def is_aws_cf_enable(self) -> bool:
         return self.__aws_cf_enable
 
 
-def get_config() -> Optional[CharonConfig]:
-    config_file_path = os.path.join(os.getenv("HOME"), ".charon", CONFIG_FILE)
+def get_config(cfgPath=None) -> Optional[CharonConfig]:
+    config_file_path = cfgPath
+    if not config_file_path or not os.path.isfile(config_file_path):
+        config_file_path = os.path.join(os.getenv("HOME"), ".charon", CONFIG_FILE)
     data = read_yaml_from_file_path(config_file_path, 'schemas/charon.json')
     return CharonConfig(data)
 

charon/pkgs/maven.py

@@ -272,7 +272,8 @@ def handle_maven_uploading(
     cf_enable=False,
     key=None,
     dry_run=False,
-    manifest_bucket_name=None
+    manifest_bucket_name=None,
+    config=None
 ) -> Tuple[str, bool]:
     """ Handle the maven product release tarball uploading process.
         * repo is the location of the tarball in filesystem
@@ -406,7 +407,7 @@ def handle_maven_uploading(
 
         # 10. Generate signature file if contain_signature is set to True
         if gen_sign:
-            conf = get_config()
+            conf = get_config(config)
             if not conf:
                 sys.exit(1)
             suffix_list = __get_suffix(PACKAGE_TYPE_MAVEN, conf)

charon/pkgs/npm.py

@@ -87,7 +87,8 @@ def handle_npm_uploading(
         cf_enable=False,
         key=None,
         dry_run=False,
-        manifest_bucket_name=None
+        manifest_bucket_name=None,
+        config=None
 ) -> Tuple[str, bool]:
     """ Handle the npm product release tarball uploading process.
         For NPM uploading, tgz file and version metadata will be relocated based
@@ -189,7 +190,7 @@ def handle_npm_uploading(
             logger.info("package.json uploading done")
 
         if gen_sign:
-            conf = get_config()
+            conf = get_config(config)
             if not conf:
                 sys.exit(1)
             suffix_list = __get_suffix(PACKAGE_TYPE_NPM, conf)

tests/test_config.py

@@ -22,8 +22,10 @@
 
 import charon.config as config
 import re
+import tempfile
 
 from charon.constants import DEFAULT_REGISTRY
+from charon.utils.files import overwrite_file
 from tests.base import BaseTest
 
 
@@ -49,6 +51,26 @@ def test_no_config(self):
         with pytest.raises(FileNotFoundError):
             config.get_config()
 
+    def test_non_default_config(self):
+        self.__base.setUp()
+        config_content = """
+ignore_patterns:
+    - ".*^(redhat).*"
+
+targets:
+    changed:
+    - bucket: changed-bucket
+      prefix: changed-prefix
+        """
+        _, tmp_config_file = tempfile.mkstemp(prefix="charon-test-config", suffix=".yaml")
+        overwrite_file(tmp_config_file, config_content)
+        conf = config.get_config(tmp_config_file)
+        self.assertIsNotNone(conf)
+        self.assertEqual("changed-bucket", conf.get_target("changed")[0].get('bucket', ''))
+        self.assertEqual("changed-prefix", conf.get_target("changed")[0].get('prefix', ''))
+        self.assertEqual([], conf.get_target("ga"))
+        os.remove(tmp_config_file)
+
     def test_config_missing_targets(self):
         content_missing_targets = """
 ignore_patterns: