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

[Michał Górny]

commit:     bfdcf993f4fb3cbaf8acd4a779b0404ca96cadc7
Author:     Michał Górny <mgorny <AT> gentoo <DOT> org>
AuthorDate: Mon Jul 25 19:07:30 2011 +0000
Commit:     Michał Górny <mgorny <AT> gentoo <DOT> org>
CommitDate: Mon Jul 25 19:07:30 2011 +0000
URL:        http://git.overlays.gentoo.org/gitweb/?p=proj/gentoopm.git;a=commit;h=bfdcf993

pkgcore: support getting deps.

---
 gentoopm/pkgcorepm/depend.py |   39 +++++++++++++++++++++++++++++++++++++++
 gentoopm/pkgcorepm/pkg.py    |   16 ++++++++++++++++
 2 files changed, 55 insertions(+), 0 deletions(-)

diff --git a/gentoopm/pkgcorepm/depend.py b/gentoopm/pkgcorepm/depend.py
new file mode 100644
index 0000000..80d697b
--- /dev/null
+++ b/gentoopm/pkgcorepm/depend.py
@@ -0,0 +1,39 @@
+#!/usr/bin/python
+#	vim:fileencoding=utf-8
+# (c) 2011 Michał Górny <mgorny@g.o>
+# Released under the terms of the 2-clause BSD license.
+
+from pkgcore.ebuild.atom import atom
+from pkgcore.restrictions.boolean import OrRestriction
+
+from gentoopm.basepm.depend import PMPackageDepSet, PMPackageFinalDepSet
+from gentoopm.pkgcorepm.atom import PkgCoreAtom
+
+class PkgCorePackageDepSet(PMPackageDepSet):
+	def __init__(self, deps, pkg):
+		self._deps = deps
+		self._pkg = pkg
+
+	@property
+	def evaluated(self):
+		return PkgCorePackageFinalDepSet(
+				self._deps.evaluate_depset(self._pkg.use))
+
+class PkgCorePackageFinalDepSet(PMPackageFinalDepSet):
+	def __init__(self, deps):
+		self._deps = deps
+
+	def _iter_deps(self, deps):
+		for d in deps:
+			if isinstance(d, atom):
+				yield d
+			elif isinstance(d, OrRestriction):
+				for d in self._iter_deps((next(iter(d)),)): # XXX
+					yield d
+			else:
+				raise NotImplementedError('Parsing %s not implemented' \
+						% repr(d))
+
+	def __iter__(self):
+		for d in self._iter_deps(self._deps):
+			yield PkgCoreAtom(d)

diff --git a/gentoopm/pkgcorepm/pkg.py b/gentoopm/pkgcorepm/pkg.py
index dff02ed..d50d0fe 100644
--- a/gentoopm/pkgcorepm/pkg.py
+++ b/gentoopm/pkgcorepm/pkg.py
@@ -10,6 +10,7 @@ from gentoopm.basepm.pkg import PMPackage, PMPackageDescription, \
 from gentoopm.basepm.pkgset import PMPackageSet, PMFilteredPackageSet
 from gentoopm.pkgcorepm.atom import PkgCoreAtom, PkgCorePackageKey
 from gentoopm.pkgcorepm.contents import PkgCorePackageContents
+from gentoopm.pkgcorepm.depend import PkgCorePackageDepSet
 from gentoopm.util import SpaceSepTuple
 
 class PkgCorePackageSet(PMPackageSet):
@@ -85,6 +86,21 @@ class PkgCorePackage(PMPackage, PkgCoreAtom):
 	def repository(self):
 		return self._pkg.repo.repo_id
 
+	@property
+	def build_dependencies(self):
+		return PkgCorePackageDepSet(self._pkg.depends,
+				self._pkg)
+
+	@property
+	def run_dependencies(self):
+		return PkgCorePackageDepSet(self._pkg.rdepends,
+				self._pkg)
+
+	@property
+	def post_dependencies(self):
+		return PkgCorePackageDepSet(self._pkg.pdepends,
+				self._pkg)
+
 	def __str__(self):
 		if self._repo_index != 0:
 			s = '%s::%s' % (self._pkg.cpvstr, self._pkg.repo.repo_id)