mirror of
https://github.com/phfroidmont/self-hosting.git
synced 2025-12-25 21:57:00 +01:00
110 lines
3.4 KiB
Python
Executable file
110 lines
3.4 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
Generate an inventory of servers from scaleway which is
|
|
suitable for use as an ansible dynamic inventory
|
|
Right now, only the group 'cci-customer' is exported
|
|
"""
|
|
|
|
import configparser
|
|
import json
|
|
import os
|
|
from typing import Dict, Any
|
|
|
|
from scaleway.apis import ComputeAPI
|
|
|
|
|
|
class SCWInventory(object):
|
|
"""
|
|
The inventory class which calls out to scaleway and digests
|
|
The returned data, making it usable by ansible as an inventory
|
|
"""
|
|
|
|
response: Dict[str, Any]
|
|
|
|
def __init__(self):
|
|
self.inventory = None
|
|
self.auth_token = None
|
|
self.environment = None
|
|
self.response = {
|
|
'_meta': {
|
|
'hostvars': {
|
|
}
|
|
}
|
|
}
|
|
|
|
def parse_config(self, creds_file='scw.ini'):
|
|
"""
|
|
Parse the ini file to get the auth token
|
|
"""
|
|
config = configparser.ConfigParser()
|
|
config.read(creds_file)
|
|
with open(os.path.expanduser(config['credentials']['token_file']), 'r') as content_file:
|
|
self.auth_token = content_file.read().replace('\n', '')
|
|
self.environment = config['config']['environment']
|
|
|
|
def get_servers(self):
|
|
"""
|
|
Query scaleway api and pull down a list of servers
|
|
"""
|
|
self.parse_config()
|
|
api_par1 = ComputeAPI(auth_token=self.auth_token, region='par1')
|
|
api_ams1 = ComputeAPI(auth_token=self.auth_token, region='ams1')
|
|
result_par1 = api_par1.query().servers.get()
|
|
result_ams1 = api_ams1.query().servers.get()
|
|
self.inventory = [
|
|
[i['name'], i['public_ip'], i['tags'], i['private_ip']] for i in
|
|
result_par1['servers'] + result_ams1['servers']
|
|
]
|
|
for host, ip_info, tags, private_ip in self.inventory:
|
|
host_vars = {
|
|
'private_ip': private_ip,
|
|
'ansible_python_interpreter': '/usr/bin/python3'
|
|
}
|
|
if ip_info:
|
|
host_vars['ansible_host'] = ip_info['address']
|
|
host_vars['public_ip'] = ip_info['address']
|
|
else:
|
|
host_vars['ansible_host'] = private_ip
|
|
|
|
self.response['_meta']['hostvars'][host] = host_vars
|
|
if tags:
|
|
for tag in tags:
|
|
self._add_to_response(
|
|
tag,
|
|
host
|
|
)
|
|
|
|
for host, variables in self.response['_meta']['hostvars'].items():
|
|
if host != 'proxy1':
|
|
variables['ansible_ssh_common_args'] = '-o ProxyCommand="ssh -W %h:%p -q root@' + \
|
|
self.response['_meta']['hostvars']['proxy1']['public_ip'] \
|
|
+ ' -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null"'
|
|
|
|
def _add_to_response(self, group, hostname):
|
|
"""
|
|
Add a host to a group within the response
|
|
"""
|
|
if group not in self.response:
|
|
self.response[group] = list()
|
|
if group in self.response:
|
|
self.response[group].append(hostname)
|
|
|
|
def print_inventory(self):
|
|
"""
|
|
Simply display the collected inventory
|
|
"""
|
|
print(json.dumps(self.response))
|
|
|
|
|
|
def main():
|
|
"""
|
|
Run the program starting here
|
|
"""
|
|
inventory = SCWInventory()
|
|
inventory.get_servers()
|
|
inventory.print_inventory()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|