"""API Client for ows-abacus-state."""

import requests


class AbacusStateAPIClient:
    """API Client for ows-abacus-state."""

    def __init__(self, base_url, headers):
        """Initialize ows_abacus_state_APIClient class."""
        self.base_url = base_url
        self.headers = headers

    def post_states(self, body):
        """Execute a POST against /abacus-state endpoint."""
        endpoint = '{}/abacus-states'.format(self.base_url)
        return requests.post(endpoint, headers=self.headers, json=body)

    def update_status(self, abacus_action_id, new_status):
        """Execute a POST against /abacus-state endpoint."""
        endpoint = '{}/abacus-state/{}'.format(self.base_url, abacus_action_id)
        return requests.put(
            endpoint, headers=self.headers, json={'action_status': new_status}
        )