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

import requests


class AbacusContractAPIClient:
    """API Client for ows-abacus-contract."""

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

    def post_contract(self, body):
        """Execute a POST against /contract endpoint."""
        endpoint = '{}/contract'.format(self.base_url)
        return requests.post(endpoint, headers=self.headers, json=body)

    def get_contract(self, contract_id):
        """Execute a GET against /contract/<contract_id> endpoint."""
        endpoint = '{}/contract/{}'.format(self.base_url, contract_id)
        return requests.get(endpoint, headers=self.headers)

    def get_contracts(self):
        """Execute a GET against /contracts endpoint."""
        endpoint = '{}/contracts'.format(self.base_url)
        return requests.get(endpoint, headers=self.headers)

    def generate_contract_body(self, contract_name, contract_type='distribution'):
        """Generate a body for POST /contract endpoint."""
        return {
            'contract_name': contract_name,
            'contract_type': contract_type,
            'term_start': '2019-06-01',
            'term_end': '2020-06-01',
            'reference_signing_entity_id': 1,
        }