import datetime

from rest_framework import serializers

from persona.models import Persona


class PersonaSerializer(serializers.ModelSerializer):
    class Meta:
        model = Persona
        fields = ('nombre', 'apellido', 'documento_identidad', 'fecha_nacimiento', 'organismo', 'edad')

    edad = serializers.SerializerMethodField(method_name='calcular_edad')

    def calcular_edad(self, persona):
        today = datetime.date.today()
        diferencia = today - persona.fecha_nacimiento
        edad = diferencia.days
        edad = edad / 365
        return int(edad)