from rest_framework import serializers
from payment.models import Transaction

class TransactionSerializer(serializers.ModelSerializer):
    name = serializers.SerializerMethodField()
    plan_name = serializers.SerializerMethodField()
    # number_of_records = serializers.SerializerMethodField()

    class Meta:
        model = Transaction
        fields = ['name', 'purpose', 'plan_name', 'record_count', 'amount', 'currency', 'transaction_date', 'transaction_id']

    def get_name(self, obj):
        return f"{obj.customer.first_name} {obj.customer.last_name}"

    def get_plan_name(self, obj):
        if obj.purpose.lower() in ['plan renew', 'plan purchase'] and obj.product:
            return obj.product.plan_name
        return None

    def get_number_of_records(self, obj):
        if obj.purpose.lower() == 'validation service':
            return obj.record_count 
        return None