mirror of
https://github.com/ansible-collections/hetzner.hcloud
synced 2024-12-14 14:32:30 +00:00
eb45ae014c
* Fix error handling and add test cases Signed-off-by: Lukas Kämmerling <lukas.kaemmerling@hetzner-cloud.de>
249 lines
7.1 KiB
Python
249 lines
7.1 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright: (c) 2019, Hetzner Cloud GmbH <info@hetzner-cloud.de>
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
|
|
from __future__ import absolute_import, division, print_function
|
|
|
|
__metaclass__ = type
|
|
|
|
DOCUMENTATION = '''
|
|
---
|
|
module: hcloud_subnetwork
|
|
|
|
short_description: Manage cloud subnetworks on the Hetzner Cloud.
|
|
|
|
|
|
description:
|
|
- Create, update and delete cloud subnetworks on the Hetzner Cloud.
|
|
|
|
author:
|
|
- Lukas Kaemmerling (@lkaemmerling)
|
|
|
|
options:
|
|
network:
|
|
description:
|
|
- The ID or Name of the Hetzner Cloud Networks.
|
|
type: str
|
|
required: true
|
|
ip_range:
|
|
description:
|
|
- IP range of the subnetwork.
|
|
type: str
|
|
required: true
|
|
type:
|
|
description:
|
|
- Type of subnetwork.
|
|
type: str
|
|
choices: [ server, cloud, vswitch ]
|
|
required: true
|
|
network_zone:
|
|
description:
|
|
- Name of network zone.
|
|
type: str
|
|
required: true
|
|
vswitch_id:
|
|
description:
|
|
- ID of the vSwitch you want to couple with your Network.
|
|
- Required if type == vswitch
|
|
type: int
|
|
state:
|
|
description:
|
|
- State of the subnetwork.
|
|
default: present
|
|
choices: [ absent, present ]
|
|
type: str
|
|
|
|
requirements:
|
|
- hcloud-python >= 1.10.0
|
|
|
|
extends_documentation_fragment:
|
|
- hetzner.hcloud.hcloud
|
|
|
|
'''
|
|
|
|
EXAMPLES = """
|
|
- name: Create a basic subnetwork
|
|
hcloud_subnetwork:
|
|
network: my-network
|
|
ip_range: 10.0.0.0/16
|
|
network_zone: eu-central
|
|
type: cloud
|
|
state: present
|
|
|
|
- name: Create a basic subnetwork
|
|
hcloud_subnetwork:
|
|
network: my-vswitch-network
|
|
ip_range: 10.0.0.0/24
|
|
network_zone: eu-central
|
|
type: vswitch
|
|
vswitch_id: 123
|
|
state: present
|
|
|
|
- name: Ensure the subnetwork is absent (remove if needed)
|
|
hcloud_subnetwork:
|
|
network: my-network
|
|
ip_range: 10.0.0.0/8
|
|
network_zone: eu-central
|
|
type: cloud
|
|
state: absent
|
|
"""
|
|
|
|
RETURN = """
|
|
hcloud_subnetwork:
|
|
description: One Subnet of a Network
|
|
returned: always
|
|
type: complex
|
|
contains:
|
|
network:
|
|
description: Name of the Network
|
|
type: str
|
|
returned: always
|
|
sample: my-network
|
|
ip_range:
|
|
description: IP range of the Network
|
|
type: str
|
|
returned: always
|
|
sample: 10.0.0.0/8
|
|
type:
|
|
description: Type of subnetwork
|
|
type: str
|
|
returned: always
|
|
sample: server
|
|
network_zone:
|
|
description: Name of network zone
|
|
type: str
|
|
returned: always
|
|
sample: eu-central
|
|
vswitch_id:
|
|
description: ID of the vswitch, null if not type vswitch
|
|
type: int
|
|
returned: always
|
|
sample: 123
|
|
gateway:
|
|
description: Gateway of the subnetwork
|
|
type: str
|
|
returned: always
|
|
sample: 10.0.0.1
|
|
"""
|
|
|
|
from ansible.module_utils.basic import AnsibleModule
|
|
from ansible.module_utils._text import to_native
|
|
from ansible_collections.hetzner.hcloud.plugins.module_utils.hcloud import Hcloud
|
|
|
|
try:
|
|
from hcloud import APIException
|
|
from hcloud.networks.domain import NetworkSubnet
|
|
except ImportError:
|
|
APIException = None
|
|
NetworkSubnet = None
|
|
|
|
|
|
class AnsibleHcloudSubnetwork(Hcloud):
|
|
def __init__(self, module):
|
|
Hcloud.__init__(self, module, "hcloud_subnetwork")
|
|
self.hcloud_network = None
|
|
self.hcloud_subnetwork = None
|
|
|
|
def _prepare_result(self):
|
|
return {
|
|
"network": to_native(self.hcloud_network.name),
|
|
"ip_range": to_native(self.hcloud_subnetwork.ip_range),
|
|
"type": to_native(self.hcloud_subnetwork.type),
|
|
"network_zone": to_native(self.hcloud_subnetwork.network_zone),
|
|
"gateway": self.hcloud_subnetwork.gateway,
|
|
"vswitch_id": self.hcloud_subnetwork.vswitch_id,
|
|
}
|
|
|
|
def _get_network(self):
|
|
try:
|
|
self.hcloud_network = self.client.networks.get_by_name(self.module.params.get("network"))
|
|
self.hcloud_subnetwork = None
|
|
except Exception as e:
|
|
self.module.fail_json(msg=e.message)
|
|
|
|
def _get_subnetwork(self):
|
|
subnet_ip_range = self.module.params.get("ip_range")
|
|
for subnetwork in self.hcloud_network.subnets:
|
|
if subnetwork.ip_range == subnet_ip_range:
|
|
self.hcloud_subnetwork = subnetwork
|
|
|
|
def _create_subnetwork(self):
|
|
params = {
|
|
"ip_range": self.module.params.get("ip_range"),
|
|
"type": self.module.params.get('type'),
|
|
"network_zone": self.module.params.get('network_zone')
|
|
}
|
|
if self.module.params.get('type') == NetworkSubnet.TYPE_VSWITCH:
|
|
self.module.fail_on_missing_params(
|
|
required_params=["vswitch_id"]
|
|
)
|
|
params["vswitch_id"] = self.module.params.get('vswitch_id')
|
|
|
|
if not self.module.check_mode:
|
|
try:
|
|
self.hcloud_network.add_subnet(subnet=NetworkSubnet(**params)).wait_until_finished()
|
|
except Exception as e:
|
|
self.module.fail_json(msg=e.message)
|
|
|
|
self._mark_as_changed()
|
|
self._get_network()
|
|
self._get_subnetwork()
|
|
|
|
def present_subnetwork(self):
|
|
self._get_network()
|
|
self._get_subnetwork()
|
|
if self.hcloud_subnetwork is None:
|
|
self._create_subnetwork()
|
|
|
|
def delete_subnetwork(self):
|
|
self._get_network()
|
|
self._get_subnetwork()
|
|
if self.hcloud_subnetwork is not None and self.hcloud_network is not None:
|
|
if not self.module.check_mode:
|
|
try:
|
|
self.hcloud_network.delete_subnet(self.hcloud_subnetwork).wait_until_finished()
|
|
except Exception as e:
|
|
self.module.fail_json(msg=e.message)
|
|
self._mark_as_changed()
|
|
self.hcloud_subnetwork = None
|
|
|
|
@staticmethod
|
|
def define_module():
|
|
return AnsibleModule(
|
|
argument_spec=dict(
|
|
network={"type": "str", "required": True},
|
|
network_zone={"type": "str", "required": True},
|
|
type={
|
|
"type": "str",
|
|
"required": True,
|
|
"choices": ["server", "cloud", "vswitch"]
|
|
},
|
|
ip_range={"type": "str", "required": True},
|
|
vswitch_id={"type": "int"},
|
|
state={
|
|
"choices": ["absent", "present"],
|
|
"default": "present",
|
|
},
|
|
**Hcloud.base_module_arguments()
|
|
),
|
|
supports_check_mode=True,
|
|
)
|
|
|
|
|
|
def main():
|
|
module = AnsibleHcloudSubnetwork.define_module()
|
|
|
|
hcloud = AnsibleHcloudSubnetwork(module)
|
|
state = module.params["state"]
|
|
if state == "absent":
|
|
hcloud.delete_subnetwork()
|
|
elif state == "present":
|
|
hcloud.present_subnetwork()
|
|
|
|
module.exit_json(**hcloud.get_result())
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|