Add ppe_matches property to UserListsValidation class for filtering PPE-related persons

Author: gabino

cuenca/resources/user_lists_validation.py

@@ -19,6 +19,17 @@ class UserListsValidation(Creatable, Retrievable):
     status: Optional[VerificationStatus] = None
     response: Optional[dict] = None
 
+    @property
+    def ppe_matches(self) -> list[dict]:
+        if not self.response or 'persons' not in self.response:
+            return []
+
+        return [
+            person
+            for person in self.response['persons']
+            if person.get('lista') == 'PPE'
+        ]
+
     @classmethod
     def create(
         cls,

cuenca/version.py

@@ -1,3 +1,3 @@
-__version__ = '2.1.1'
+__version__ = '2.1.3.dev0'
 CLIENT_VERSION = __version__
 API_VERSION = '2020-03-19'

tests/resources/test_user_lists_validation.py

@@ -14,13 +14,15 @@ def test_create_user_validation_with_response():
         second_surname='Beltrán',
     )
     assert user_validation.status == VerificationStatus.rejected
-    assert user_validation.response['persons'][0]['parentesco'] == 'HIJO'
+    assert len(user_validation.ppe_matches) == 1
+    assert user_validation.ppe_matches[0]['parentesco'] == 'HIJO'
 
 
 @pytest.mark.vcr
 def test_create_user_validation(user_lists_request):
     user_validation = UserListsValidation.create(**user_lists_request)
     assert user_validation.status == VerificationStatus.succeeded
+    assert len(user_validation.ppe_matches) == 0
     assert user_validation.id