Add a cli argument for overriding the locker branch (#13)

* Add a cli argument for overriding the locker branch

* Fix formatting

plant/cli.py

@@ -21,6 +21,7 @@
 
 from compliance.evidence import ExternalEvidence, YEAR
 from compliance.utils.credentials import Config
+from compliance.config import get_config
 
 from ilcli import Command
 
@@ -37,6 +38,9 @@ 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",
@@ -106,6 +110,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