60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# @author: Giancarlo Panichi
|
|
# revisited by @author: Francesco Mangiacrapa
|
|
|
|
# Created on 2023/02/01
|
|
#
|
|
import requests
|
|
import urllib.parse
|
|
|
|
|
|
class IAM_Token_Client:
|
|
|
|
def __init__(self, iamURL, the_clientId, the_clientSecret):
|
|
self.iamURL = iamURL
|
|
self.clientId = the_clientId
|
|
self.clientSecret = the_clientSecret
|
|
print('IAM URL: ' + self.iamURL)
|
|
#print('ClientId: ' + self.clientId)
|
|
#print('ClientSecret: ' + self.clientSecret)
|
|
|
|
def getAccessToken(self):
|
|
#print("getAccessToken()")
|
|
|
|
loginheaders = {'Accept': 'application/json', 'Content-Type': 'application/x-www-form-urlencoded'}
|
|
logindata = {'grant_type': 'client_credentials', 'client_id': self.clientId, 'client_secret': self.clientSecret}
|
|
|
|
# Get Access Token by client_id
|
|
resp1 = requests.post(self.iamURL, data=logindata, headers=loginheaders)
|
|
jwt1 = resp1.json()
|
|
#print("Resp AccessToken: ", jwt1)
|
|
accessToken = jwt1["access_token"]
|
|
return accessToken
|
|
|
|
def getUmaToken(self, gcube_context):
|
|
print("getUmaToken()")
|
|
context = gcube_context
|
|
|
|
#print('Context: ' + context)
|
|
|
|
context = urllib.parse.quote(context, safe='')
|
|
#print('Context safe: ' + context)
|
|
|
|
accessToken = self.getAccessToken()
|
|
|
|
umaheaders = {'Accept': 'application/json', 'Content-Type': 'application/x-www-form-urlencoded'}
|
|
umadata = {'grant_type': 'urn:ietf:params:oauth:grant-type:uma-ticket', 'audience': context}
|
|
|
|
# Get UMA token for context
|
|
umaheaders["Authorization"] = "Bearer " + accessToken
|
|
resp2 = requests.post(self.iamURL, data=umadata, headers=umaheaders)
|
|
jwt2 = resp2.json()
|
|
#print("Resp UmaToken: ", jwt2)
|
|
umaToken = jwt2["access_token"]
|
|
return umaToken
|
|
|
|
def __str__(self):
|
|
return 'IAMExample'
|