Allow packages specified on the command line to be retroactively upgraded from "nomerge" to "merge" nodes.

svn path=/main/trunk/; revision=4494

1 files changed, 26 insertions, 0 deletions

diff --git a/pym/portage.py b/pym/portage.py
index 104f1a279..a97c50171 100644
--- a/pym/portage.py
+++ b/pym/portage.py
@@ -364,6 +364,32 @@ class digraph:
 		del self.nodes[node]
 		self.order.remove(node)
 
+	def replace(self, oldnode, newnode):
+		"""Replace all references to oldnode with references to newnode.
+		This is useful for upgrading a "nomerge" node to a "merge" node.
+
+		@return:
+			1. None on success.
+			2. Raise a KeyError if oldnode does not exist.
+			3. Raise a ValueError if newnode already exists.
+		"""
+		if not self.contains(oldnode):
+			raise KeyError(oldnode)
+		if oldnode == newnode:
+			return
+		if newnode in self.nodes:
+			raise ValueError(newnode)
+		for parent in self.nodes[oldnode][1]:
+			self.nodes[parent][0][newnode] = self.nodes[parent][0][oldnode]
+			del self.nodes[parent][0][oldnode]
+		for child in self.nodes[oldnode][0]:
+			self.nodes[child][1][newnode] = self.nodes[child][1][oldnode]
+			del self.nodes[child][1][oldnode]
+		self.nodes[newnode] = self.nodes[oldnode]
+		del self.nodes[oldnode]
+		oldindex = self.order.index(oldnode)
+		self.order[oldindex] = newnode
+
 	def contains(self, node):
 		"""Checks if the digraph contains mynode"""
 		return node in self.nodes