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

Split out separate methods for the OPTIONS call and massaging each asset dict

Browse Source
This commit is contained in:
Jeff Bradberry
2020-02-03 11:06:08 -05:00
parent 9280198b0f
commit 37cbf7691b
1 changed files with 17 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -178,18 +178,25 @@ class Export(CustomCommand):
# 3) the resource flag is used with an argument, and the attr will be that argument's value # 3) the resource flag is used with an argument, and the attr will be that argument's value
resources.add_argument('--{}'.format(resource), nargs='?', const='') resources.add_argument('--{}'.format(resource), nargs='?', const='')
def get_resources(self, client, resource, value): def get_resource_options(self, endpoint):
api_resource = getattr(client.v2, resource) return endpoint.options().json['actions']['POST']
post_options = api_resource.options().json['actions']['POST']
def get_assets(self, endpoint, value):
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(client.v2, resource, value)
results = api_resource.get(id=pk).json['results'] results = endpoint.get(id=pk).json['results']
else: else:
results = api_resource.get(all_pages=True).json['results'] results = endpoint.get(all_pages=True).json['results']
return [{key: r[key] for key in post_options if key in r} for r in results] return results
def enhance_asset(self, endpoint, asset, options):
fields = {key: asset[key] for key in options if key in asset}
fk_fields = {}
related_fields = {}
return dict(**fields, **fk_fields, **related_fields)
def handle(self, client, parser): def handle(self, client, parser):
self.extend_parser(parser) self.extend_parser(parser)
@@ -209,9 +216,11 @@ class Export(CustomCommand):
value = getattr(parsed, resource, None) value = getattr(parsed, resource, None)
if value is None and not all_resources: if value is None and not all_resources:
continue continue
resources = self.get_resources(client, resource, value) endpoint = getattr(client.v2, resource)
options = self.get_resource_options(endpoint)
assets = self.get_assets(endpoint, value)
data[resource] = resources data[resource] = [self.enhance_asset(endpoint, asset, options) for asset in assets]
return data return data