python.d/ipfs.chart.py

   1 # -*- coding: utf-8 -*-
   2 # Description: IPFS netdata python.d module
   3 # Authors: Pawel Krupa (paulfantom), davidak
   4
   5 from base import UrlService
   6 import json
   7
   8 # default module values (can be overridden per job in `config`)
   9 # update_every = 2
  10 priority = 60000
  11 retries = 60
  12
  13 # default job configuration (overridden by python.d.plugin)
  14 # config = {'local': {
  15 #     'update_every': update_every,
  16 #     'retries': retries,
  17 #     'priority': priority,
  18 #     'url': 'http://localhost:5001'
  19 # }}
  20
  21 # charts order (can be overridden if you want less charts, or different order)
  22 ORDER = ['bandwidth', 'peers']
  23
  24 CHARTS = {
  25     'bandwidth': {
  26         'options': [None, 'IPFS Bandwidth', 'kbits/s', 'Bandwidth', 'ipfs.bandwidth', 'line'],
  27         'lines': [
  28             ["in", None, "absolute", 8, 1000],
  29             ["out", None, "absolute", -8, 1000]
  30         ]},
  31     'peers': {
  32         'options': [None, 'IPFS Peers', 'peers', 'Peers', 'ipfs.peers', 'line'],
  33         'lines': [
  34             ["peers", None, 'absolute']
  35         ]}
  36 }
  37
  38
  39 class Service(UrlService):
  40     def __init__(self, configuration=None, name=None):
  41         UrlService.__init__(self, configuration=configuration, name=name)
  42         try:
  43             self.baseurl = str(self.configuration['url'])
  44         except (KeyError, TypeError):
  45             self.baseurl = "http://localhost:5001"
  46         self.order = ORDER
  47         self.definitions = CHARTS
  48
  49     def _get_bandwidth(self):
  50         """
  51         Format data received from http request
  52         :return: int, int
  53         """
  54         self.url = self.baseurl + "/api/v0/stats/bw"
  55         try:
  56             raw = self._get_raw_data()
  57         except AttributeError:
  58             return None
  59
  60         try:
  61             parsed = json.loads(raw)
  62             bw_in = int(parsed['RateIn'])
  63             bw_out = int(parsed['RateOut'])
  64         except:
  65             return None
  66
  67         return bw_in, bw_out
  68
  69     def _get_peers(self):
  70         """
  71         Format data received from http request
  72         :return: int
  73         """
  74         self.url = self.baseurl + "/api/v0/swarm/peers"
  75         try:
  76             raw = self._get_raw_data()
  77         except AttributeError:
  78             return None
  79
  80         try:
  81             parsed = json.loads(raw)
  82             peers = len(parsed['Strings'])
  83         except:
  84             return None
  85
  86         return peers
  87
  88     def _get_data(self):
  89         """
  90         Get data from API
  91         :return: dict
  92         """
  93         try:
  94             peers = self._get_peers()
  95             bandwidth_in, bandwidth_out = self._get_bandwidth()
  96         except:
  97             return None
  98         data = {}
  99         if peers is not None:
 100             data['peers'] = peers
 101         if bandwidth_in is not None and bandwidth_out is not None:
 102             data['in'] = bandwidth_in
 103             data['out'] = bandwidth_out
 104
 105         if len(data) == 0:
 106             return None
 107         return data