def get_acls(env, provider, token):
""" Method used to get all ACLs for given provider """
logger = util.get_logger(env)
logger.debug('TEA configuraton ACL')
cmr_url = util.get_env(env)
headers = util.standard_headers({
'Authorization': token,
'Content-Type': 'application/json'
})
url = (f'{cmr_url}/access-control/acls'
f'?provider={provider}'
f'&identity_type=catalog_item'
f'&page_size=2000')
try:
response = requests.get(url, headers=headers)
json_data = response.json()
logging.debug('get_acls: response=%s', json_data)
if response.status_code == 200:
if 'items' in json_data:
items = json_data['items']
return items
except requests.exceptions.RequestException as error:
logging.error('Error occurred in get_acl: %s', error)
return []
def test_standard_headers(self):
"""
Test that the standard header function always returns the correct user agent
"""
test = lambda e, p, m: self.assertEqual(e, p['User-Agent'], m)
expected = 'ESDIS TEA Config Generator'
test(expected, util.standard_headers(), 'dictionary not provided')
test(expected, util.standard_headers(None), 'none dictionary')
test(expected, util.standard_headers({'u': 'you'}),
'dictionary provided')
test(expected, util.standard_headers({'User-Agent': 'Wrong'}),
'overwrite check')
actual = util.standard_headers({'Other-Header': 'Keep'})
self.assertEqual('Keep', actual['Other-Header'], 'other values')
def get_collection(env:dict, token, concept_id):
""" Method returns collection for given concept_id """
logger = util.get_logger(env)
headers = util.standard_headers({'Authorization': token, 'Content-Type': 'application/json'})
cmr_base = util.get_env(env)
url = f'{cmr_base}/search/concepts/{concept_id}.umm_json'
try:
response = requests.get(env, url, headers=headers)
json_data = response.json()
logger.debug('get_collection: response=%s', json_data)
if response.status_code == 200:
return json_data
except requests.exceptions.RequestException as error:
logger.error('Error occurred in get_collection: %s', error)
return {}
def get_acl(env, acl_url, token):
""" Method retrieves ACL for given ACL URL """
logger = util.get_logger(env)
headers = util.standard_headers({
'Authorization': token,
'Content-Type': 'application/json'
})
try:
response = requests.get(acl_url, headers=headers)
json_data = response.json()
logger.debug("get_acl: response=%s", json_data)
if response.status_code == 200:
return json_data
except requests.exceptions.RequestException as error:
logging.error('Error occurred in get_acl: %s', error)
return {}
def get_group(env: dict, group_id, token):
""" Method used to get group for given group ID """
logger = util.get_logger({})
cmr_base = util.get_env(env)
url = f'{cmr_base}/access-control/groups/{group_id}'
headers = util.standard_headers({
'Authorization': token,
'Content-Type': 'application/json'
})
try:
response = requests.get(url, headers=headers)
json_data = response.json()
logger.debug('get_group: response=%s', json_data)
if response.status_code == 200:
return json_data
except requests.exceptions.RequestException as error:
logger.error('Error occurred in get_group: %s', error)
return {}
def get_groups(env: dict, acl_url, token):
""" Method gets groups for given ACL URL """
logger = util.get_logger(env)
headers = util.standard_headers({
'Authorization': token,
'Content-Type': 'application/json'
})
try:
response = requests.get(acl_url, headers=headers)
json_data = response.json()
logger.debug('get_groups: response=%s', json_data)
if response.status_code == 200:
if 'group_permissions' in json_data:
items = json_data['group_permissions']
return items
except requests.exceptions.RequestException as error:
logger.error('Error occurred in get_groups: %s', error)
return []
def get_collections(env:dict, token, provider, page_num, page_size):
""" Method returns collections for given provider """
logger = util.get_logger(env)
headers = util.standard_headers({'Authorization': token})
cmr_base = util.get_env(env)
url = (f'{cmr_base}/search/collections.umm-json'
f'?provider={provider}'
f'&sort_key=entry_title'
f'&page_num={page_num}'
f'&page_size={page_size}')
try:
logger.debug('request url: %s', url)
response = requests.get(url, headers=headers)
json_data = response.json()
logger.debug('get_collections: response=%s', json_data)
if response.status_code == 200:
return json_data
except requests.exceptions.RequestException as error:
logger.error('Error occurred in get_collections from calling %s:\n%s', url, error)
return {}