Source code for wolframclient.evaluation.cloud.server

# -*- coding: utf-8 -*-

from __future__ import absolute_import, print_function, unicode_literals

from wolframclient.utils import six

__all__ = ['WolframServer']


class WolframServer(object):
    ''' Represents the cloud server.

    Contains the authentication endpoints information, the API endpoint aka. the
    cloud base (`$CloudBase` in the Wolfram Language), and eventually the xauth
    consumer key and secret.
    '''

    def __init__(self,
                 cloudbase,
                 request_token_endpoint,
                 access_token_endpoint,
                 xauth_consumer_key=None,
                 xauth_consumer_secret=None,
                 certificate=None):
        self.cloudbase = cloudbase
        self.request_token_endpoint = request_token_endpoint
        self.access_token_endpoint = access_token_endpoint
        self.xauth_consumer_key = xauth_consumer_key
        self.xauth_consumer_secret = xauth_consumer_secret
        if certificate is None or isinstance(certificate, six.string_types):
            self.certificate = certificate
        else:
            raise ValueError(
                'Invalid certificate. Must be a string type or None.')

    def is_xauth(self):
        return self.xauth_consumer_key is not None and self.xauth_consumer_secret is not None

    def __repr__(self):
        return '<%s: cloudbase=%s, request_token=%s, access_token=%s, certificate=%s, xauth support=%s>' % (
            self.__class__.__name__, self.cloudbase,
            self.request_token_endpoint, self.access_token_endpoint,
            self.certificate, self.is_xauth())


# A built-in instance representing the Wolfram Public Cloud.
WOLFRAM_PUBLIC_CLOUD_SERVER = WolframServer(
    'https://www.wolframcloud.com',
    'https://account.wolfram.com/auth/request-token',
    'https://account.wolfram.com/auth/access-token',
    xauth_consumer_key=None,
    xauth_consumer_secret=None)