Fixed #3: 'git up' didn't return to current branch

msiemens · May 3, 2013 · 6bd3aae · 6bd3aae
1 parent a7e048f
commit 6bd3aae
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/PyGitUp/gitup.py b/PyGitUp/gitup.py
@@ -205,7 +205,9 @@ def returning_to_current_branch(self):
 
         yield
 
-        print(colored('returning to {0}'.format(branch_name), 'magenta'))
+        if not self.repo.head.ref.name == branch_name:
+            print(colored('returning to {0}'.format(branch_name), 'magenta'))
+            self.git.checkout(branch_name)
 
     def config(self, key):
         """ Get a git-up-specific config value. """