from uuid import UUID

from owsresponse import response

from account.models import (
    company_brand as company_brand_model,
)
from account.utils.dataloader_util import format_for_dataloader


def lookup_company_brands_by_uuids(uuids: list[UUID]) -> response.Response:
    """Lookup company brands using uuids.

    NOTE: This function is configured for an endpoint which requires NO access rule checks.
    This is because it is serving as a Policy Information Point for Permission Platform.
    Do not expose other properties or attributes via this endpoint.

    Args:
        uuids: list of uuids.
    """
    result = company_brand_model.lookup_company_brands_by_uuids(uuids)
    if result:
        result.message = {
            'company_brands': format_for_dataloader(
                result.message,
                uuids,
                'uuid',
            )
        }
    return result


def lookup_company_brands_by_ids(company_brand_ids: list[int]) -> response.Response:
    """Lookup company brands using ids.

    NOTE: This function is configured for an endpoint which requires NO access rule checks.
    This is because it is serving as a Policy Information Point for Permission Platform.
    Do not expose other properties or attributes via this endpoint.

    Args:
        company_brand_ids: list of company brand ids.
    """
    result = company_brand_model.lookup_company_brands_by_ids(company_brand_ids)
    if result:
        result.message = {
            'company_brands': format_for_dataloader(
                result.message,
                company_brand_ids,
                'company_brand_id',
            )
        }
    return result