from rest_framework import serializers
from core.models import PromoCode

class PromoCodeSerializer(serializers.ModelSerializer):
    valid_from = serializers.SerializerMethodField()
    valid_to = serializers.SerializerMethodField()
    
    class Meta:
        model = PromoCode
        fields = ['id', 'code', 'discount_amount', 'discount_type', 'valid_from', 'valid_to', 'total_count', 
        'taken_count', 'code_status', 'created_at', 'edited_at']

    def get_valid_from(self, obj):
        return obj.valid_from.replace(tzinfo=None)

    def get_valid_to(self, obj):
        return obj.valid_to.replace(tzinfo=None)