git_remote_helpers/git/exporter.py

   1 import os
   2 import subprocess
   3 import sys
   4
   5
   6 class GitExporter(object):
   7     """An exporter for testgit repositories.
   8
   9     The exporter simply delegates to git fast-export.
  10     """
  11
  12     def __init__(self, repo):
  13         """Creates a new exporter for the specified repo.
  14         """
  15
  16         self.repo = repo
  17
  18     def export_repo(self, base, refs=None):
  19         """Exports a fast-export stream for the given directory.
  20
  21         Simply delegates to git fast-epxort and pipes it through sed
  22         to make the refs show up under the prefix rather than the
  23         default refs/heads. This is to demonstrate how the export
  24         data can be stored under it's own ref (using the refspec
  25         capability).
  26
  27         If None, refs defaults to ["HEAD"].
  28         """
  29
  30         if not refs:
  31             refs = ["HEAD"]
  32
  33         dirname = self.repo.get_base_path(base)
  34         path = os.path.abspath(os.path.join(dirname, 'testgit.marks'))
  35
  36         if not os.path.exists(dirname):
  37             os.makedirs(dirname)
  38
  39         print "feature relative-marks"
  40         if os.path.exists(os.path.join(dirname, 'git.marks')):
  41             print "feature import-marks=%s/git.marks" % self.repo.hash
  42         print "feature export-marks=%s/git.marks" % self.repo.hash
  43         sys.stdout.flush()
  44
  45         args = ["git", "--git-dir=" + self.repo.gitpath, "fast-export", "--export-marks=" + path]
  46
  47         if os.path.exists(path):
  48             args.append("--import-marks=" + path)
  49
  50         args.extend(refs)
  51
  52         p1 = subprocess.Popen(args, stdout=subprocess.PIPE)
  53
  54         args = ["sed", "s_refs/heads/_" + self.repo.prefix + "_g"]
  55
  56         child = subprocess.Popen(args, stdin=p1.stdout)
  57         if child.wait() != 0:
  58             raise CalledProcessError