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

Modify get_natural_key to work on Pages

Browse Source
This commit is contained in:
Jeff Bradberry
2020-02-17 15:16:21 -05:00
parent e4383c505f
commit 0877e5305c
1 changed files with 8 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
awxkit/awxkit/cli/resource.py
View File

@@ -200,30 +200,15 @@ class Export(CustomCommand):
def get_options(self, endpoint): def get_options(self, endpoint):
return endpoint.options().json['actions']['POST'] return endpoint.options().json['actions']['POST']
def register_natural_key(self, asset): def get_natural_key(self, page):
if asset['url'] in self._natural_keys: natural_key = {'type': page.type}
return lookup = NATURAL_KEYS.get(page.type)
natural_key = {'type': asset['type']}
lookup = NATURAL_KEYS.get(asset['type'])
if callable(lookup): if callable(lookup):
natural_key.update(lookup(asset)) natural_key.update(lookup(page))
else: else:
natural_key.update((key, asset.get(key)) for key in lookup or ()) natural_key.update((key, page[key]) for key in lookup or ())
self._natural_keys[asset['url']] = natural_key return natural_key
def get_natural_key(self, url=None, asset=None):
if url is None:
url = asset['url']
if url not in self._natural_keys:
if asset is None:
# get the asset by following the url
raise Exception("Oops!")
self.register_natural_key(asset)
return self._natural_keys[url]
def get_assets(self, resource, value): def get_assets(self, resource, value):
endpoint = getattr(self.v2, resource) endpoint = getattr(self.v2, resource)
@@ -235,9 +220,6 @@ class Export(CustomCommand):
else: else:
results = endpoint.get(all_pages=True).results results = endpoint.get(all_pages=True).results
for asset in results:
self.register_natural_key(asset.json)
options = self.get_options(endpoint) options = self.get_options(endpoint)
return [self.serialize_asset(asset, options) for asset in results] return [self.serialize_asset(asset, options) for asset in results]
@@ -248,7 +230,7 @@ class Export(CustomCommand):
} }
fk_fields = { fk_fields = {
key: self.get_natural_key(url=asset.related[key].endpoint) for key in options key: self.get_natural_key(asset.related[key].get()) for key in options
if key in asset.json and options[key]['type'] == 'id' if key in asset.json and options[key]['type'] == 'id'
} }
@@ -258,7 +240,7 @@ class Export(CustomCommand):
continue continue
data = related_endpoint.get(all_pages=True).json data = related_endpoint.get(all_pages=True).json
if 'results' in data: if 'results' in data:
related[k] = [self.get_natural_key(asset=x) for x in data['results']] related[k] = [self.get_natural_key(x) for x in data.results]
related_fields = {'related': related} if related else {} related_fields = {'related': related} if related else {}
return dict(**fields, **fk_fields, **related_fields) return dict(**fields, **fk_fields, **related_fields)