add expiration date field

Author: rcabrera-py

mati/types/enums.py

@@ -184,6 +184,40 @@ def ocr_number(self) -> str:
             return self.fields['ocr_number']['value']
         return ''
 
+    @property
+    def expiration_date(self) -> None:
+        """
+        This property fills the expiration date direct from the ocr
+        fields `expiration_date`.
+
+        Returns None if the field is missing or fields is None.
+        Unlike other fields that return empty strings, dates return None
+        to indicate absence of a date value.
+        """
+        if self.fields is None:
+            return None
+        try:
+            return self.fields['expiration_date']['value']
+        except (KeyError, TypeError):
+            return None
+
+    @property
+    def emission_date(self) -> None:
+        """
+        This property fills the emission date direct from the ocr
+        fields `emission_date`.
+
+        Returns None if the field is missing or fields is None.
+        Unlike other fields that return empty strings, dates return None
+        to indicate absence of a date value.
+        """
+        if self.fields is None:
+            return None
+        try:
+            return self.fields['emission_date']['value']
+        except (KeyError, TypeError):
+            return None
+
     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.dev3'

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,63 @@ def test_excess_fields():
     data = {'some': 'data', 'aditional': 'data', 'id': 'foo', 'status': 10}
     VerificationDocumentStep._filter_excess_fields(data)
     assert 'some' not in data
+
+
+@pytest.mark.parametrize(
+    ('fields', 'expected'),
+    [
+        (
+            {
+                'expiration_date': {
+                    'value': '2030-12-31',
+                    'label': 'Date of Expiration',
+                    'format': 'date',
+                }
+            },
+            '2030-12-31',
+        ),
+        (None, None),
+        ({'address': {'value': 'Test Address', 'label': 'Address'}}, None),
+        ({'expiration_date': {'label': 'Date of Expiration'}}, None),
+    ],
+)
+def test_expiration_date_property(fields, expected):
+    doc = VerificationDocument(
+        country='MX',
+        region='',
+        photos=[],
+        steps=[],
+        type='national-id',
+        fields=fields,
+    )
+    assert doc.expiration_date == expected
+
+
+@pytest.mark.parametrize(
+    ('fields', 'expected'),
+    [
+        (
+            {
+                'emission_date': {
+                    'value': '2023-01-15',
+                    'label': 'Emission Date',
+                    'format': 'date',
+                }
+            },
+            '2023-01-15',
+        ),
+        (None, None),
+        ({'address': {'value': 'Test Address', 'label': 'Address'}}, None),
+        ({'emission_date': {'label': 'Emission Date'}}, None),
+    ],
+)
+def test_emission_date_property(fields, expected):
+    doc = VerificationDocument(
+        country='MX',
+        region='',
+        photos=[],
+        steps=[],
+        type='proof-of-residency',
+        fields=fields,
+    )
+    assert doc.emission_date == expected