pkg.ebuild: expose PROPERTIES and RESTRICT depset attrs

Author: radhermit

src/pkgcraft/pkg/ebuild.pxd

@@ -13,7 +13,9 @@ cdef class EbuildPkg(Pkg):
     cdef object _pdepend
     cdef object _rdepend
     cdef object _license
+    cdef object _properties
     cdef object _required_use
+    cdef object _restrict
     cdef object _src_uri
     cdef tuple _homepage
     cdef tuple _keywords

src/pkgcraft/pkg/ebuild.pyx

@@ -16,8 +16,10 @@ cdef class EbuildPkg(Pkg):
         self._pdepend = SENTINEL
         self._rdepend = SENTINEL
         self._license = SENTINEL
-        self._src_uri = SENTINEL
+        self._properties = SENTINEL
         self._required_use = SENTINEL
+        self._restrict = SENTINEL
+        self._src_uri = SENTINEL
 
     @property
     def repo(self):
@@ -127,6 +129,15 @@ cdef class EbuildPkg(Pkg):
             self._license = None if deps is NULL else DepSet.from_ptr(deps)
         return self._license
 
+    @property
+    def properties(self):
+        """Get a package's PROPERTIES."""
+        cdef C.DepSet *deps
+        if self._properties is SENTINEL:
+            deps = C.pkgcraft_ebuild_pkg_properties(self._ebuild_pkg)
+            self._properties = None if deps is NULL else DepSet.from_ptr(deps)
+        return self._properties
+
     @property
     def required_use(self):
         """Get a package's REQUIRED_USE."""
@@ -136,6 +147,15 @@ cdef class EbuildPkg(Pkg):
             self._required_use = None if deps is NULL else DepSet.from_ptr(deps)
         return self._required_use
 
+    @property
+    def restrict(self):
+        """Get a package's RESTRICT."""
+        cdef C.DepSet *deps
+        if self._restrict is SENTINEL:
+            deps = C.pkgcraft_ebuild_pkg_restrict(self._ebuild_pkg)
+            self._restrict = None if deps is NULL else DepSet.from_ptr(deps)
+        return self._restrict
+
     @property
     def src_uri(self):
         """Get a package's SRC_URI."""

src/pkgcraft/pkgcraft_c.pxd

@@ -436,6 +436,14 @@ cdef extern from "pkgcraft.h":
     # The argument must be a non-null EbuildPkg pointer.
     DepSet *pkgcraft_ebuild_pkg_pdepend(EbuildPkg *p);
 
+    # Return a package's PROPERTIES.
+    #
+    # Returns NULL on nonexistence.
+    #
+    # # Safety
+    # The argument must be a non-null EbuildPkg pointer.
+    DepSet *pkgcraft_ebuild_pkg_properties(EbuildPkg *p);
+
     # Return a package's RDEPEND.
     #
     # Returns NULL on nonexistence.
@@ -452,6 +460,14 @@ cdef extern from "pkgcraft.h":
     # The argument must be a non-null EbuildPkg pointer.
     DepSet *pkgcraft_ebuild_pkg_required_use(EbuildPkg *p);
 
+    # Return a package's RESTRICT.
+    #
+    # Returns NULL on nonexistence.
+    #
+    # # Safety
+    # The argument must be a non-null EbuildPkg pointer.
+    DepSet *pkgcraft_ebuild_pkg_restrict(EbuildPkg *p);
+
     # Return a package's slot.
     #
     # # Safety

tests/pkg/test_ebuild.py

@@ -98,12 +98,24 @@ def test_license(self, repo):
         pkg = repo.create_pkg(license='BSD')
         assert str(pkg.license) == 'BSD'
 
+    def test_properties(self, repo):
+        pkg = repo.create_pkg()
+        assert not pkg.properties
+        pkg = repo.create_pkg(properties='live')
+        assert str(pkg.properties) == 'live'
+
     def test_required_use(self, repo):
         pkg = repo.create_pkg()
         assert not pkg.required_use
         pkg = repo.create_pkg(required_use='use')
         assert str(pkg.required_use) == 'use'
 
+    def test_restrict(self, repo):
+        pkg = repo.create_pkg()
+        assert not pkg.restrict
+        pkg = repo.create_pkg(restrict='fetch')
+        assert str(pkg.restrict) == 'fetch'
+
     def test_src_uri(self, repo):
         pkg = repo.create_pkg()
         assert not pkg.src_uri