mirror of https://github.com/trailofbits/algo
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
94 lines
2.9 KiB
Python
94 lines
2.9 KiB
Python
5 years ago
|
#!/usr/bin/python
|
||
|
# -*- coding: utf-8 -*-
|
||
|
|
||
|
from __future__ import absolute_import, division, print_function
|
||
|
|
||
|
__metaclass__ = type
|
||
|
|
||
|
################################################################################
|
||
|
# Documentation
|
||
|
################################################################################
|
||
|
|
||
|
ANSIBLE_METADATA = {'metadata_version': '1.1', 'status': ["preview"], 'supported_by': 'community'}
|
||
|
|
||
|
################################################################################
|
||
|
# Imports
|
||
|
################################################################################
|
||
|
from ansible.module_utils.gcp_utils import navigate_hash, GcpSession, GcpModule, GcpRequest
|
||
|
import json
|
||
|
|
||
|
################################################################################
|
||
|
# Main
|
||
|
################################################################################
|
||
|
|
||
|
|
||
|
def main():
|
||
|
module = GcpModule(argument_spec=dict(filters=dict(type='list', elements='str'), scope=dict(required=True, type='str')))
|
||
|
|
||
|
if module._name == 'gcp_compute_image_facts':
|
||
|
module.deprecate("The 'gcp_compute_image_facts' module has been renamed to 'gcp_compute_regions_info'", version='2.13')
|
||
|
|
||
|
if not module.params['scopes']:
|
||
|
module.params['scopes'] = ['https://www.googleapis.com/auth/compute']
|
||
|
|
||
|
items = fetch_list(module, collection(module), query_options(module.params['filters']))
|
||
|
if items.get('items'):
|
||
|
items = items.get('items')
|
||
|
else:
|
||
|
items = []
|
||
|
return_value = {'resources': items}
|
||
|
module.exit_json(**return_value)
|
||
|
|
||
|
|
||
|
def collection(module):
|
||
|
return "https://www.googleapis.com/compute/v1/projects/{project}/{scope}".format(**module.params)
|
||
|
|
||
|
|
||
|
def fetch_list(module, link, query):
|
||
|
auth = GcpSession(module, 'compute')
|
||
|
response = auth.get(link, params={'filter': query})
|
||
|
return return_if_object(module, response)
|
||
|
|
||
|
|
||
|
def query_options(filters):
|
||
|
if not filters:
|
||
|
return ''
|
||
|
|
||
|
if len(filters) == 1:
|
||
|
return filters[0]
|
||
|
else:
|
||
|
queries = []
|
||
|
for f in filters:
|
||
|
# For multiple queries, all queries should have ()
|
||
|
if f[0] != '(' and f[-1] != ')':
|
||
|
queries.append("(%s)" % ''.join(f))
|
||
|
else:
|
||
|
queries.append(f)
|
||
|
|
||
|
return ' '.join(queries)
|
||
|
|
||
|
|
||
|
def return_if_object(module, response):
|
||
|
# If not found, return nothing.
|
||
|
if response.status_code == 404:
|
||
|
return None
|
||
|
|
||
|
# If no content, return nothing.
|
||
|
if response.status_code == 204:
|
||
|
return None
|
||
|
|
||
|
try:
|
||
|
module.raise_for_status(response)
|
||
|
result = response.json()
|
||
|
except getattr(json.decoder, 'JSONDecodeError', ValueError) as inst:
|
||
|
module.fail_json(msg="Invalid JSON response with error: %s" % inst)
|
||
|
|
||
|
if navigate_hash(result, ['error', 'errors']):
|
||
|
module.fail_json(msg=navigate_hash(result, ['error', 'errors']))
|
||
|
|
||
|
return result
|
||
|
|
||
|
|
||
|
if __name__ == "__main__":
|
||
|
main()
|