infomaniak.py

#! /usr/bin/env python3
# -*- coding: utf-8; py-indent-offset: 4 -*-
#
# Author:  Linuxfabrik GmbH, Zurich, Switzerland
# Contact: info (at) linuxfabrik (dot) ch
#          https://www.linuxfabrik.ch/
# License: The Unlicense, see LICENSE file.

# https://github.com/Linuxfabrik/monitoring-plugins/blob/main/CONTRIBUTING.rst

"""Provides functions using the Infomanik REST-API.
"""

__author__ = 'Linuxfabrik GmbH, Zurich/Switzerland'
__version__ = '2024031401'

from . import url


BASE_URL = 'https://api.infomaniak.com'


def get_events(token, insecure=False, no_proxy=False, timeout=8):
    """Get all Infomaniak Events.
    https://developer.infomaniak.com/docs/api/get/2/events
    """
    uri = '{}/2/events?locale=en'.format(BASE_URL)
    success, events = url.fetch_json(
        uri,
        header={'Authorization':'Bearer {}'.format(token)},
        insecure=insecure,
        no_proxy=no_proxy,
        timeout=timeout,
    )
    if not success:
        return (success, events)
    if not events:
        return (False, 'There was no result from {}.'.format(uri))
    if events.get('result') != 'success':
        return (False, events.get('error').get('description'))
    return (True, events)


def get_swiss_backup_products(account_id, token, insecure=False, no_proxy=False, timeout=8):
    """Get all Infomaniak Swiss Backup products.
    https://developer.infomaniak.com/docs/api/get/1/swiss_backups
    """
    uri = '{}/1/swiss_backups?account_id={}'.format(BASE_URL, account_id)
    success, products = url.fetch_json(
        uri,
        header={'Authorization':'Bearer {}'.format(token)},
        insecure=insecure,
        no_proxy=no_proxy,
        timeout=timeout,
    )
    if not success:
        return (success, products)
    if not products:
        return (False, 'There was no result from {}.'.format(uri))
    if products.get('result') != 'success':
        return (False, products.get('error').get('description'))
    return (True, products)


def get_swiss_backup_slots(account_id, token, insecure=False, no_proxy=False, timeout=8):
    """Get all devices / slots for each Infomaniak Swiss Backup product.
    https://developer.infomaniak.com/docs/api/get/1/swiss_backups/%7Bswiss_backup_id%7D/slots/%7Bslot_id%7D
    """
    success, products = get_swiss_backup_products(account_id, token)
    if not success:
        return (success, products)
    slots = []
    for product in products.get('data', {}):
        uri = '{}/1/swiss_backups/{}/slots'.format(
            BASE_URL,
            product.get('id'),
        )
        success, slot = url.fetch_json(
            uri,
            header={'Authorization':'Bearer {}'.format(token)},
            insecure=insecure,
            no_proxy=no_proxy,
            timeout=timeout,
        )
        if not success:
            return (success, slot)
        if not slot:
            return (False, 'There was no result from {}.'.format(uri))
        if slot.get('result') != 'success':
            return (False, slot.get('error').get('description'))
        # append some information from the parent, too:
        slot['product_customer_name'] = product.get('customer_name')
        slot['product_tags'] = product.get('tags')
        slots.append(slot)
    return (True, slots)