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

Fold the other methods doing API calls into get_assets()

Browse Source
This commit is contained in:
Jeff Bradberry
2020-02-17 11:07:52 -05:00
parent 6162ff0439
commit fb066eb52e
1 changed files with 9 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -225,11 +225,12 @@ class Export(CustomCommand):
return self._natural_keys[url] return self._natural_keys[url]
def get_assets(self, endpoint, value): def get_assets(self, resource, value):
endpoint = getattr(self.v2, resource)
if value: if value:
from .options import pk_or_name from .options import pk_or_name
pk = pk_or_name(client.v2, resource, value) pk = pk_or_name(self.v2, resource, value)
results = endpoint.get(id=pk).json['results'] results = endpoint.get(id=pk).json['results']
else: else:
results = endpoint.get(all_pages=True).json['results'] results = endpoint.get(all_pages=True).json['results']
@@ -237,7 +238,8 @@ class Export(CustomCommand):
for asset in results: for asset in results:
self.register_natural_key(asset) self.register_natural_key(asset)
return results options = self.get_resource_options(endpoint)
return [self.enhance_asset(endpoint, asset, options) for asset in results]
def enhance_asset(self, endpoint, asset, options): def enhance_asset(self, endpoint, asset, options):
fields = { fields = {
@@ -275,16 +277,13 @@ class Export(CustomCommand):
# If no resource flags are explicitly used, export everything. # If no resource flags are explicitly used, export everything.
all_resources = all(getattr(parsed, resource, None) is None for resource in EXPORTABLE_RESOURCES) all_resources = all(getattr(parsed, resource, None) is None for resource in EXPORTABLE_RESOURCES)
self.v2 = client.v2
data = {} data = {}
for resource in EXPORTABLE_RESOURCES: for resource in EXPORTABLE_RESOURCES:
value = getattr(parsed, resource, None) value = getattr(parsed, resource, None)
if value is None and not all_resources: if all_resources or value is not None:
continue data[resource] = self.get_assets(resource, value)
endpoint = getattr(client.v2, resource)
options = self.get_resource_options(endpoint)
assets = self.get_assets(endpoint, value)
data[resource] = [self.enhance_asset(endpoint, asset, options) for asset in assets]
return data return data