add expiration date field

Author: rcabrera-py

mati/types/enums.py

@@ -184,6 +184,26 @@ def ocr_number(self) -> str:
             return self.fields['ocr_number']['value']
         return ''
 
+    @property
+    def expiration_date(self) -> str:
+        """
+        This property fills the expiration date direct from the ocr
+        fields `expiration_date`
+        """
+        if self.fields and 'expiration_date' in self.fields:
+            return self.fields['expiration_date']['value']
+        return ''
+
+    @property
+    def emission_date(self) -> str:
+        """
+        This property fills the emission date direct from the ocr
+        fields `emission_date`
+        """
+        if self.fields and 'emission_date' in self.fields:
+            return self.fields['emission_date']['value']
+        return ''
+
     def add_expired_step(self) -> None:
         '''
         Appends an expired error step to the document if missing.

mati/version.py

@@ -1 +1 @@
-__version__ = '2.0.8'
+__version__ = '2.0.9.dev2'

tests/test_types.py

@@ -2,7 +2,7 @@
 from pytest_lazyfixture import lazy_fixture
 
 from mati.types import ValidationInputType
-from mati.types.enums import VerificationDocumentStep
+from mati.types.enums import VerificationDocument, VerificationDocumentStep
 
 
 def test_type_to_str():
@@ -44,3 +44,57 @@ def test_excess_fields():
     data = {'some': 'data', 'aditional': 'data', 'id': 'foo', 'status': 10}
     VerificationDocumentStep._filter_excess_fields(data)
     assert 'some' not in data
+
+
+def test_expiration_date_property():
+    doc = VerificationDocument(
+        country='MX',
+        region='',
+        photos=[],
+        steps=[],
+        type='national-id',
+        fields={
+            'expiration_date': {
+                'value': '2030-12-31',
+                'label': 'Date of Expiration',
+                'format': 'date',
+            }
+        },
+    )
+    assert doc.expiration_date == '2030-12-31'
+    doc_no_fields = VerificationDocument(
+        country='MX',
+        region='',
+        photos=[],
+        steps=[],
+        type='national-id',
+        fields=None,
+    )
+    assert doc_no_fields.expiration_date == ''
+
+
+def test_emission_date_property():
+    doc = VerificationDocument(
+        country='MX',
+        region='',
+        photos=[],
+        steps=[],
+        type='proof-of-residency',
+        fields={
+            'emission_date': {
+                'value': '2023-01-15',
+                'label': 'Emission Date',
+                'format': 'date',
+            }
+        },
+    )
+    assert doc.emission_date == '2023-01-15'
+    doc_no_fields = VerificationDocument(
+        country='MX',
+        region='',
+        photos=[],
+        steps=[],
+        type='proof-of-residency',
+        fields=None,
+    )
+    assert doc_no_fields.emission_date == ''