Mercurial > hg-git
view hggit/git2hg.py @ 794:1350e43e662f
git2hg.find_incoming: move graph traversal into a function
This is preparation for upcoming changes to find_incoming that will allow it to
import certain Git branches as Mercurial named branches.
author | Siddharth Agarwal <sid0@fb.com> |
---|---|
date | Wed, 15 Oct 2014 14:21:09 -0700 |
parents | 388944fca782 |
children | c19835c3c60d |
line wrap: on
line source
# git2hg.py - convert Git repositories and commits to Mercurial ones from dulwich.objects import Commit, Tag def find_incoming(git_object_store, git_map, refs): '''find what commits need to be imported git_object_store is a dulwich object store. git_map is a map with keys being Git commits that have already been imported refs is a map of refs to SHAs that we're interested in.''' done = set() commit_cache = {} # sort by commit date def commitdate(sha): obj = git_object_store[sha] return obj.commit_time-obj.commit_timezone # get a list of all the head shas def get_heads(refs): todo = [] seenheads = set() for sha in refs.itervalues(): # refs could contain refs on the server that we haven't pulled down # the objects for if sha in git_object_store: obj = git_object_store[sha] while isinstance(obj, Tag): obj_type, sha = obj.object obj = git_object_store[sha] if isinstance(obj, Commit) and sha not in seenheads: seenheads.add(sha) todo.append(sha) todo.sort(key=commitdate, reverse=True) return todo def get_unseen_commits(todo): '''get all unseen commits reachable from todo in topological order 'unseen' means not reachable from the done set and not in the git map. Mutates todo and the done set in the process.''' commits = [] while todo: sha = todo[-1] if sha in done or sha in git_map: todo.pop() continue assert isinstance(sha, str) if sha in commit_cache: obj = commit_cache[sha] else: obj = git_object_store[sha] commit_cache[sha] = obj assert isinstance(obj, Commit) for p in obj.parents: if p not in done and p not in git_map: todo.append(p) # process parents of a commit before processing the # commit itself, and come back to this commit later break else: commits.append(sha) done.add(sha) todo.pop() return commits todo = get_heads(refs) commits = get_unseen_commits(todo) return GitIncomingResult(commits, commit_cache) class GitIncomingResult(object): '''struct to store result from find_incoming''' def __init__(self, commits, commit_cache): self.commits = commits self.commit_cache = commit_cache