예제 #1
0
    def test_credentials(self):

        datadir = os.path.join(os.path.dirname(__file__), 'data')
        os.environ['HOME'] = datadir

        # Make sure we don't have have the test solvebio directory
        if os.path.isdir(self.solvebiodir):
            shutil.rmtree(self.solvebiodir)

        cred_file = creds.netrc.path()
        self.assertTrue(os.path.exists(cred_file),
                        "cred file created when it doesn't exist first")

        self.assertEqual(creds.get_credentials(), None,
                         'Should not find credentials')

        test_credentials_file = os.path.join(datadir, 'test_creds')
        shutil.copy(test_credentials_file, cred_file)

        auths = creds.get_credentials()
        self.assertTrue(auths is not None, 'Should find credentials')

        solvebio.api_host = 'https://example.com'

        auths = creds.get_credentials()
        self.assertEqual(
            auths, None, 'Should not find credentials for host {0}'.format(
                solvebio.api_host))

        solvebio.api_host = 'https://api.solvebio.com'
        creds.delete_credentials()
        auths = creds.get_credentials()
        self.assertEqual(
            auths, None, 'Should not find removed credentials for '
            'host {0}'.format(solvebio.api_host))

        pair = (
            '*****@*****.**',
            'b00b00',
        )
        creds.save_credentials(*pair)
        auths = creds.get_credentials()
        self.assertTrue(
            auths is not None, 'Should get newly set credentials for '
            'host {0}'.format(solvebio.api_host))

        expected = (solvebio.api_host, pair[0], 'Token', pair[1])
        self.assertEqual(auths, expected, 'Should get back creds we saved')
예제 #2
0
    def test_credentials(self):

        datadir = os.path.join(os.path.dirname(__file__), 'data')
        os.environ['HOME'] = datadir

        # Make sure we don't have have the test solvebio directory
        if os.path.isdir(self.solvebiodir):
            shutil.rmtree(self.solvebiodir)

        cred_file = creds.netrc.path()
        self.assertTrue(os.path.exists(cred_file),
                        "cred file created when it doesn't exist first")

        self.assertEqual(creds.get_credentials(), None,
                         'Should not find credentials')

        test_credentials_file = os.path.join(datadir, 'test_creds')
        shutil.copy(test_credentials_file, cred_file)

        auths = creds.get_credentials()
        self.assertTrue(auths is not None, 'Should find credentials')

        solvebio.api_host = 'https://example.com'

        auths = creds.get_credentials()
        self.assertEqual(auths, None,
                         'Should not find credentials for host {0}'
                         .format(solvebio.api_host))

        solvebio.api_host = 'https://api.solvebio.com'
        creds.delete_credentials()
        auths = creds.get_credentials()
        self.assertEqual(auths, None,
                         'Should not find removed credentials for '
                         'host {0}'.format(solvebio.api_host))

        pair = ('*****@*****.**', 'b00b00',)
        creds.save_credentials(*pair)
        auths = creds.get_credentials()
        self.assertTrue(auths is not None,
                        'Should get newly set credentials for '
                        'host {0}'.format(solvebio.api_host))

        self.assertEqual(auths, pair[1], 'Should get back creds we saved')
예제 #3
0
    def _get_api_key(self):
        """
        Helper function to get the current user's API key or None.
        """
        if solvebio.api_key:
            return solvebio.api_key

        try:
            return get_credentials()[1]
        except:
            pass

        return None
예제 #4
0
from django.conf import settings
import solvebio

API_HOST = getattr(settings, 'SOLVEBIO_API_HOST', None)
API_KEY = getattr(settings, 'SOLVEBIO_API_KEY', None)
DATASET_ALIASES = getattr(settings, 'SOLVEBIO_DATASET_ALIASES', {})


if API_HOST:
    solvebio.api_host = API_HOST

if API_KEY:
    solvebio.api_key = API_KEY
else:
    # attempt to get API key from get_credentials (local .netrc)
    from solvebio.cli.credentials import get_credentials

    try:
        solvebio.api_key = API_KEY = get_credentials()[1]
    except:
        pass