[gentoo-commits] proj/gentoopm:master commit in: gentoopm/paludispm/

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

commit:     87eed3cc74ea06f0b6cb4cf2ebb18b27ee508e6f
Author:     Michał Górny <mgorny <AT> gentoo <DOT> org>
AuthorDate: Fri Jul  8 21:00:18 2011 +0000
Commit:     Michał Górny <mgorny <AT> gentoo <DOT> org>
CommitDate: Fri Jul  8 21:00:18 2011 +0000
URL:        http://git.overlays.gentoo.org/gitweb/?p=proj/gentoopm.git;a=commit;h=87eed3cc

Paludis: support atom matching.

---
 gentoopm/paludispm/repo.py |   32 ++++++++++++++++++++++++++++++++
 1 files changed, 32 insertions(+), 0 deletions(-)

diff --git a/gentoopm/paludispm/repo.py b/gentoopm/paludispm/repo.py
index 01f57a8..caa29e8 100644
--- a/gentoopm/paludispm/repo.py
+++ b/gentoopm/paludispm/repo.py
@@ -40,6 +40,38 @@ class PaludisRepository(PMRepository, PaludisPackageSet):
 				paludis.FilteredGenerator(self._gen, self._filt))]):
 			yield PaludisID(p, i, enum)
 
+	def filter(self, *args, **kwargs):
+		pset = self
+		newargs = []
+
+		for f in args:
+			if not callable(f): # an atom!
+				pset = PaludisAtomFilteredRepo(pset, f)
+			else:
+				newargs.append(f)
+
+		if pset == self:
+			return PaludisPackageSet.filter(self, args, kwargs)
+		elif newargs or kwargs:
+			return pset.filter(self, newargs, kwargs)
+		else:
+			return pset
+
+class PaludisAtomFilteredRepo(PaludisRepository):
+	@property
+	def _gen(self):
+		return self._mygen
+
+	@property
+	def _filt(self):
+		return self._myfilt
+
+	def __init__(self, repo, atom):
+		PaludisRepository.__init__(self, repo._env)
+		self._myfilt = repo._filt
+		self._mygen = repo._gen & paludis.Generator.Matches(atom._atom,
+				paludis.MatchPackageOptions())
+
 class PaludisStackRepo(PaludisRepository):
 	@property
 	def _gen(self):