librenms-librenms-agent/snmp/seafile.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

# https://download.seafile.com/published/web-api/v2.1-admin

# user -> libraries (count)
# user -> trash-libraries (count)
# user -> space consumption (count)
# user -> is activated (bool)

# connected_devices (count)
# groups (count)

# Clients -> plattform (count)
# Clients -> version (count)

import requests
import json

# Configfile content example:
# {"url": "https://seafile.mydomain.org",
#  "username": "some_admin_login@mail.address",
#  "password": "password",
#  "account_identifier": "name",
#  "hide_monitoring_account": true
# }

CONFIGFILE='/etc/snmp/seafile.json'
error = 0
error_string = ''
version = 1


def get_data(url_path, data=None, token=None):
        complete_url = "%s/%s" % (url, url_path)
        headers = {'Accept': 'application/json'}
        if token:
            headers['Authorization'] = "Token %s" % token

        try:
            if token:
                r = requests.get(complete_url, data=data, headers=headers)
            else:
                r = requests.post(complete_url, data=data, headers=headers)
            try:
                return r.json()
            except json.decoder.JSONDecodeError:
                return 'no valid json returned - url correct?'
        except requests.exceptions.RequestException as err:
            return str(err)


def get_devices():
    # get all devices
    url_path = 'api/v2.1/admin/devices/'
    return get_data(url_path, token=token)


def get_groups():
    # get all groups
    url_path = 'api/v2.1/admin/groups/'
    return get_data(url_path, token=token)


def get_sysinfo():
    # get all groups
    url_path = 'api/v2.1/admin/sysinfo/'
    return get_data(url_path, token=token)


def get_account_information():
    # get all accounts withs details
    account_list = []
    for account in get_data('api2/accounts/', token=token):

        # get account details
        url_path = 'api2/accounts/%s/' % account['email']
        account_data = get_data(url_path, token=token)

        # get libraries by owner
        url_path = 'api/v2.1/admin/libraries/?owner=%s' % account['email']
        account_data['repos'] = get_data(url_path, token=token)['repos']

        # get deleted libraries by owner
        url_path = 'api/v2.1/admin/trash-libraries/?owner=%s' % account['email']
        account_data['trash_repos'] = get_data(url_path, token=token)['repos']

        account_list.append(account_data)
    return account_list


def resort_devices(device_list):
    data = {}
    platform = {}
    client_version = {}
    for device in device_list:
        # don't list information assigned to monitor account
        if hide_monitoring_account:
            if device['user'] == configfile['username']:
                continue

        if device['platform'] not in platform.keys():
            platform[device['platform']] = 1
        else:
            platform[device['platform']] += 1

        if device['client_version'] not in client_version.keys():
            client_version[device['client_version']] = 1
        else:
            client_version[device['client_version']] += 1

    data['platform'] = []
    for k, v in platform.items():
        data['platform'].append({'os_name': k,
                                 'clients':v})
    data['client_version'] = []
    for k, v in client_version.items():
        data['client_version'].append({'client_version': k,
                                 'clients':v})

    return data


def resort_groups(group_list):
    data = {'count': len(group_list)}
    return data


def resort_accounts(account_list):
    if account_identifier in ['name', 'email']:
        identifier = account_identifier
    else:
        identifier = 'name'

    accepted_key_list = ['is_active', 'usage']

    data = []
    for user_account in account_list:
        # don't list information assigned to monitor account
        if hide_monitoring_account:
            if user_account['email'] == configfile['username']:
                continue

        new_account = {}
        new_account['owner'] = user_account[identifier]
        new_account['repos'] = len(user_account['repos'])
        new_account['trash_repos'] = len(user_account['trash_repos'])

        for k in user_account.keys():
            if k not in accepted_key_list:
                continue
            new_account[k] = user_account[k]
        data.append(new_account)

    return sorted(data, key=lambda k: k['owner'].lower())


# ------------------------ MAIN --------------------------------------------------------
with open(CONFIGFILE, 'r') as json_file:
    try:
        configfile = json.load(json_file)
    except json.decoder.JSONDecodeError as e:
        error = 1
        error_string = "Configfile Error: '%s'" % e

if not error:
    url = configfile['url']
    username = configfile['username']
    password = configfile['password']
    try:
        account_identifier = configfile['account_identifier']
    except KeyError:
        account_identifier = None
    try:
        hide_monitoring_account = configfile['hide_monitoring_account']
    except KeyError:
        hide_monitoring_account = False

    # get token
    login_data = {'username': username, 'password': password}
    ret = get_data('api2/auth-token/', data=login_data)
    if type(ret) != str:
        if 'token' in ret.keys():
            token = ret['token']
        else:
            error = 1
            try:
                error_string = json.dumps(ret)
            except:
                error_string = ret
    else:
        error = 1
        error_string = ret

data = {}
if not error:
    ret= get_account_information()
if not error:
    data['accounts'] = resort_accounts(ret)
    data['devices'] = resort_devices(get_devices()['devices'])
    data['groups'] = resort_groups(get_groups()['groups'])
    data['sysinfo'] = get_sysinfo()

output = {'error': error,
          'errorString': error_string,
          'version': version,
          'data': data
         }

print(json.dumps(output))