expose branch name

universalcore · May 19, 2015 · 8a04cf3 · 8a04cf3
1 parent 1c9c8b5
commit 8a04cf3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/unicore/distribute/api/tests/test_utils.py b/unicore/distribute/api/tests/test_utils.py
@@ -67,7 +67,12 @@ def test_format_repo(self):
         formatted = format_repo(self.workspace.repo)
         self.assertEqual(
             set(formatted.keys()),
-            set(['name', 'commit', 'timestamp', 'author', 'schemas']))
+            set(['name',
+                 'commit',
+                 'timestamp',
+                 'author',
+                 'schemas',
+                 'branch']))
         last_commit = self.workspace.repo.commit()
         self.assertEqual(formatted['commit'], last_commit.hexsha)
         self.assertEqual(

diff --git a/unicore/distribute/utils.py b/unicore/distribute/utils.py
@@ -150,6 +150,7 @@ def format_repo(repo):
     commit = repo.commit()
     return {
         'name': os.path.basename(repo.working_dir),
+        'branch': repo.active_branch.name,
         'commit': commit.hexsha,
         'timestamp': datetime.fromtimestamp(
             commit.committed_date).isoformat(),