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

from typing import Optional

from src.connectors.requests import get
from src.exceptions import OwsRoyaltiesException
from src.models import PaymentEntity

SERVICE = 'ows-royalties'


def get_payment_entity(payment_entity_id: int) -> Optional[PaymentEntity]:
    """Get the payment_entity details.

    Args:
        payment_entity_id (int): Id of payment_entity
    """
    path = f'/reference-payment-entity/{payment_entity_id}'
    response = get(SERVICE, path)

    if response.status_code != 200:
        raise OwsRoyaltiesException(f'ERROR in GET {path}')

    return PaymentEntity.model_validate(response.json())