Add fecha_nacimiento_date computed field to Person model

Author: gmorales96

quienesquien/person.py

@@ -1,3 +1,5 @@
+import datetime as dt
+
 from pydantic import (
     BaseModel,
     ConfigDict,
@@ -27,12 +29,22 @@ class Person(BaseModel):
         extra='allow',
     )
 
-    @computed_field  # type: ignore[misc]
-    @property
+    @computed_field
     def peso1(self) -> str:
         # peso1 is required for backward compatibility with previous version.
         return str(self.coincidencia)
 
+    @computed_field
+    def fecha_nacimiento_date(self) -> dt.date | None:
+        if not self.fecha_nacimiento:
+            return None
+        try:
+            return dt.datetime.strptime(
+                self.fecha_nacimiento, '%d/%m/%Y'
+            ).date()
+        except (TypeError, ValueError):
+            return None
+
     @model_validator(mode='after')
     def collect_extra_fields(self):
         if self.model_extra:

tests/test_person.py

@@ -1,8 +1,13 @@
+import datetime as dt
+from typing import Any
+
+import pytest
+
 from quienesquien.person import Person
 
 
-def test_collect_extra_fields():
-    person_data = {
+def test_collect_extra_fields() -> None:
+    person_data: dict[str, Any] = {
         'LISTA': 'lista1',
         'COINCIDENCIA': 100,
         'NOMBRECOMP': 'Juan Pérez',
@@ -14,5 +19,37 @@ def test_collect_extra_fields():
     assert person.peso1 == '100'
     assert person.coincidencia == 100
     assert person.nombrecomp == 'Juan Pérez'
-    assert person.campo_extra1 == 'valor1'
-    assert person.campo_extra2 == 'valor2'
+
+    # Access extra fields through model_extra after lowercase conversion
+    assert person.model_extra is not None
+    assert person.model_extra['campo_extra1'] == 'valor1'
+    assert person.model_extra['campo_extra2'] == 'valor2'
+
+
+@pytest.mark.parametrize(
+    "fecha_nacimiento_input, expected_fecha_nacimiento, expected_date",
+    [
+        ('13/11/1953', '13/11/1953', dt.date(1953, 11, 13)),
+        (None, None, None),
+        ('', '', None),
+        ('invalid-date', 'invalid-date', None),
+        ('1953-11-13', '1953-11-13', None),
+        ('13/11', '13/11', None),
+        ('13-11-1953', '13-11-1953', None),
+    ],
+)
+def test_fecha_nacimiento_date(
+    fecha_nacimiento_input: str | None,
+    expected_fecha_nacimiento: str | None,
+    expected_date: dt.date | None,
+) -> None:
+    person_data: dict[str, Any] = {
+        'LISTA': 'PPE',
+        'COINCIDENCIA': 100,
+        'NOMBRECOMP': 'Andres Manuel López Obrador',
+        'FECHA_NACIMIENTO': fecha_nacimiento_input,
+    }
+    person = Person(**person_data)
+
+    assert person.fecha_nacimiento == expected_fecha_nacimiento
+    assert person.fecha_nacimiento_date == expected_date