mirror of
https://github.com/ansible/awx.git
synced 2026-03-03 01:38:50 -03:30
Refactor Inventory Plugin module to import from module_utils
This commit is contained in:
@@ -98,15 +98,15 @@ inventory_id: the_ID_of_targeted_ansible_tower_inventory
|
|||||||
# ansible-inventory -i @tower_inventory --list
|
# ansible-inventory -i @tower_inventory --list
|
||||||
'''
|
'''
|
||||||
|
|
||||||
import re
|
|
||||||
import os
|
import os
|
||||||
import json
|
import re
|
||||||
from ansible.module_utils import six
|
from ansible.module_utils import six
|
||||||
from ansible.module_utils.urls import Request, urllib_error, ConnectionError, socket, httplib
|
from ansible.module_utils.urls import Request
|
||||||
from ansible.module_utils._text import to_text, to_native
|
from ansible.module_utils._text import to_text, to_native
|
||||||
from ansible.errors import AnsibleParserError, AnsibleOptionsError
|
from ansible.errors import AnsibleOptionsError
|
||||||
from ansible.plugins.inventory import BaseInventoryPlugin
|
from ansible.plugins.inventory import BaseInventoryPlugin
|
||||||
from ansible.config.manager import ensure_type
|
from ansible.config.manager import ensure_type
|
||||||
|
from ..module_utils.ansible_tower import make_request
|
||||||
|
|
||||||
# Python 2/3 Compatibility
|
# Python 2/3 Compatibility
|
||||||
try:
|
try:
|
||||||
@@ -121,25 +121,6 @@ class InventoryModule(BaseInventoryPlugin):
|
|||||||
# If the user supplies '@tower_inventory' as path, the plugin will read from environment variables.
|
# If the user supplies '@tower_inventory' as path, the plugin will read from environment variables.
|
||||||
no_config_file_supplied = False
|
no_config_file_supplied = False
|
||||||
|
|
||||||
def make_request(self, request_handler, tower_url):
|
|
||||||
"""Makes the request to given URL, handles errors, returns JSON
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
response = request_handler.get(tower_url)
|
|
||||||
except (ConnectionError, urllib_error.URLError, socket.error, httplib.HTTPException) as e:
|
|
||||||
n_error_msg = 'Connection to remote host failed: {err}'.format(err=to_native(e))
|
|
||||||
# If Tower gives a readable error message, display that message to the user.
|
|
||||||
if callable(getattr(e, 'read', None)):
|
|
||||||
n_error_msg += ' with message: {err_msg}'.format(err_msg=to_native(e.read()))
|
|
||||||
raise AnsibleParserError(n_error_msg)
|
|
||||||
|
|
||||||
# Attempt to parse JSON.
|
|
||||||
try:
|
|
||||||
return json.loads(response.read())
|
|
||||||
except (ValueError, TypeError) as e:
|
|
||||||
# If the JSON parse fails, print the ValueError
|
|
||||||
raise AnsibleParserError('Failed to parse json from host: {err}'.format(err=to_native(e)))
|
|
||||||
|
|
||||||
def verify_file(self, path):
|
def verify_file(self, path):
|
||||||
if path.endswith('@tower_inventory'):
|
if path.endswith('@tower_inventory'):
|
||||||
self.no_config_file_supplied = True
|
self.no_config_file_supplied = True
|
||||||
|
|||||||
@@ -29,8 +29,12 @@
|
|||||||
from __future__ import (absolute_import, division, print_function)
|
from __future__ import (absolute_import, division, print_function)
|
||||||
__metaclass__ = type
|
__metaclass__ = type
|
||||||
|
|
||||||
|
import json
|
||||||
import os
|
import os
|
||||||
import traceback
|
import traceback
|
||||||
|
from ansible.module_utils._text import to_native
|
||||||
|
from ansible.module_utils.urls import Request, urllib_error, ConnectionError, socket, httplib
|
||||||
|
from ansible.errors import AnsibleParserError
|
||||||
|
|
||||||
TOWER_CLI_IMP_ERR = None
|
TOWER_CLI_IMP_ERR = None
|
||||||
try:
|
try:
|
||||||
@@ -46,8 +50,30 @@ except ImportError:
|
|||||||
from ansible.module_utils.basic import AnsibleModule, missing_required_lib
|
from ansible.module_utils.basic import AnsibleModule, missing_required_lib
|
||||||
|
|
||||||
|
|
||||||
|
def make_request(self, request_handler, tower_url):
|
||||||
|
'''
|
||||||
|
Makes the request to given URL, handles errors, returns JSON
|
||||||
|
'''
|
||||||
|
try:
|
||||||
|
response = request_handler.get(tower_url)
|
||||||
|
except (ConnectionError, urllib_error.URLError, socket.error, httplib.HTTPException) as e:
|
||||||
|
n_error_msg = 'Connection to remote host failed: {err}'.format(err=to_native(e))
|
||||||
|
# If Tower gives a readable error message, display that message to the user.
|
||||||
|
if callable(getattr(e, 'read', None)):
|
||||||
|
n_error_msg += ' with message: {err_msg}'.format(err_msg=to_native(e.read()))
|
||||||
|
raise AnsibleParserError(n_error_msg)
|
||||||
|
|
||||||
|
# Attempt to parse JSON.
|
||||||
|
try:
|
||||||
|
return json.loads(response.read())
|
||||||
|
except (ValueError, TypeError) as e:
|
||||||
|
# If the JSON parse fails, print the ValueError
|
||||||
|
raise AnsibleParserError('Failed to parse json from host: {err}'.format(err=to_native(e)))
|
||||||
|
|
||||||
|
|
||||||
def tower_auth_config(module):
|
def tower_auth_config(module):
|
||||||
'''tower_auth_config attempts to load the tower-cli.cfg file
|
'''
|
||||||
|
`tower_auth_config` attempts to load the tower-cli.cfg file
|
||||||
specified from the `tower_config_file` parameter. If found,
|
specified from the `tower_config_file` parameter. If found,
|
||||||
if returns the contents of the file as a dictionary, else
|
if returns the contents of the file as a dictionary, else
|
||||||
it will attempt to fetch values from the module params and
|
it will attempt to fetch values from the module params and
|
||||||
|
|||||||
Reference in New Issue
Block a user