Specify branch for hook

tupui · Aug 24, 2024 · 993cb08 · 993cb08
1 parent 0a91e41
commit 993cb08
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pre-commit/soroban_versioning_pre_push.py b/pre-commit/soroban_versioning_pre_push.py
@@ -22,6 +22,8 @@
 if PROJECT_KEY is None:
     raise ValueError("'TANSU_PROJECT_KEY' is missing from the environment")
 
+BRANCH = os.getenv("TANSU_BRANCH", "main")
+
 
 def main():
     project_key = bytes.fromhex(PROJECT_KEY)
@@ -31,7 +33,7 @@ def main():
         .stdout.decode()
         .strip("\n")
     )
-    if branch != "main":
+    if branch != BRANCH:
         exit(0)
 
     commit_hash = (