"""
Kerberos aware Requester
"""
from jenkinsapi.utils.requester import Requester
from requests_kerberos import HTTPKerberosAuth, OPTIONAL


# pylint: disable=W0222
class KrbRequester(Requester):

    """
    A class which carries out HTTP requests with Kerberos/GSSAPI authentication.
    """

    def __init__(self, *args, **kwargs):
        """
        :param ssl_verify: flag indicating if server certificate in HTTPS requests should be
                           verified
        :param baseurl: Jenkins' base URL
        :param mutual_auth: type of mutual authentication, use one of
                            REQUIRED, OPTIONAL or DISABLED
                            from requests_kerberos package
        """

        super(KrbRequester, self).__init__(*args, **kwargs)
        self.mutual_auth = kwargs['mutual_auth'] \
            if 'mutual_auth' in kwargs else OPTIONAL

    def get_request_dict(
            self, params=None, data=None, files=None, headers=None, **kwargs):

        req_dict = super(KrbRequester, self).get_request_dict(
            params=params,
            data=data,
            files=files,
            headers=headers,
            **kwargs
        )
        if self.mutual_auth:
            auth = HTTPKerberosAuth(self.mutual_auth)
        else:
            auth = HTTPKerberosAuth()

        req_dict['auth'] = auth
        return req_dict