"""Account payee dataload logic."""

from owsresponse import response

from abacus_account.models.account import Account
from abacus_account.schemas.account import AccountDetailSchema
from abacus_account.utils.format_response import prepare_dataload_response

account_detail_schema = AccountDetailSchema()


def dataload_accounts_by_ids(account_ids):
    """Dataload accounts by ids."""
    accounts_by_params = Account.get_filtered_query(account_ids=account_ids).all()
    accounts_list = account_detail_schema.dump(accounts_by_params, many=True)
    result = prepare_dataload_response(account_ids, accounts_list, 'account_id')

    return response.Response({'items': result})