Make repo branch configurable (#29)

ComplianceAsCode · Aug 23, 2024 · c2f01c1 · c2f01c1
1 parent 190cbc2
commit c2f01c1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/harvest/__init__.py b/harvest/__init__.py
@@ -13,4 +13,4 @@
 # limitations under the License.
 """The Auditree file collating and reporting tool."""
 
-__version__ = "1.3.0"
+__version__ = "1.3.1"
diff --git a/harvest/cli.py b/harvest/cli.py
@@ -44,6 +44,14 @@ def _init_arguments(self):
                 "or: local - if working exclusively with a local git repo"
             ),
         )
+        self.add_argument(
+            "--branch",
+            help=(
+                "which branch you wish to retrieve git repo evidence from - "
+                "default is master"
+            ),
+            default="master",
+        )
         self.add_argument(
             "--repo-path",
             help=(
@@ -140,7 +148,7 @@ def _run(self, args):
         collator = Collator(
             args.repo,
             Config(args.creds) if args.creds else None,
-            "master",
+            args.branch,
             args.repo_path,
             args.no_validate,
             include_file_path=args.include_file_path,
@@ -196,7 +204,7 @@ def _run(self, args):
         reporter = self.report(
             args.repo,
             Config(args.creds) if args.creds else None,
-            "master",
+            args.branch,
             args.repo_path,
             self.template_dir,
             args.no_validate,