External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-05-20 07:17:40 -02:30
Code Issues Packages Projects Releases Wiki Activity

Export full related objects under some conditions

Browse Source
This commit is contained in:
Jeff Bradberry
2020-03-27 16:08:45 -04:00
parent a1f7d0b781
commit 972d3ab535
1 changed files with 21 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
awxkit/awxkit/api/pages/api.py
View File

@@ -24,6 +24,10 @@ EXPORTABLE_RESOURCES = [
EXPORTABLE_RELATIONS = [ EXPORTABLE_RELATIONS = [
'Roles', 'Roles',
'NotificationTemplates', 'NotificationTemplates',
]
EXPORTABLE_DEPENDENT_OBJECTS = [
'Labels', 'Labels',
'SurveySpec', 'SurveySpec',
'WorkflowJobTemplateNodes', 'WorkflowJobTemplateNodes',
@@ -105,7 +109,7 @@ class ApiV2(base.Base):
# Note: doing asset[key] automatically parses json blob strings, which can be a problem. # Note: doing asset[key] automatically parses json blob strings, which can be a problem.
fields = { fields = {
key: asset.json[key] for key in options key: asset.json[key] for key in options
if key in asset.json and key not in asset.related if key in asset.json and key not in asset.related and key != 'id'
} }
fields['natural_key'] = get_natural_key(asset) fields['natural_key'] = get_natural_key(asset)
@@ -115,19 +119,29 @@ class ApiV2(base.Base):
} }
related = {} related = {}
for k, related_endpoint in asset.related.items(): for key, related_endpoint in asset.related.items():
if not related_endpoint: if key in asset.json or not related_endpoint:
continue continue
if k == 'object_roles': if key == 'object_roles':
continue continue
rel = related_endpoint._create() rel = related_endpoint._create()
if rel.__class__.__name__ not in EXPORTABLE_RELATIONS:
if rel.__class__.__name__ in EXPORTABLE_RELATIONS:
by_natural_key = True
elif rel.__class__.__name__ in EXPORTABLE_DEPENDENT_OBJECTS:
by_natural_key = False
else:
continue continue
data = related_endpoint.get(all_pages=True) data = related_endpoint.get(all_pages=True)
if 'results' in data: if 'results' in data:
related[k] = [get_natural_key(x) for x in data.results] related_options = self._get_options(related_endpoint)
related[key] = [
get_natural_key(x) if by_natural_key else self._serialize_asset(x, related_options)
for x in data.results
]
else: else:
related[k] = data.json related[key] = data.json
except exc.Forbidden: except exc.Forbidden:
return None return None