from rest_framework import serializers
from .models import BlacklistRecord, ScrapeLog
import json

class BlacklistRecordSerializer(serializers.ModelSerializer):
    associated = serializers.SerializerMethodField()

    class Meta:
        model = BlacklistRecord
        fields = ['sno', 'blacklist_no', 'blacklist_date_bs', 'blacklist_date_ad', 'borrower_name', 'associated']

    def get_associated(self, obj):
        try:
            return json.loads(obj.associated_entities)
        except:
            return []

class ScrapeLogSerializer(serializers.ModelSerializer):
    class Meta:
        model = ScrapeLog
        fields = '__all__'