From 94e3f8fe18ff3bd86b4ef4cbb474e4c52994edc7 Mon Sep 17 00:00:00 2001
From: Zac Medico <zmedico@gentoo.org>
Date: Wed, 23 May 2007 00:05:30 +0000
Subject: Update /Packages as each package is built (MD5 is done here since it
 won't hurt performance).

svn path=/main/trunk/; revision=6586
---
 pym/emerge/__init__.py       |  3 +--
 pym/portage/dbapi/bintree.py | 50 ++++++++++++++++++++++++++++++++++++++++++--
 2 files changed, 49 insertions(+), 4 deletions(-)

(limited to 'pym')

diff --git a/pym/emerge/__init__.py b/pym/emerge/__init__.py
index a18fbf833..8a8bbfbe8 100644
--- a/pym/emerge/__init__.py
+++ b/pym/emerge/__init__.py
@@ -3163,8 +3163,7 @@ class MergeTask(object):
 						if retval != os.EX_OK:
 							return retval
 						bintree = self.trees[myroot]["bintree"]
-						if bintree.populated:
-							bintree.inject(pkg_key)
+						bintree.inject(pkg_key)
 						if "--buildpkgonly" not in self.myopts:
 							msg = " === (%s of %s) Merging (%s::%s)" % \
 								(mergecount, len(mymergelist), pkg_key, y)
diff --git a/pym/portage/dbapi/bintree.py b/pym/portage/dbapi/bintree.py
index 6e0800b08..27331ba37 100644
--- a/pym/portage/dbapi/bintree.py
+++ b/pym/portage/dbapi/bintree.py
@@ -480,7 +480,11 @@ class binarytree(object):
 						self.dbapi._aux_cache[mycpv] = aux_cache
 
 			self._pkg_paths = pkg_paths
-			if update_pkgindex and os.access(self.pkgdir, os.W_OK):
+			# Do not bother to write the Packages index if $PKGDIR/All/ exists
+			# since it will provide no benefit due to the need to read CATEGORY
+			# from xpak.
+			if update_pkgindex and not self._all_directory and \
+				os.access(self.pkgdir, os.W_OK):
 				cpv_all = self._pkg_paths.keys()
 				stale = set(metadata).difference(cpv_all)
 				for cpv in stale:
@@ -543,7 +547,49 @@ class binarytree(object):
 		self.populated=1
 
 	def inject(self, cpv):
-		return self.dbapi.cpv_inject(cpv)
+		"""Add a freshly built package to the database.  This updates
+		$PKGDIR/Packages with the new package metadata (including MD5)."""
+		if not self.populated and self._all_directory:
+			# There's nothing to update in this case, since the Packages
+			# index is not created when $PKGDIR/All/ exists.
+			return
+		if not self.populated:
+			self.populate()
+		full_path = self.getname(cpv)
+		try:
+			s = os.stat(full_path)
+		except OSError, e:
+			if e.errno != errno.ENOENT:
+				raise
+			del e
+			writemsg("!!! Binary package does not exist: '%s'\n" % full_path,
+				noiselevel=-1)
+			return
+		mytbz2 = portage.xpak.tbz2(full_path)
+		slot = mytbz2.getfile("SLOT")
+		if slot is None:
+			writemsg("!!! Invalid binary package: '%s'\n" % full_path,
+				noiselevel=-1)
+			return
+		slot = slot.strip()
+		from portage.checksum import perform_md5
+		md5 = perform_md5(full_path)
+		self.dbapi.cpv_inject(cpv)
+		self.dbapi._aux_cache.pop(cpv, None)
+		self._pkgindex.packages.pop(cpv, None)
+		d = {}
+		d["CPV"] = cpv
+		d["SLOT"] = slot
+		d["MTIME"] = str(long(s.st_mtime))
+		d["SIZE"] = str(s.st_size)
+		d["MD5"] = str(md5)
+		self._pkgindex.packages[cpv] = d
+		from portage.util import atomic_ofstream
+		f = atomic_ofstream(os.path.join(self.pkgdir, "Packages"))
+		try:
+			self._pkgindex.write(f)
+		finally:
+			f.close()
 
 	def exists_specific(self, cpv):
 		if not self.populated:
-- 
cgit v1.2.3-1-g7c22