[gentoo-commits] proj/gentoopm:master commit in: gentoopm/portagepm/, gentoopm/basepm/

["Michał Górny" <mgorny@gentoo.org>]

commit:     a23f4f0a03bb3a09aa588c6f819bad94e1e27e67
Author:     Michał Górny <mgorny <AT> gentoo <DOT> org>
AuthorDate: Thu Jul  7 19:55:50 2011 +0000
Commit:     Michał Górny <mgorny <AT> gentoo <DOT> org>
CommitDate: Thu Jul  7 19:55:50 2011 +0000
URL:        http://git.overlays.gentoo.org/gitweb/?p=proj/gentoopm.git;a=commit;h=a23f4f0a

Support checking for matching atoms through 'atom in pkgset'.

---
 gentoopm/basepm/pkg.py    |   13 +++++++++++++
 gentoopm/portagepm/pkg.py |    2 +-
 2 files changed, 14 insertions(+), 1 deletions(-)

diff --git a/gentoopm/basepm/pkg.py b/gentoopm/basepm/pkg.py
index f7e534a..2e801e0 100644
--- a/gentoopm/basepm/pkg.py
+++ b/gentoopm/basepm/pkg.py
@@ -67,6 +67,19 @@ class PMPackageSet(ABCObject):
 		except KeyError:
 			raise ValueError('Ambiguous filter (matches more than a single package name).')
 
+	def __contains__(self, arg):
+		"""
+		Check whether the package set contains at least a single package
+		matching the filter or package atom passed as an argument.
+		"""
+
+		i = iter(self.filter(arg))
+		try:
+			next(i)
+		except StopIteration:
+			return False
+		return True
+
 class PMFilteredPackageSet(PMPackageSet):
 	def __init__(self, it, args, kwargs):
 		self._iter = it

diff --git a/gentoopm/portagepm/pkg.py b/gentoopm/portagepm/pkg.py
index ca2c054..d2ea1ad 100644
--- a/gentoopm/portagepm/pkg.py
+++ b/gentoopm/portagepm/pkg.py
@@ -52,7 +52,7 @@ class PortageCPV(PortageDBCPV):
 	def __cmp__(self, other):
 		if not isinstance(other, PortageCPV):
 			raise TypeError('Unable to compare %s against %s' % \
-					self, other)
+					(self, other))
 		return cmp(cpv_getkey(self._cpv), cpv_getkey(other._cpv)) \
 				or vercmp(cpv_getversion(self._cpv), cpv_getversion(other._cpv)) \
 				or cmp(self._repo_prio, other._repo_prio)