"""API Client for airflow."""

import requests


class AirflowAPIClient:
    """API Client for airflow."""

    def __init__(self, base_url):
        """Set API URL."""
        self.base_url = base_url

    def get_dag_runs(self, dag_id):
        """Get execution statuses for a dag."""
        endpoint = '{}/api/experimental/dags/{}/dag_runs'.format(self.base_url, dag_id)
        return requests.get(endpoint)

    def get_status(self):
        """Execute a POST against /accounting-period endpoint."""
        endpoint = '{}/api/experimental/test'.format(self.base_url)
        return requests.get(endpoint)