"""API Client for ows-product."""
import requests

PRODUCT_STRING = 'product'


class APIClient:
    """API Client for ows-product."""

    def __init__(self, base_url, session):
        """Initialize APIClient class."""
        self.base_url = base_url
        self.session = session

    def bulk_get_track_localization(self, tuids):
        """Execute a POST against localization/track."""
        endpoint = '{}/{}/localization/track'.format(
            self.base_url, PRODUCT_STRING)
        return requests.post(endpoint, headers={'session': self.session},
                             json={'tuids': tuids})

    def get_track_localization(self, tuids):
        """Execute a GET against localization/track."""
        endpoint = '{}/{}/localization/track/{}'.format(
            self.base_url, PRODUCT_STRING, ','.join(map(str, tuids)))
        return requests.get(endpoint, headers={'session': self.session})