"""Logic for payee KYC notification."""

import logging

from payee.models.payee_kyc_notification import PayeeKycNotification

logger = logging.getLogger('payee_kyc_notification')


def get_payee_kyc_notification_dataloader(
    payee_ids: list[int],
) -> 'dict':
    notifications = PayeeKycNotification.get_latest_by_payee_ids(payee_ids)
    notification_map = {
        notification.payee_id: notification for notification in notifications
    }

    return {
        'payee_kyc_notifications': [
            {'data': notification_map.get(payee_id)} for payee_id in payee_ids
        ]
    }