From 479c3a35af4aabf0b3d35a156295cd29e47224b9 Mon Sep 17 00:00:00 2001
From: Zac Medico <zmedico@gentoo.org>
Date: Mon, 26 May 2008 08:02:40 +0000
Subject: [PATCH] * Enable config.setcpv() to use a Package instance in place
 of a cpv. * Make depgraph._select_package() pass Package instances into
 setcpv() calls. * Enable the Package constructor to use an iterable for
 metadata items.

svn path=/main/trunk/; revision=10427
---
 pym/_emerge/__init__.py | 50 +++++++++++++++++++++--------------------
 pym/portage/__init__.py | 12 +++++++++-
 2 files changed, 37 insertions(+), 25 deletions(-)

diff --git a/pym/_emerge/__init__.py b/pym/_emerge/__init__.py
index 173eba75f..e60dd212d 100644
--- a/pym/_emerge/__init__.py
+++ b/pym/_emerge/__init__.py
@@ -1296,7 +1296,12 @@ class Package(Task):
 		def __init__(self, pkg, metadata):
 			dict.__init__(self)
 			self._pkg = pkg
-			for k, v in metadata.iteritems():
+			i = getattr(metadata, "iteritems", None)
+			if i is None:
+				i = metadata
+			else:
+				i = i()
+			for k, v in i:
 				self[k] = v
 
 		def __setitem__(self, k, v):
@@ -3042,21 +3047,22 @@ class depgraph(object):
 					if pkg is None:
 						calculated_use = False
 						try:
-							metadata = dict(izip(self._mydbapi_keys,
-								db.aux_get(cpv, self._mydbapi_keys)))
+							metadata = zip(self._mydbapi_keys,
+								db.aux_get(cpv, self._mydbapi_keys))
 						except KeyError:
 							continue
+						pkg = Package(built=built, cpv=cpv,
+							installed=installed, metadata=metadata,
+							onlydeps=onlydeps, root=root, type_name=pkg_type)
+						metadata = pkg.metadata
 						if not built and ("?" in metadata["LICENSE"] or \
 							"?" in metadata["PROVIDE"]):
 							# This is avoided whenever possible because
 							# it's expensive. It only needs to be done here
 							# if it has an effect on visibility.
-							pkgsettings.setcpv(cpv, mydb=metadata)
+							pkgsettings.setcpv(pkg)
 							metadata["USE"] = pkgsettings["PORTAGE_USE"]
 							calculated_use = True
-						pkg = Package(built=built, cpv=cpv,
-							installed=installed, metadata=metadata,
-							onlydeps=onlydeps, root=root, type_name=pkg_type)
 						self._pkg_cache[pkg] = pkg
 
 					if not installed or (installed and matched_packages):
@@ -3095,7 +3101,7 @@ class depgraph(object):
 					if not pkg.built and not calculated_use:
 						# This is avoided whenever possible because
 						# it's expensive.
-						pkgsettings.setcpv(cpv, mydb=pkg.metadata)
+						pkgsettings.setcpv(pkg)
 						pkg.metadata["USE"] = pkgsettings["PORTAGE_USE"]
 
 					myarg = None
@@ -3153,14 +3159,11 @@ class depgraph(object):
 						"--reinstall" in self.myopts):
 						iuses = set(filter_iuse_defaults(
 							pkg.metadata["IUSE"].split()))
-						old_use = pkg.metadata["USE"].split()
-						mydb = pkg.metadata
-						if myeb and not usepkgonly:
-							mydb = portdb
+						old_use = pkg.use.enabled
 						if myeb:
-							pkgsettings.setcpv(myeb, mydb=mydb)
+							pkgsettings.setcpv(myeb)
 						else:
-							pkgsettings.setcpv(cpv, mydb=mydb)
+							pkgsettings.setcpv(pkg)
 						now_use = pkgsettings["PORTAGE_USE"].split()
 						forced_flags = set()
 						forced_flags.update(pkgsettings.useforce)
@@ -3168,8 +3171,7 @@ class depgraph(object):
 						cur_iuse = iuses
 						if myeb and not usepkgonly:
 							cur_iuse = set(filter_iuse_defaults(
-								portdb.aux_get(myeb,
-								["IUSE"])[0].split()))
+								myeb.metadata["IUSE"].split()))
 						if self._reinstall_for_flags(forced_flags,
 							old_use, iuses,
 							now_use, cur_iuse):
@@ -3180,7 +3182,7 @@ class depgraph(object):
 						("--newuse" in self.myopts or \
 						"--reinstall" in self.myopts) and \
 						cpv in vardb.match(atom):
-						pkgsettings.setcpv(cpv, mydb=pkg.metadata)
+						pkgsettings.setcpv(pkg)
 						forced_flags = set()
 						forced_flags.update(pkgsettings.useforce)
 						forced_flags.update(pkgsettings.usemask)
@@ -3197,7 +3199,7 @@ class depgraph(object):
 						if reinstall_for_flags:
 							reinstall = True
 					if not built:
-						myeb = cpv
+						myeb = pkg
 					matched_packages.append(pkg)
 					if reinstall_for_flags:
 						self._reinstall_nodes[pkg] = \
@@ -5185,23 +5187,23 @@ class depgraph(object):
 				continue
 			mydb = trees[myroot][self.pkg_tree_map[pkg_type]].dbapi
 			try:
-				metadata = dict(izip(self._mydbapi_keys,
-					mydb.aux_get(pkg_key, self._mydbapi_keys)))
+				metadata = zip(self._mydbapi_keys,
+					mydb.aux_get(pkg_key, self._mydbapi_keys))
 			except KeyError:
 				# It does no exist or it is corrupt.
 				if action == "uninstall":
 					continue
 				raise portage.exception.PackageNotFound(pkg_key)
-			if pkg_type == "ebuild":
-				pkgsettings = self.pkgsettings[myroot]
-				pkgsettings.setcpv(pkg_key, mydb=metadata)
-				metadata["USE"] = pkgsettings["PORTAGE_USE"]
 			installed = action == "uninstall"
 			built = pkg_type != "ebuild"
 			pkg = Package(built=built, cpv=pkg_key,
 				installed=installed, metadata=metadata,
 				operation=action, root=myroot,
 				type_name=pkg_type)
+			if pkg_type == "ebuild":
+				pkgsettings = self.pkgsettings[myroot]
+				pkgsettings.setcpv(pkg)
+				pkg.metadata["USE"] = pkgsettings["PORTAGE_USE"]
 			self._pkg_cache[pkg] = pkg
 
 			root_config = self.roots[pkg.root]
diff --git a/pym/portage/__init__.py b/pym/portage/__init__.py
index d2f4cc98f..f8dbc2d22 100644
--- a/pym/portage/__init__.py
+++ b/pym/portage/__init__.py
@@ -1909,6 +1909,13 @@ class config(object):
 		"""
 
 		self.modifying()
+
+		pkg = None
+		if not isinstance(mycpv, basestring):
+			pkg = mycpv
+			mycpv = pkg.cpv
+			mydb = pkg.metadata
+
 		if self.mycpv == mycpv:
 			return
 		ebuild_phase = self.get("EBUILD_PHASE")
@@ -1924,7 +1931,10 @@ class config(object):
 				iuse = mydb["IUSE"]
 			else:
 				slot, iuse = mydb.aux_get(self.mycpv, ["SLOT", "IUSE"])
-			cpv_slot = "%s:%s" % (self.mycpv, slot)
+			if pkg is None:
+				cpv_slot = "%s:%s" % (self.mycpv, slot)
+			else:
+				cpv_slot = pkg
 			pkginternaluse = []
 			for x in iuse.split():
 				if x.startswith("+"):
-- 
2.26.2