markuman.hetzner_dns/plugins/module_utils/helper.py

#!/usr/bin/python
# -*- coding: utf-8 -*-

# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
# Part of ansible markuman.hetzner_dns collection

import os
import requests
import json
from ansible.errors import AnsibleError

def ZoneInfo(zones, name):
    zone_id = None
    zone_info = {}
    for item in zones.json()['zones']:
        if item.get('name') == name:
            zone_id = item.get('id')
            zone_info = item

    return zone_id, zone_info

class HetznerAPIHandler:
    def __init__(self, kwargs):
        self.TOKEN = kwargs.get('api_token') or os.environ.get('HETZNER_DNS_TOKEN')
        if self.TOKEN is None:
            raise AnsibleError('Unable to continue. No Hetzner DNS API Token is given.')

    def get_zone_info(self):
        try:
            r = requests.get(
                url="https://dns.hetzner.com/api/v1/zones",
                headers={
                    "Auth-API-Token": self.TOKEN,
                },
            )

            if r.status_code == 200:
                return r
            else:
                raise AnsibleError('Unknown Error')
        except requests.exceptions.RequestException:
            raise AnsibleError('HTTP Request failed')

    def get_record_info(self, zone_id):
        try:
            r = requests.get(
                url="https://dns.hetzner.com/api/v1/records",
                params={
                    "zone_id": zone_id,
                },
                headers={
                    "Auth-API-Token": self.TOKEN,
                },
            )
            if r.status_code == 200:
                return r
            else:
                raise AnsibleError('Unknown Error')
        except requests.exceptions.RequestException:
            raise AnsibleError('HTTP Request failed')


    def create_record(self, record):
        try:
            r = requests.post(
                url="https://dns.hetzner.com/api/v1/records",
                headers={
                    "Content-Type": "application/json",
                    "Auth-API-Token": self.TOKEN,
                },
                data=json.dumps(record)
            )
            if r.status_code == 200:
                return r
            else:
                raise AnsibleError('Unknown Error')
        except requests.exceptions.RequestException:
            raise AnsibleError('HTTP Request failed')


    def update_record(self, record, record_id):
        try:
            r = requests.put(
                url="https://dns.hetzner.com/api/v1/records/{RecordID}".format(RecordID=record_id),
                headers={
                    "Content-Type": "application/json",
                    "Auth-API-Token": self.TOKEN,
                },
                data=json.dumps(record)
            )
            if r.status_code == 200:
                return r
            else:
                raise AnsibleError('Unknown Error')
        except requests.exceptions.RequestException:
            raise AnsibleError('HTTP Request failed')

    def delete_record(self, record_id):
        try:
            r = requests.delete(
                url="https://dns.hetzner.com/api/v1/records/{RecordID}".format(RecordID=record_id),
                headers={
                    "Auth-API-Token": self.TOKEN,
                },
            )
            if r.status_code == 200:
                return r
            else:
                raise AnsibleError('Unknown Error')
        except requests.exceptions.RequestException:
            raise AnsibleError('HTTP Request failed')