Exemplos de get_api_key em Python

Exemplo n.º 1

    def test_get_api_key_api_error(self, mock_validate_api_key, mock_prompt):
        with fake_creds('.jovian-get-api-key', 'credentials.json'):
            creds = {
                "WEBAPP_URL": "https://staging.jovian.ml/",
                "ORG_ID": "staging",
                "API_URL": "https://api-staging.jovian.ai"
            }
            write_creds(creds)

            with self.assertRaises(ApiError):
                get_api_key()

Exemplo n.º 2

Arquivo: test_credentials.py Projeto: pplonski/jovian-py

def test_get_api_key_api_error(mock_validate_api_key, mock_prompt):
    with fake_creds():
        creds = {
            "WEBAPP_URL": "https://staging.jovian.ai/",
            "ORG_ID": "staging",
            "API_URL": "https://api-staging.jovian.ai"
        }
        write_creds(creds)

        with pytest.raises(ApiError):
            get_api_key()

Exemplo n.º 3

Arquivo: api.py Projeto: vaishnavipatil29/jovian-py

def _h():
    """Create authorization header with API key"""
    return {"Authorization": "Bearer " + get_api_key(),
            "x-jovian-source": "library",
            "x-jovian-library-version": __version__,
            "x-jovian-guest": get_guest_key(),
            "x-jovian-org": read_org_id()}

Exemplo n.º 4

Arquivo: slack.py Projeto: vaishnavipatil29/jovian-py

def _h():
    """Create a header to provide library metadata"""
    return {"Authorization": "Bearer " + get_api_key(),
            "x-jovian-source": "library",
            "x-jovian-library-version": __version__,
            "x-jovian-command": "add-slack",
            "x-jovian-guest": get_guest_key(),
            "x-jovian-org": read_org_id()}

Exemplo n.º 5

Arquivo: test_credentials.py Projeto: pplonski/jovian-py

def test_get_api_key_request_once(mock_validate_api_key, mock_prompt):
    with fake_creds():
        creds = {
            "WEBAPP_URL": "https://staging.jovian.ai/",
            "ORG_ID": "staging",
            "API_URL": "https://api-staging.jovian.ai"
        }
        write_creds(creds)

        assert get_api_key() == "fake_api_key"

Exemplo n.º 6

Arquivo: test_credentials.py Projeto: vaishnavipatil29/jovian-py

def test_get_api_key(mock_validate_api_key):
    with fake_creds():
        creds = {
            "WEBAPP_URL": "https://staging.jovian.ml/",
            "ORG_ID": "staging",
            "API_KEY": "fake_api_key",
            "API_URL": "https://api-staging.jovian.ai"
        }
        write_creds(creds)

        assert get_api_key() == "fake_api_key"

Exemplo n.º 7

    def test_get_api_key_request_once(self, mock_validate_api_key,
                                      mock_prompt):
        with fake_creds('.jovian-get-api-key', 'credentials.json'):
            creds = {
                "WEBAPP_URL": "https://staging.jovian.ml/",
                "ORG_ID": "staging",
                "API_URL": "https://api-staging.jovian.ai"
            }
            write_creds(creds)

            assert get_api_key() == "fake_api_key"

Exemplo n.º 8

Arquivo: request.py Projeto: vaishnavipatil29/jovian-py

    def _request_wrapper(*args, **kwargs):
        for i in range(2):
            res = request(*args, **kwargs)
            if res.status_code == 401:
                log('The current API key is invalid or expired.', error=True)
                purge_api_key()

                # This will ensure that fresh api token is requested
                if 'headers' in kwargs:
                    kwargs['headers'][
                        'Authorization'] = "Bearer " + get_api_key()
            else:
                return res
        return res