SDK-1559: Add missing filter methods for Doc scan

Author: MrBurtyyy

yoti_python_sdk/doc_scan/session/retrieve/id_document_resource_response.py

@@ -6,6 +6,9 @@
 )
 from yoti_python_sdk.doc_scan.session.retrieve.page_response import PageResponse
 from yoti_python_sdk.doc_scan.session.retrieve.resource_response import ResourceResponse
+from yoti_python_sdk.doc_scan.session.retrieve.task_response import (
+    TextExtractionTaskResponse,
+)
 
 
 class IdDocumentResourceResponse(ResourceResponse):
@@ -71,3 +74,16 @@ def document_fields(self):
         :rtype: DocumentFieldsResponse
         """
         return self.__document_fields
+
+    @property
+    def text_extraction_tasks(self):
+        """
+        Returns a list of text extraction tasks associated
+        with the id document
+
+        :return: list of text extraction tasks
+        :rtype: list[TextExtractionTaskResponse]
+        """
+        return [
+            task for task in self.tasks if isinstance(task, TextExtractionTaskResponse)
+        ]

yoti_python_sdk/doc_scan/session/retrieve/resource_container.py

@@ -6,8 +6,6 @@
 )
 from yoti_python_sdk.doc_scan.session.retrieve.liveness_resource_response import (
     LivenessResourceResponse,
-)
-from yoti_python_sdk.doc_scan.session.retrieve.liveness_resource_response import (
     ZoomLivenessResourceResponse,
 )
 
@@ -75,3 +73,17 @@ def liveness_capture(self):
         :rtype: list[LivenessResourceResponse]
         """
         return self.__liveness_capture
+
+    @property
+    def zoom_liveness_resources(self):
+        """
+        Returns a filtered list of zoom liveness capture resources
+
+        :return: list of zoom liveness captures
+        :rtype: list[ZoomLivenessResourceResponse]
+        """
+        return [
+            liveness
+            for liveness in self.__liveness_capture
+            if isinstance(liveness, ZoomLivenessResourceResponse)
+        ]

yoti_python_sdk/doc_scan/session/retrieve/task_response.py

@@ -156,4 +156,10 @@ class TextExtractionTaskResponse(TaskResponse):
     Represents a Text Extraction task response
     """
 
-    pass
+    @property
+    def generated_text_data_checks(self):
+        return [
+            check
+            for check in self.generated_checks
+            if isinstance(check, GeneratedTextDataCheckResponse)
+        ]

yoti_python_sdk/tests/doc_scan/session/retrieve/test_id_document_resource_response.py

@@ -68,6 +68,14 @@ def test_should_parse_tasks_with_type(self):
         assert isinstance(result.tasks[0], TextExtractionTaskResponse)
         assert isinstance(result.tasks[1], TaskResponse)
 
+    def test_should_filter_text_extraction_tasks(self):
+        data = {"tasks": self.SOME_TASKS}
+
+        result = IdDocumentResourceResponse(data)
+
+        assert len(result.tasks) == 2
+        assert len(result.text_extraction_tasks) == 1
+
 
 if __name__ == "__main__":
     unittest.main()

yoti_python_sdk/tests/doc_scan/session/retrieve/test_resource_container.py

@@ -34,6 +34,19 @@ def test_should_parse_with_none(self):
         assert len(result.id_documents) == 0
         assert len(result.liveness_capture) == 0
 
+    def test_should_filter_zoom_liveness_resources(self):
+        data = {
+            "liveness_capture": [
+                {"liveness_type": "ZOOM"},
+                {"liveness_type": "someUnknown"},
+            ]
+        }
+
+        result = ResourceContainer(data)
+
+        assert len(result.liveness_capture) == 2
+        assert len(result.zoom_liveness_resources) == 1
+
 
 if __name__ == "__main__":
     unittest.main()

yoti_python_sdk/tests/doc_scan/session/retrieve/test_task_response.py

@@ -9,7 +9,10 @@
 from yoti_python_sdk.doc_scan.session.retrieve.generated_check_response import (
     GeneratedTextDataCheckResponse,
 )
-from yoti_python_sdk.doc_scan.session.retrieve.task_response import TaskResponse
+from yoti_python_sdk.doc_scan.session.retrieve.task_response import (
+    TaskResponse,
+    TextExtractionTaskResponse,
+)
 
 
 class TaskResponseTest(unittest.TestCase):
@@ -74,6 +77,14 @@ def test_should_parse_with_none(self):
         assert len(result.generated_checks) == 0
         assert len(result.generated_media) == 0
 
+    def test_should_filter_generated_text_data_checks(self):
+        data = {"generated_checks": self.SOME_GENERATED_CHECKS}
+
+        result = TextExtractionTaskResponse(data)
+
+        assert len(result.generated_checks) == 2
+        assert len(result.generated_text_data_checks) == 1
+
 
 if __name__ == "__main__":
     unittest.main()