summaryrefslogtreecommitdiff
path: root/roles/cloud/install/filter_plugins/hetzner.py
blob: 0aa19c175321cd6ca957e5680a9dfdc00325c095 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
from __future__ import (absolute_import, division, print_function)
__metaclass__ = type


from ansible import errors


def hetzner_install_image_name_glob(data):
    try:
        if data[0] == 'debian' and data[1] != 'buster':
            return "debian-*-%s-amd64-base.tar.gz" % (data[1])
        return "%s-*-%s-64-minimal.tar.gz" % (data[0], data[1])
    except Exception as e:
        raise errors.AnsibleFilterError("hetzner_install_image_name(): %s" % str(e))


def hroot_extract_ssh_key_fingerprints(data):
    try:
        return [k['key']['fingerprint'] for k in data]
    except Exception as e:
        raise errors.AnsibleFilterError("hroot_ssh_key_fingerprints(): %s" % str(e))


def hroot_extract_serverip(data, server_name):
    try:
        ips = [s['server']['server_ip'] for s in data if s['server']['server_name'] == server_name]
        if len(ips):
            return ips[0]
        raise errors.AnsibleFilterError("hroot_extract_serverip(): server %s not found" % server_name)
    except Exception as e:
        raise errors.AnsibleFilterError("hroot_extract_serverip(): %s" % str(e))


class FilterModule(object):

    ''' Ansible math jinja2 filters '''

    def filters(self):
        return {
            'hetzner_install_image_name_glob': hetzner_install_image_name_glob,
            'hroot_extract_ssh_key_fingerprints': hroot_extract_ssh_key_fingerprints,
            'hroot_extract_serverip': hroot_extract_serverip,
        }