Handle detached head in legit branches

frostming · Apr 9, 2012 · 684d7b7 · 684d7b7
1 parent ee3325d
commit 684d7b7
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/legit/cli.py b/legit/cli.py
@@ -441,8 +441,13 @@ def display_available_branches():
 
     for branch in branches:
 
-        marker = '*' if (branch.name == repo.head.ref.name) else ' '
-        color = colored.green if (branch.name == repo.head.ref.name) else colored.yellow
+        try:
+            branch_is_selected = (branch.name == repo.head.ref.name)
+        except TypeError:
+            branch_is_selected = False
+
+        marker = '*' if branch_is_selected else ' '
+        color = colored.green if branch_is_selected else colored.yellow
         pub = '(published)' if branch.is_published else '(unpublished)'
 
         print columns(
@@ -520,4 +525,4 @@ def handle_abort(aborted):
     hv='harvest',
     har='harvest',
     h='help'
-)
+)