global_configuration.py

from requests.structures import CaseInsensitiveDict

from apimatic_core.http.configurations.http_client_configuration import HttpClientConfiguration
from apimatic_core.utilities.api_helper import ApiHelper


class GlobalConfiguration:

    def get_http_client_configuration(self):
        return self._http_client_configuration

    def get_global_errors(self):
        return self._global_errors

    def get_global_headers(self):
        return self._global_headers

    def get_additional_headers(self):
        return self._additional_headers

    def get_auth_managers(self):
        return self._auth_managers

    def get_base_uri(self, server):
        return self._base_uri_executor(server)

    def __init__(
            self, http_client_configuration=HttpClientConfiguration()
    ):
        self._http_client_configuration = http_client_configuration
        self._global_errors = {}
        self._global_headers = CaseInsensitiveDict()
        self._additional_headers = CaseInsensitiveDict()
        self._auth_managers = {}
        self._base_uri_executor = None

    def global_errors(self, global_errors):
        self._global_errors = global_errors
        return self

    def global_headers(self, global_headers):
        self._global_headers = global_headers
        return self

    def global_header(self, key, value):
        self._global_headers[key] = value
        return self

    def additional_headers(self, additional_headers):
        self._additional_headers = additional_headers
        return self

    def additional_header(self, key, value):
        self._additional_headers[key] = value
        return self

    def auth_managers(self, auth_managers):
        self._auth_managers = auth_managers
        return self

    def user_agent(self, user_agent, user_agent_parameters={}):
        self.add_useragent_in_global_headers(user_agent, user_agent_parameters)
        return self

    def base_uri_executor(self, base_uri_executor):
        self._base_uri_executor = base_uri_executor
        return self

    def add_useragent_in_global_headers(self, user_agent, user_agent_parameters):
        if user_agent_parameters:
            user_agent = ApiHelper.append_url_with_template_parameters(
                user_agent, user_agent_parameters).replace('  ', ' ')
        if user_agent:
            self._global_headers['user-agent'] = user_agent