hg-git: git_handler.py comparison

comparison git_handler.py @ 51:1421d04f1ad2

merged from dulwich

author	Scott Chacon <schacon@gmail.com>
date	Wed, 29 Apr 2009 11:04:45 -0700
parents	3b62270c1fad d274092e3b24
children	87d462a6b796

comparison

equal deleted inserted replaced

-:3b62270c1fad
+:1421d04f1ad2
 ShaFile,
 Tag,
 Tree,
 hex_to_sha
 )
 import math
 def seconds_to_offset(time):
 hours = (float(time) / 60 / 60)
 hour_diff = math.fmod(time, 60)
 def map_git_get(self, hgsha):
 if hgsha in self._map_hg:
 return self._map_hg[hgsha]
 else:
 return None
 def load_map(self):
 self._map_git = {}
 self._map_hg = {}
 if os.path.exists(self.repo.join('git-mapfile')):
 for line in self.repo.opener('git-mapfile'):
 if key in self._config:
 name = self._config[key]
 print "URL for " + remote_name + " : " + name
 else:
 print "No remote named : " + remote_name
 return
 def remote_list(self):
 for key, value in self._config.iteritems():
 if key[0:6] == 'remote':
 print key + "\t" + value
 def remote_name_to_url(self, remote_name):
 return self._config['remote.' + remote_name + '.url']
 def update_references(self):
 # TODO : if bookmarks exist, add them as git branches
 c = self.map_git_get(hex(self.repo.changelog.tip()))
 self.git.set_ref('refs/heads/master', c)
 def export_git_objects(self):
 print "exporting git objects"
 for rev in self.repo.changelog:
 self.export_hg_commit(rev)
 # convert this commit into git objects
 # go through the manifest, convert all blobs/trees we don't have
 # write the commit object (with metadata info)
 def export_hg_commit(self, rev):
 # return if we've already processed this
 node = self.repo.changelog.lookup(rev)
 phgsha = hex(node)
 pgit_sha = self.map_git_get(phgsha)
 if pgit_sha:
 return pgit_sha
 print "converting revision " + str(rev)
 # make sure parents are converted first
 parents = self.repo.parents(rev)
 for parent in parents:
 p_rev = parent.rev()
 hgsha = hex(parent.node())
 git_sha = self.map_git_get(hgsha)
 if not p_rev == -1:
 if not git_sha:
 self.export_hg_commit(p_rev)
 ctx = self.repo.changectx(rev)
 tree_sha = self.write_git_tree(ctx)
 # TODO : something with tags?
 # TODO : explicit file renaming, copying?
 commit = {}
 commit['tree'] = tree_sha
 (time, timezone) = ctx.date()
 commit['author'] = ctx.user() + ' ' + str(int(time)) + ' ' + seconds_to_offset(timezone)
 message = ctx.description()
 commit['message'] = ctx.description()
 commit['message'] += "\n\n--HG--\n"
 commit['message'] += "branch : " + ctx.branch() + "\n"
 commit['parents'] = []
 for parent in parents:
 hgsha = hex(parent.node())
 git_sha = self.map_git_get(hgsha)
 if git_sha:
 commit['parents'].append(git_sha)
 commit_sha = self.git.write_commit_hash(commit) # writing new blobs to git
 self.map_set(commit_sha, phgsha)
 return commit_sha
 def write_git_tree(self, ctx):
 trees = {}
 man = ctx.manifest()
 for filenm in man.keys():
 # write blob if not in our git database
 fctx = ctx.filectx(filenm)
 is_exec = 'x' in fctx.flags()
 is_link = 'l' in fctx.flags()
 file_id = hex(fctx.filenode())
 if parentpath not in trees:
 trees[parentpath] = []
 if treeentry not in trees[parentpath]:
 trees[parentpath].append( treeentry )
 else:
 fileentry = ['blob', parts[0], blob_sha, is_exec, is_link]
 if '/' not in trees:
 trees['/'] = []
 trees['/'].append(fileentry)
 # sort by tree depth, so we write the deepest trees first
 dirs = trees.keys()
 dirs.sort(lambda a, b: len(b.split('/'))-len(a.split('/')))
 dirs.remove('/')
 dirs.append('/')
 # write all the trees
 tree_sha = None
 tree_shas = {}
 for dirnm in dirs:
 tree_data = []
 entry[2] = sha
 tree_data.append(entry)
 tree_sha = self.git.write_tree_array(tree_data) # writing new trees to git
 tree_shas[dirnm] = tree_sha
 return tree_sha # should be the last root tree sha
 def remote_head(self, remote_name):
 for head, sha in self.git.remote_refs(remote_name).iteritems():
 if head == 'HEAD':
 return self.map_hg_get(sha)
 return None
 except:
 raise
 # TODO : for now, we'll just push all heads that match remote heads
 #        * we should have specified push, tracking branches and --all
 # takes a dict of refs:shas from the server and returns what should be
 # pushed up
 def get_changed_refs(self, refs):
 keys = refs.keys()
 changed = []
 if not keys:
 return None
 # TODO : this is a huge hack
 if keys[0] == 'capabilities^{}': # nothing on the server yet - first push
 changed.append(("0"*40, self.git.ref('master'), 'refs/heads/master'))
 for ref_name in keys:
 parts = ref_name.split('/')
 if parts[0] == 'refs': # strip off 'refs/heads'
 if parts[1] == 'heads':
 head = "/".join([v for v in parts[2:]])
 local_ref = self.git.ref(ref_name)
 if local_ref:
 if not local_ref == refs[ref_name]:
 changed.append((refs[ref_name], local_ref, ref_name))
 return changed
 # takes a list of shas the server wants and shas the server has
 # and generates a list of commit shas we need to push up
 def generate_pack_contents(self, want, have):
 graph_walker = SimpleFetchGraphWalker(want, self.git.get_parents)
 next = graph_walker.next()
 if next in have:
 graph_walker.ack(next)
 else:
 shas.append(next)
 next = graph_walker.next()
 # so now i have the shas, need to turn them into a list of
 # tuples (sha, path) for ALL the objects i'm sending
 # TODO : don't send blobs or trees they already have
 def get_objects(tree, path):
 changes = list()
 changes.append((tree, path))
 if isinstance (obj, Blob):
 changes.append((obj, path + name))
 elif isinstance(obj, Tree):
 changes.extend (get_objects (obj, path + name + '/'))
 return changes
 objects = []
 for commit_sha in shas:
 commit = self.git.commit(commit_sha)
 objects.append((commit, 'commit'))
 tree = self.git.get_object(commit.tree)
 objects.extend( get_objects(tree, '/') )
 return objects
 def fetch_pack(self, remote_name):
 git_url = self.remote_name_to_url(remote_name)
 client, path = self.get_transport_and_path(git_url)
 graphwalker = SimpleFetchGraphWalker(self.git.heads().values(), self.git.get_parents)
 f, commit = self.git.object_store.add_pack()
 def import_git_commit(self, commit):
 print "importing: " + commit.id
 # TODO : look for HG metadata in the message and use it
 # TODO : add extra Git data (committer info) as extras to changeset
 # TODO : (?) have to handle merge contexts at some point (two parent files, etc)
-# TODO : throw IOError for removed files
+# TODO : Do something less coarse-grained than try/except on the
+#        get_file call for removed files
 def getfilectx(repo, memctx, f):
-(e, sha, data) = self.git.get_file(commit, f)
+try:
+(e, sha, data) = self.git.get_file(commit, f)
+except TypeError:
+raise IOError()
 e = '' # TODO : make this a real mode
 return context.memfilectx(f, data, 'l' in e, 'x' in e, None)
 p1 = "0" * 40
 p2 = "0" * 40
 return SubprocessGitClient(), uri
 def clear(self):
 git_dir = self.repo.join('git')
 mapfile = self.repo.join('git-mapfile')
 if os.path.exists(git_dir):
 for root, dirs, files in os.walk(git_dir, topdown=False):
 for name in files:
 os.remove(os.path.join(root, name))
 for name in dirs:
 os.rmdir(os.path.join(root, name))
 os.rmdir(git_dir)
 if os.path.exists(mapfile):
 os.remove(mapfile)
 ''
 """
 Tarjan's algorithm and topological sorting implementation in Python
 by Paul Harrison

Mercurial > hg > hg-git

comparison git_handler.py @ 51:1421d04f1ad2