from .utils import NamespacedClient, query_params, _make_path

class ClusterClient(NamespacedClient):
    @query_params('level', 'local', 'master_timeout', 'timeout', 'wait_for_active_shards', 'wait_for_nodes', 'wait_for_relocating_shards', 'wait_for_status')
    def health(self, index=None, params=None):
        """
        The cluster health API allows to get a very simple status on the health of the cluster.
        http://elasticsearch.org/guide/reference/api/admin-cluster-health/

        :arg index: Limit the information returned to a specific index
        :arg level: Specify the level of detail for returned information, default u'cluster'
        :arg local: Return local information, do not retrieve the state from master node (default: false)
        :arg master_timeout: Explicit operation timeout for connection to master node
        :arg timeout: Explicit operation timeout
        :arg wait_for_active_shards: Wait until the specified number of shards is active
        :arg wait_for_nodes: Wait until the specified number of nodes is available
        :arg wait_for_relocating_shards: Wait until the specified number of relocating shards is finished
        :arg wait_for_status: Wait until cluster is in a specific state, default None
        """
        status, data = self.transport.perform_request('GET', _make_path('_cluster', 'health', index), params=params)
        return data

    @query_params('dry_run', 'filter_metadata')
    def reroute(self, body=None, params=None):
        """
        The reroute command allows to explicitly execute a cluster reroute allocation command including specific commands.
        http://elasticsearch.org/guide/reference/api/admin-cluster-reroute/

        :arg body: The definition of `commands` to perform (`move`, `cancel`, `allocate`)
        :arg dry_run: Simulate the operation only and return the resulting state
        :arg filter_metadata: Don't return cluster state metadata (default: false)
        """
        status, data = self.transport.perform_request('POST', _make_path('_cluster', 'reroute'), params=params, body=body)
        return data

    @query_params()
    def get_settings(self, params=None):
        """
        Allows to update cluster wide specific settings.
        http://elasticsearch.org/guide/reference/api/admin-cluster-update-settings/
        """
        status, data = self.transport.perform_request('GET', _make_path('_cluster', 'settings'), params=params)
        return data

    @query_params()
    def put_settings(self, body, params=None):
        """
        Allows to update cluster wide specific settings.
        http://elasticsearch.org/guide/reference/api/admin-cluster-update-settings/

        :arg body: The settings to be updated. Can be either `transient` or `persistent` (survives cluster restart).
        """
        status, data = self.transport.perform_request('PUT', _make_path('_cluster', 'settings'), params=params, body=body)
        return data