put Tree and Blob onto a new base class suitable to deal with IndexObjects

Author: Byron

lib/git/base.py

@@ -40,7 +40,7 @@ class Object(LazyMixin):
 	Implements an Object which may be Blobs, Trees, Commits and Tags
 	"""
 	TYPES = ("blob", "tree", "commit", "tag")
-	__slots__ = ("repo", "id", "size")
+	__slots__ = ("repo", "id", "size", "_data_cached" )
 	type = None			# to be set by subclass
 
 	def __init__(self, repo, id, size=None):
@@ -61,6 +61,7 @@ def __init__(self, repo, id, size=None):
 		self.repo = repo
 		self.id = id
 		self.size = size
+		self._data_cached = type(None)
 
 	def __bake__(self):
 		"""
@@ -103,6 +104,20 @@ def __repr__(self):
 		"""
 		return '<git.%s "%s">' % (self.__class__.__name__, self.id)
 
+	@property
+	def data(self):
+		"""
+		The binary contents of this object.
+
+		Returns
+			str
+			
+		NOTE
+			The data will be cached after the first access.
+		"""
+		self._data_cached = ( self._data_cached is not type(None) and self._data_cached ) or self.repo.git.cat_file(self.id, p=True, with_raw_output=True)
+		return self._data_cached
+	
 	@classmethod
 	def get_type_by_name(cls, object_type_name):
 		"""
@@ -132,6 +147,45 @@ def get_type_by_name(cls, object_type_name):
 			raise ValueError("Cannot handle unknown object type: %s" % object_type_name)
 
 
+class IndexObject(Object):
+	"""
+	Base for all objects that can be part of the index file , namely Tree, Blob and
+	SubModule objects
+	"""
+	__slots__ = ("path", "mode") 
+	
+	def __init__(self, repo, id, mode=None, path=None, size = None):
+		"""
+		Initialize a newly instanced IndexObject
+		``repo``
+			is the Repo we are located in
+
+		``id`` : string
+			is the git object id as hex sha
+
+		``mode`` : int
+			is the file mode as int, use the stat module to evaluate the infomration
+
+		``path`` : str
+			is the path to the file in the file system, relative to the git repository root, i.e.
+			file.ext or folder/other.ext
+			
+		 ``size`` : int
+		 	size of the object data in bytes
+		"""
+		super(IndexObject, self).__init__(repo, id, size)
+		self.mode = mode
+		self.path = path
+		
+	@property
+	def basename(self):
+	  """
+	  Returns
+		  The basename of the IndexObject's file path
+	  """
+	  return os.path.basename(self.path)
+	  
+		
 class Ref(object):
 	"""
 	Represents a named reference to any object

lib/git/blob.py

@@ -12,56 +12,19 @@
 from commit import Commit
 import base
 
-class Blob(base.Object):
+class Blob(base.IndexObject):
 	"""A Blob encapsulates a git blob object"""
 	DEFAULT_MIME_TYPE = "text/plain"
 	type = "blob"
-	__slots__ = ("mode", "path", "_data_stored")
+
+	__slots__ = tuple()
 
 	# precompiled regex
 	re_whitespace = re.compile(r'\s+')
 	re_hexsha_only = re.compile('^[0-9A-Fa-f]{40}$')
 	re_author_committer_start = re.compile(r'^(author|committer)')
 	re_tab_full_line = re.compile(r'^\t(.*)$')
 
-	def __init__(self, repo, id, mode=None, path=None):
-		"""
-		Create an unbaked Blob containing just the specified attributes
-
-		``repo``
-			is the Repo
-
-		``id``
-			is the git object id
-
-		``mode``
-			is the file mode
-
-		``path``
-			is the path to the file
-
-		Returns
-			git.Blob
-		"""
-		super(Blob,self).__init__(repo, id)
-		self.mode = mode
-		self.path = path
-		self._data_stored  = type(None)	# serves as marker to prevent baking in this case
-
-	@property
-	def data(self):
-		"""
-		The binary contents of this blob.
-
-		Returns
-			str
-			
-		NOTE
-			The data will be cached after the first access.
-		"""
-		self._data_stored = ( self._data_stored is not type(None) and self._data_stored ) or self.repo.git.cat_file(self.id, p=True, with_raw_output=True)
-		return self._data_stored
-
 	@property
 	def mime_type(self):
 		"""
@@ -78,14 +41,6 @@ def mime_type(self):
 			guesses = mimetypes.guess_type(self.path)
 		return guesses and guesses[0] or self.DEFAULT_MIME_TYPE
 
-	@property
-	def basename(self):
-	  """
-	  Returns
-		  The basename of the Blobs file path
-	  """
-	  return os.path.basename(self.path)
-
 	@classmethod
 	def blame(cls, repo, commit, file):
 		"""

lib/git/tree.py

@@ -8,14 +8,13 @@
 import blob
 import base
 
-class Tree(base.Object):
+class Tree(base.IndexObject):
 
 	type = "tree"
+	__slots__ = "_contents"
 
-	def __init__(self, repo, id, mode=None, path=None):
-		super(Tree, self).__init__(repo, id)
-		self.mode = mode
-		self.path = path
+	def __init__(self, repo, id, mode=None, path=None, size=None):
+		super(Tree, self).__init__(repo, id, mode, path, size)
 		self._contents = None
 
 	def __bake__(self):
@@ -71,9 +70,6 @@ def __div__(self, file):
 		"""
 		return self.get(file)
 
-	@property
-	def basename(self):
-		os.path.basename(self.path)
 
 	def __repr__(self):
 		return '<git.Tree "%s">' % self.id