"""Requests to ows-royalties."""

import httpx
from lambdacommon.common_config import logger

from src.connectors.requests import get
from src.constants.constants import GET_STATEMENT_PERIOD_PAYMENT_ENTITY_STATES

SERVICE = 'ows-royalties'


def get_statement_period_payment_entity_states(
    statement_period_id: int | str,
) -> httpx.Response:
    """Get statement_period_payment_entity states by statement_period_id.

    Args:
        statement_period_id: ID of the statement period

    Returns:
        OWS Response
    """
    logger.info(
        GET_STATEMENT_PERIOD_PAYMENT_ENTITY_STATES.format(
            statement_period_id=statement_period_id
        )
    )

    path = f'/statement-period/{statement_period_id}/payment-entities/states'
    response = get(SERVICE, path)
    return response