GitRepository.restore_head(): new method

Author: mhagger

git-imerge

@@ -310,19 +310,15 @@ class GitTemporaryHead(object):
     """
 
     def __enter__(self, git, message):
+        self.git = git
         self.message = message
         self.head_name = git.get_head_refname()
         return self
 
     def __exit__(self, exc_type, exc_val, exc_tb):
         if self.head_name:
             try:
-                check_call([
-                    'git', 'symbolic-ref',
-                    '-m', self.message, 'HEAD',
-                    self.head_name,
-                    ])
-                check_call(['git', 'reset', '--hard'])
+                self.git.restore_head(self.head_name, self.message)
             except CalledProcessError as e:
                 raise Failure(
                     'Could not restore HEAD to %r!:    %s\n'
@@ -989,6 +985,10 @@ class GitRepository(object):
         except CalledProcessError:
             return None
 
+    def restore_head(self, refname, message):
+        check_call(['git', 'symbolic-ref', '-m', message, 'HEAD', refname])
+        check_call(['git', 'reset', '--hard'])
+
     def checkout(self, refname, quiet=False):
         cmd = ['git', 'checkout']
         if quiet: