"""API Client for pure ows-features."""
import requests


class FeaturesAPIClient:
    """API Client for ows-features."""

    def __init__(self, base_url):
        """Initialize ows_accounting_APIClient class."""
        self.base_url = base_url
        self.headers = {'Correlation-Id': 'test',
                        'Content-Type': 'application/json'}

    def update_feature(self, feature_name, variant_name, user_id):
        """Update a feature flag for a given user."""
        endpoint = '{}/features/{}/user/alw:{}'.format(
            self.base_url, feature_name, user_id)
        return requests.put(
            endpoint,
            headers=self.headers,
            json={'variant_name': variant_name})