155 lines
3.6 KiB
Python
155 lines
3.6 KiB
Python
import json
|
|
|
|
from ansible.module_utils.common.text.converters import to_text
|
|
from ansible.module_utils.urls import open_url
|
|
from ansible.plugins.inventory import BaseInventoryPlugin
|
|
|
|
|
|
DOCUMENTATION = '''
|
|
---
|
|
module: centurion
|
|
plugin_type: inventory
|
|
short_description: Centurion ERP Inventory
|
|
description:
|
|
- "An Inventory Plugin to fetch inventory from Centurion ERP."
|
|
options:
|
|
plugin:
|
|
description: token that ensures this is a source file for the 'centurion' plugin.
|
|
required: True
|
|
choices: ['nofusscomputing.centurion.centurion']
|
|
api_endpoint:
|
|
description: Endpoint of the Centurion API
|
|
required: true
|
|
env:
|
|
- name: CENTURION_API
|
|
token:
|
|
required: false
|
|
description:
|
|
- Centurion API token to be able to read against Centurion.
|
|
env:
|
|
- name: CENTURION_TOKEN
|
|
validate_certs:
|
|
description:
|
|
- Allows skipping of validation of SSL certificates. Set to C(false) to disable certificate validation.
|
|
default: true
|
|
type: boolean
|
|
env:
|
|
- name: VALIDATE_CENTURION_CERTS
|
|
author:
|
|
- jon @jon_nfc
|
|
'''
|
|
|
|
|
|
|
|
EXAMPLES = """
|
|
# inventory.yml file in YAML format
|
|
|
|
plugin: nofusscomputing.centurion.centurion
|
|
|
|
api_endpoint: http://localhost:8000
|
|
token: <token value here>
|
|
validate_certs: false
|
|
|
|
|
|
# Example Ansible Tower credential Input Configuration:
|
|
|
|
fields:
|
|
- id: CENTURION_API
|
|
type: string
|
|
label: Centurion Host URL
|
|
- id: CENTURION_TOKEN
|
|
type: string
|
|
label: Centurion API Token
|
|
secret: true
|
|
- id: VALIDATE_CENTURION_CERTS
|
|
label: Validate Centurion SSL Certificate
|
|
type: boolean
|
|
required:
|
|
- CENTURION_API
|
|
- CENTURION_TOKEN
|
|
|
|
# Example Ansible Tower credential Injector Configuration:
|
|
|
|
env:
|
|
CENTURION_API: '{{ CENTURION_API }}'
|
|
CENTURION_TOKEN: '{{ CENTURION_TOKEN }}'
|
|
VALIDATE_CENTURION_CERTS: '{{ VALIDATE_CENTURION_CERTS | default(true) }}'
|
|
|
|
"""
|
|
|
|
|
|
|
|
class InventoryModule(BaseInventoryPlugin):
|
|
|
|
NAME = 'nofusscomputing.centurion.centurion'
|
|
|
|
|
|
def verify_file(self, path):
|
|
|
|
return True
|
|
|
|
|
|
def fetch_devices(self) -> dict:
|
|
|
|
self.display.v("Fetching Devices ")
|
|
|
|
response = open_url(
|
|
url = f'{self.api_endpoint}/api/device/',
|
|
headers = self.headers,
|
|
validate_certs = self.validate_certs,
|
|
)
|
|
|
|
devices = json.loads(to_text(response.read()))['results']
|
|
|
|
|
|
return devices
|
|
|
|
|
|
def fetch_device_config(self, url: str) -> dict:
|
|
|
|
|
|
self.display.vv(f"Fetching Device Config {url} ")
|
|
|
|
response = open_url(
|
|
url = url,
|
|
headers = self.headers,
|
|
validate_certs = self.validate_certs,
|
|
)
|
|
|
|
config = json.loads(to_text(response.read()))
|
|
|
|
return config
|
|
|
|
|
|
def parse(self, inventory, loader, path, cache=True):
|
|
super().parse(inventory, loader, path)
|
|
|
|
self._read_config_data(path=path)
|
|
|
|
|
|
self.api_endpoint = self.get_option("api_endpoint").strip("/")
|
|
self.token = self.get_option("token")
|
|
self.validate_certs = self.get_option("validate_certs")
|
|
|
|
self.headers = {
|
|
'Authorization': f'Token {self.token}'
|
|
}
|
|
|
|
devices = self.fetch_devices()
|
|
|
|
self.display.v(f"Parsing returned devices")
|
|
|
|
for device in devices:
|
|
|
|
|
|
self.display.vv(f"Adding device {device['name']} to inventory")
|
|
|
|
|
|
self.inventory.add_host(host=device['name'])
|
|
|
|
config = self.fetch_device_config(device['config'])
|
|
|
|
for key, val in config.items():
|
|
|
|
self.inventory.set_variable(device['name'], key, val)
|