"""Test user for integration API testing."""
from os import getenv

from auth.auth import GrassAuth, UserType
import requests

from tests.consts.api import (
    DEMOGRAPHIC_INSIGHTS, GEOGRAPHIC_INSIGHTS_BY_REGION,
    GEOGRAPHIC_INSIGHTS_BY_TERRITORY, GEOGRAPHIC_ORCHARD_LATEST_DATE,
    GEOGRAPHIC_ORCHARD_REGIONS, PLAYLIST_METADATA, PLAYLIST_PLACEMENTS,
    STREAM_CHARTS)


class User(object):
    """Test user for integration API testing."""

    def __init__(self, user_id):
        """Create a test user of ows-analytics."""
        self.auth = {}
        self.id = user_id

    def get_demographic_insights(self, **body):
        """Get demographic insights response.

        Args:
            body (dict): dict should include:
                            from_date
                            to_date
                            stores
                            isrcs
        """
        body['session'] = self.auth.get('session_token', '')
        url = self.workstation_url + DEMOGRAPHIC_INSIGHTS
        response = requests.get(url, params=body)

        return response

    def get_stream_charts(self, **body):
        """Get stream charts response.

        Args:
            body (dict): dict should include:
                            from_date
                            to_date
                            stores
                            isrcs
        """
        headers = {}
        body['session'] = self.auth.get('session_token', '')
        url = self.workstation_url + STREAM_CHARTS

        response = requests.get(url, headers=headers, params=body)

        return response

    def get_playlist_placements(self, **body):
        """Get playlist placements response.

        Args:
            body (dict): dict should include:
                            from_date
                            to_date
                            stores
                            isrcs
        """
        headers = {}
        body['session'] = self.auth.get('session_token', '')
        url = self.workstation_url + PLAYLIST_PLACEMENTS

        response = requests.get(url, headers=headers, params=body)

        return response

    def get_playlist_metadata(self, **body):
        """Get playlist metadata response.

        Args:
            body (dict): dict should include:
                            playlist_links
        """
        headers = {}
        body['session'] = self.auth.get('session_token', '')
        url = self.workstation_url + PLAYLIST_METADATA

        response = requests.get(url, headers=headers, params=body)

        return response

    def get_geographic_by_territory(self, **body):
        """Get geographic insights by territory response.

        Args:
            body (dict): dict should include:
                            from_date
                            to_date
        """
        headers = {}
        body['session'] = self.auth.get('session_token', '')
        url = self.workstation_url + GEOGRAPHIC_INSIGHTS_BY_TERRITORY

        response = requests.get(url, headers=headers, params=body)

        return response

    def get_geographic_by_region(self, **body):
        """Get geographic insights by region response.

        Args:
            body (dict): dict should include:
                            from_date
                            to_date
        """
        headers = {}
        body['session'] = self.auth.get('session_token', '')
        url = self.workstation_url + GEOGRAPHIC_INSIGHTS_BY_REGION

        response = requests.get(url, headers=headers, params=body)

        return response

    def get_geographic_orchard_regions(self):
        """Get geographic orchard regions response."""
        headers = {}
        body = {}
        body['session'] = self.auth.get('session_token', '')
        url = self.workstation_url + GEOGRAPHIC_ORCHARD_REGIONS

        response = requests.get(url, headers=headers, params=body)

        return response

    def get_geographics_latest_date(self):
        """Get geographic orchard latest available date response."""
        headers = {}
        body = {}
        body['session'] = self.auth.get('session_token', '')
        url = self.workstation_url + GEOGRAPHIC_ORCHARD_LATEST_DATE

        response = requests.get(url, headers=headers, params=body)

        return response


class LocalUser(User):
    """Test user for integration API testing in a local environment."""

    def __init__(self, user_id):
        """Create a test user of ows-analytics."""
        super().__init__(user_id)
        self.workstation_url = getenv('LOCAL_BASE_URL')


class QAUser(User):
    """Test user for integration API testing in QA environment."""

    def __init__(self, user_id):
        """Create a test user of ows-analytics."""
        super().__init__(user_id)
        self.auth['session_token'] = GrassAuth(UserType.WORKSTATION,
                                               user_id).grass_token
        self.workstation_url = getenv('QA_BASE_URL')