Add a cli argument for overriding the locker branch

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