Remove hard-coded master branch reference

adobe · Dec 13, 2023 · 7311d56 · 7311d56
1 parent c44b13c
commit 7311d56
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pydc_control/commands.py b/pydc_control/commands.py
@@ -243,9 +243,14 @@ def run_checkout(args: argparse.Namespace):
             continue
 
         if os.path.exists(project.path):
-            log.get_logger().info(f'########### {project.name} (pulling changes) ###########')
-            commands = ['git', 'pull', 'upstream', 'master']
             cwd = project.path
+            head_branch = subprocess.check_output(
+                ['git', 'rev-parse', '--abbrev-ref', 'HEAD'],
+                encoding='utf8',
+                cwd=cwd,
+            ).strip()
+            log.get_logger().info(f'########### {project.name} (pulling changes for {head_branch}) ###########')
+            commands = ['git', 'pull', 'upstream', head_branch]
         else:
             log.get_logger().info(f'########### {project.name} (cloning) ###########')
             commands = ['git', 'clone', '--origin', 'upstream', project.repository]