diff --git a/prune/cli.py b/prune/cli.py index e5d4594..ae5684e 100644 --- a/prune/cli.py +++ b/prune/cli.py @@ -20,6 +20,7 @@ from urllib.parse import urlparse from compliance.utils.credentials import Config +from compliance.config import get_config from ilcli import Command @@ -36,6 +37,11 @@ def _init_arguments(self): "as an example https://github.com/my-org/my-repo" ), ) + self.add_argument( + "--branch", + help="Branch name for locker repository", + default=False, + ) self.add_argument( "--creds", metavar="~/path/creds", @@ -88,6 +94,10 @@ def _run(self, args): gitconfig = None if args.git_config or args.git_config_file: gitconfig = args.git_config or json.loads(open(args.git_config_file).read()) + if args.branch: + c = get_config() + c.load() + c.raw_config["locker"]["default_branch"] = args.branch # self.name drives the Locker push mode. # - dry-run translates to locker no-push mode # - push-remote translates to locker full-remote mode