restClient.py

import requests
from configs import default


class RestClient(object):
    _instance = None
    APP_SCOPES = {
        "publisher": 'apim:api_view apim:api_create apim:api_publish apim:tier_view apim:tier_manage apim:subscription_view apim:subscription_block apim:subscribe',
        "store": 'apim:subscribe apim:signup apim:workflow_approve'
    }

    def __init__(self, application="publisher"):
        self.base_path = "https://{host}:{port}".format(
            host=default['connection']['hostname'], port=default['connection']['port'])
        self.session = requests.Session()
        self.verify = False
        self.application = application
        self._get_access_token()
        RestClient._client = self

    def _get_access_token(self):
        form_data = {
            'username': 'admin',
            'password': 'admin',
            'grant_type': 'password',
            'validity_period': '3600',
            'scopes': RestClient.get_scopes_for_app(self.application)
        }
        login_endpoint = self.base_path + \
            "/login/token/{application}".format(application=self.application)
        login = self.session.post(
            login_endpoint, data=form_data, verify=self.verify)
        if not login.ok:
            print("Error: {}".format(login.reason))
            return False
        # This is to ignore the environment prefix and get the token value
        token_part_1 = login.json()['partialToken']
        token_part_2 = [v for k, v in self.session.cookies.iteritems(
        ) if k.startswith("WSO2_AM_TOKEN_2")].pop()
        self.access_token = token_part_1 + token_part_2
        print("DEBUG: Access token = {}".format(self.access_token))
        self.session.headers['Authorization'] = 'Bearer ' + token_part_1
        print("session headers: {}".format(self.session.headers))

    def create_application(self):
        pass
    @staticmethod
    def get_scopes_for_app(application):
        return RestClient.APP_SCOPES[application]


# stuff to run always here such as class/def
def main():
    pass


if __name__ == "__main__":
    # stuff only to run when not called via 'import' here
    main()