summaryrefslogtreecommitdiff
path: root/git_remote_helpers/git/importer.py
blob: d3f90e1024a1b3ac8e7b4204b873e0ac2fffeaf9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import os
import subprocess

from git_remote_helpers.util import check_call, check_output


class GitImporter(object):
    """An importer for testgit repositories.

    This importer simply delegates to git fast-import.
    """

    def __init__(self, repo):
        """Creates a new importer for the specified repo.
        """

        self.repo = repo

    def get_refs(self, gitdir):
        """Returns a dictionary with refs.

        Note that the keys in the returned dictionary are byte strings as
        read from git.
        """
        args = ["git", "--git-dir=" + gitdir, "for-each-ref", "refs/heads"]
        lines = check_output(args).strip().split('\n'.encode('ascii'))
        refs = {}
        for line in lines:
            value, name = line.split(' '.encode('ascii'))
            name = name.strip('commit\t'.encode('ascii'))
            refs[name] = value
        return refs

    def do_import(self, base):
        """Imports a fast-import stream to the given directory.

        Simply delegates to git fast-import.
        """

        dirname = self.repo.get_base_path(base)
        if self.repo.local:
            gitdir = self.repo.gitpath
        else:
            gitdir = os.path.abspath(os.path.join(dirname, '.git'))
        path = os.path.abspath(os.path.join(dirname, 'testgit.marks'))

        if not os.path.exists(dirname):
            os.makedirs(dirname)

        refs_before = self.get_refs(gitdir)

        args = ["git", "--git-dir=" + gitdir, "fast-import", "--quiet", "--export-marks=" + path]

        if os.path.exists(path):
            args.append("--import-marks=" + path)

        check_call(args)

        refs_after = self.get_refs(gitdir)

        changed = {}

        for name, value in refs_after.iteritems():
            if refs_before.get(name) == value:
                continue

            changed[name] = value

        return changed