Merge branch 'revparse'

Author: Byron

lib/git/db.py

@@ -1,11 +1,18 @@
-"""Module with our own gitdb implementation - it uses the git command""" 
+"""Module with our own gitdb implementation - it uses the git command"""
+from exc import (
+					GitCommandError, 
+					BadObject
+				)
+
 from gitdb.base import (
 								OInfo,
 								OStream
 							)
 
-from gitdb.util import bin_to_hex
-
+from gitdb.util import (
+							bin_to_hex, 
+							hex_to_bin
+						)
 from gitdb.db import GitDB
 from gitdb.db import LooseObjectDB
 
@@ -35,3 +42,20 @@ def stream(self, sha):
 		t = self._git.stream_object_data(bin_to_hex(sha))
 		return OStream(*t)
 
+	
+	# { Interface
+	
+	def partial_to_complete_sha_hex(self, partial_hexsha):
+		""":return: Full binary 20 byte sha from the given partial hexsha
+		:raise AmbiguousObjectName:
+		:raise BadObject:
+		:note: currently we only raise BadObject as git does not communicate 
+			AmbiguousObjects separately"""
+		try:
+			hexsha, typename, size = self._git.get_object_header(partial_hexsha)
+			return hex_to_bin(hexsha)
+		except (GitCommandError, ValueError):
+			raise BadObject(partial_hexsha)
+		# END handle exceptions
+	
+	#} END interface

lib/git/exc.py

@@ -1,10 +1,12 @@
-# errors.py
+# exc.py
 # Copyright (C) 2008, 2009 Michael Trier ([email protected]) and contributors
 #
 # This module is part of GitPython and is released under
 # the BSD License: http://www.opensource.org/licenses/bsd-license.php
 """ Module containing all exceptions thrown througout the git package, """
 
+from gitdb.exc import *
+
 class InvalidGitRepositoryError(Exception):
 	""" Thrown if the given repository appears to have an invalid format.  """
 

lib/git/ext/gitdb

@@ -1,4 +1,4 @@
-"""Initialize the index module"""
+"""Initialize the index package"""
 
 from base import *
 from typ import *

lib/git/index/__init__.py

@@ -1122,7 +1122,7 @@ def diff(self, other=diff.Diffable.Index, paths=None, create_patch=False, **kwar
 		# item. Handle existing -R flags properly. Transform strings to the object
 		# so that we can call diff on it
 		if isinstance(other, basestring):
-			other = Object.new(self.repo, other)
+			other = self.repo.rev_parse(other)
 		# END object conversion
 
 		if isinstance(other, Object):

lib/git/index/base.py

@@ -49,10 +49,18 @@ def new(cls, repo, id):
 			
 		:note: This cannot be a __new__ method as it would always call __init__
 			with the input id which is not necessarily a binsha."""
-		hexsha, typename, size = repo.git.get_object_header(id)
-		inst = get_object_type_by_name(typename)(repo, hex_to_bin(hexsha))
-		inst.size = size
-		return inst
+		return repo.rev_parse(str(id))
+		
+	@classmethod
+	def new_from_sha(cls, repo, sha1):
+		"""
+		:return: new object instance of a type appropriate to represent the given 
+			binary sha1
+		:param sha1: 20 byte binary sha1"""
+		oinfo = repo.odb.info(sha1)
+		inst = get_object_type_by_name(oinfo.type)(repo, oinfo.binsha)
+		inst.size = oinfo.size
+		return inst 
 
 	def _set_self_from_args_(self, args_dict):
 		"""Initialize attributes on self from the given dict that was retrieved

lib/git/objects/base.py

@@ -68,7 +68,7 @@ def name(self):
 		:return:
 			In case of symbolic references, the shortest assumable name 
 			is the path itself."""
-		return self.path	
+		return self.path
 
 	def _abs_path(self):
 		return join_path_native(self.repo.git_dir, self.path)
@@ -109,6 +109,19 @@ def _iter_packed_refs(cls, repo):
 		# I believe files are closing themselves on destruction, so it is 
 		# alright.
 
+	@classmethod
+	def dereference_recursive(cls, repo, ref_path):
+		"""
+		:return: hexsha stored in the reference at the given ref_path, recursively dereferencing all
+			intermediate references as required
+		:param repo: the repository containing the reference at ref_path"""
+		while True:
+			ref = cls(repo, ref_path)
+			hexsha, ref_path = ref._get_ref_info()
+			if hexsha is not None:
+				return hexsha
+		# END recursive dereferencing
+		
 	def _get_ref_info(self):
 		"""Return: (sha, target_ref_path) if available, the sha the file at 
 		rela_path points to, or None. target_ref_path is the reference we 
@@ -195,9 +208,8 @@ def _set_reference(self, ref):
 			try:
 				write_value = ref.commit.hexsha
 			except AttributeError:
-				sha = str(ref)
 				try:
-					obj = Object.new(self.repo, sha)
+					obj = self.repo.rev_parse(ref+"^{}")	# optionally deref tags
 					if obj.type != "commit":
 						raise TypeError("Invalid object type behind sha: %s" % sha)
 					write_value = obj.hexsha
@@ -333,7 +345,7 @@ def _create(cls, repo, path, resolve, reference, force):
 		# figure out target data
 		target = reference
 		if resolve:
-			target = Object.new(repo, reference)
+			target = repo.rev_parse(str(reference))
 
 		if not force and isfile(abs_ref_path):
 			target_data = str(target)
@@ -523,7 +535,7 @@ def _get_object(self):
 			always point to the actual object as it gets re-created on each query"""
 		# have to be dynamic here as we may be a tag which can point to anything
 		# Our path will be resolved to the hexsha which will be used accordingly
-		return Object.new(self.repo, self.path)
+		return Object.new_from_sha(self.repo, hex_to_bin(self.dereference_recursive(self.repo, self.path)))
 
 	def _set_object(self, ref):
 		"""

lib/git/refs.py

@@ -391,7 +391,7 @@ def _from_line(cls, repo, line, fetch_line):
 				split_token = '...'
 				if control_character == ' ':
 					split_token = split_token[:-1]
-				old_commit = Commit.new(repo, operation.split(split_token)[0])
+				old_commit = repo.rev_parse(operation.split(split_token)[0])
 			# END handle refspec
 		# END reference flag handling
 

lib/git/remote.py

@@ -0,0 +1,3 @@
+"""Initialize the Repo package"""
+
+from base import *

lib/git/repo/__init__.py

@@ -4,51 +4,39 @@
 # This module is part of GitPython and is released under
 # the BSD License: http://www.opensource.org/licenses/bsd-license.php
 
-from exc import InvalidGitRepositoryError, NoSuchPathError
-from cmd import Git
-from objects import Actor
-from refs import *
-from index import IndexFile
-from objects import *
-from config import GitConfigParser
-from remote import Remote
-
-from db import (
+from git.exc import InvalidGitRepositoryError, NoSuchPathError
+from git.cmd import Git
+from git.objects import Actor
+from git.refs import *
+from git.index import IndexFile
+from git.objects import *
+from git.config import GitConfigParser
+from git.remote import Remote
+from git.db import (
 				GitCmdObjectDB, 
 				GitDB
 				)
 
+
 from gitdb.util import (
 							join,
-							isdir, 
 							isfile,
-							join,
 							hex_to_bin
 						)
+
+from fun import (
+					rev_parse,
+					is_git_dir,
+					touch
+				)
+
 import os
 import sys
 import re
 
 
 __all__ = ('Repo', )
 
-def touch(filename):
-	fp = open(filename, "a")
-	fp.close()
-
-def is_git_dir(d):
-	""" This is taken from the git setup.c:is_git_directory
-	function."""
-
-	if isdir(d) and \
-			isdir(join(d, 'objects')) and \
-			isdir(join(d, 'refs')):
-		headref = join(d, 'HEAD')
-		return isfile(headref) or \
-				(os.path.islink(headref) and
-				os.readlink(headref).startswith('refs'))
-	return False
-
 
 class Repo(object):
 	"""Represents a git repository and allows you to query references, 
@@ -70,6 +58,7 @@ class Repo(object):
 	# precompiled regex
 	re_whitespace = re.compile(r'\s+')
 	re_hexsha_only = re.compile('^[0-9A-Fa-f]{40}$')
+	re_hexsha_shortened = re.compile('^[0-9A-Fa-f]{4,40}$')
 	re_author_committer_start = re.compile(r'^(author|committer)')
 	re_tab_full_line = re.compile(r'^\t(.*)$')
 
@@ -109,7 +98,7 @@ def __init__(self, path=None, odbt = GitDB):
 				self.git_dir = curpath
 				self._working_tree_dir = os.path.dirname(curpath)
 				break
-			gitpath = os.path.join(curpath, '.git')
+			gitpath = join(curpath, '.git')
 			if is_git_dir(gitpath):
 				self.git_dir = gitpath
 				self._working_tree_dir = curpath
@@ -139,7 +128,7 @@ def __init__(self, path=None, odbt = GitDB):
 		self.git = Git(self.working_dir)
 
 		# special handling, in special times
-		args = [os.path.join(self.git_dir, 'objects')]
+		args = [join(self.git_dir, 'objects')]
 		if issubclass(odbt, GitCmdObjectDB):
 			args.append(self.git)
 		self.odb = odbt(*args)
@@ -160,11 +149,11 @@ def __repr__(self):
 
 	# Description property
 	def _get_description(self):
-		filename = os.path.join(self.git_dir, 'description')
+		filename = join(self.git_dir, 'description')
 		return file(filename).read().rstrip()
 
 	def _set_description(self, descr):
-		filename = os.path.join(self.git_dir, 'description')
+		filename = join(self.git_dir, 'description')
 		file(filename, 'w').write(descr+'\n')
 
 	description = property(_get_description, _set_description,
@@ -334,11 +323,9 @@ def commit(self, rev=None):
 		:param rev: revision specifier, see git-rev-parse for viable options.
 		:return: ``git.Commit``"""
 		if rev is None:
-			rev = self.active_branch
-		
-		c = Object.new(self, rev)
-		assert c.type == "commit", "Revision %s did not point to a commit, but to %s" % (rev, c)
-		return c
+			return self.active_branch.commit
+		else:
+			return self.rev_parse(str(rev)+"^0")
 
 	def iter_trees(self, *args, **kwargs):
 		""":return: Iterator yielding Tree objects
@@ -359,14 +346,9 @@ def tree(self, rev=None):
 			it cannot know about its path relative to the repository root and subsequent 
 			operations might have unexpected results."""
 		if rev is None:
-			rev = self.active_branch
-		
-		c = Object.new(self, rev)
-		if c.type == "commit":
-			return c.tree
-		elif c.type == "tree":
-			return c
-		raise ValueError( "Revision %s did not point to a treeish, but to %s" % (rev, c))
+			return self.active_branch.commit.tree
+		else:
+			return self.rev_parse(str(rev)+"^{tree}")
 
 	def iter_commits(self, rev=None, paths='', **kwargs):
 		"""A list of Commit objects representing the history of a given ref/commit
@@ -393,11 +375,11 @@ def iter_commits(self, rev=None, paths='', **kwargs):
 		return Commit.iter_items(self, rev, paths, **kwargs)
 
 	def _get_daemon_export(self):
-		filename = os.path.join(self.git_dir, self.DAEMON_EXPORT_FILE)
+		filename = join(self.git_dir, self.DAEMON_EXPORT_FILE)
 		return os.path.exists(filename)
 
 	def _set_daemon_export(self, value):
-		filename = os.path.join(self.git_dir, self.DAEMON_EXPORT_FILE)
+		filename = join(self.git_dir, self.DAEMON_EXPORT_FILE)
 		fileexists = os.path.exists(filename)
 		if value and not fileexists:
 			touch(filename)
@@ -413,7 +395,7 @@ def _get_alternates(self):
 		"""The list of alternates for this repo from which objects can be retrieved
 
 		:return: list of strings being pathnames of alternates"""
-		alternates_path = os.path.join(self.git_dir, 'objects', 'info', 'alternates')
+		alternates_path = join(self.git_dir, 'objects', 'info', 'alternates')
 
 		if os.path.exists(alternates_path):
 			try:
@@ -436,9 +418,9 @@ def _set_alternates(self, alts):
 		:note:
 			The method does not check for the existance of the paths in alts
 			as the caller is responsible."""
-		alternates_path = os.path.join(self.git_dir, 'objects', 'info', 'alternates') 
+		alternates_path = join(self.git_dir, 'objects', 'info', 'alternates') 
 		if not alts:
-			if os.path.isfile(alternates_path):
+			if isfile(alternates_path):
 				os.remove(alternates_path)
 		else:
 			try:
@@ -466,7 +448,7 @@ def is_dirty(self, index=True, working_tree=True, untracked_files=False):
 		default_args = ('--abbrev=40', '--full-index', '--raw')
 		if index: 
 			# diff index against HEAD
-			if os.path.isfile(self.index.path) and self.head.is_valid() and \
+			if isfile(self.index.path) and self.head.is_valid() and \
 				len(self.git.diff('HEAD', '--cached', *default_args)):
 				return True
 		# END index handling
@@ -674,7 +656,7 @@ def clone(self, path, **kwargs):
 		# our git command could have a different working dir than our actual 
 		# environment, hence we prepend its working dir if required
 		if not os.path.isabs(path) and self.git.working_dir:
-			path = os.path.join(self.git._working_dir, path)
+			path = join(self.git._working_dir, path)
 		return Repo(os.path.abspath(path), odbt = odbt)
 
 
@@ -693,11 +675,13 @@ def archive(self, ostream, treeish=None, prefix=None,  **kwargs):
 		if treeish is None:
 			treeish = self.active_branch
 		if prefix and 'prefix' not in kwargs:
-			kwargs['prefix'] = prefix
+			kwargs['prefix'] = prefix 
 		kwargs['output_stream'] = ostream
 
 		self.git.archive(treeish, **kwargs)
 		return self
-
+	
+	rev_parse = rev_parse
+		
 	def __repr__(self):
 		return '<git.Repo "%s">' % self.git_dir