collecty/client.py

   1 #!/usr/bin/python
   2 ###############################################################################
   3 #                                                                             #
   4 # collecty - A system statistics collection daemon for IPFire                 #
   5 # Copyright (C) 2012 IPFire development team                                  #
   6 #                                                                             #
   7 # This program is free software: you can redistribute it and/or modify        #
   8 # it under the terms of the GNU General Public License as published by        #
   9 # the Free Software Foundation, either version 3 of the License, or           #
  10 # (at your option) any later version.                                         #
  11 #                                                                             #
  12 # This program is distributed in the hope that it will be useful,             #
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of              #
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the               #
  15 # GNU General Public License for more details.                                #
  16 #                                                                             #
  17 # You should have received a copy of the GNU General Public License           #
  18 # along with this program.  If not, see <http://www.gnu.org/licenses/>.       #
  19 #                                                                             #
  20 ###############################################################################
  21
  22 import daemon
  23
  24 import logging
  25 log = logging.getLogger("collectly.client")
  26
  27 class CollectyClient(object):
  28         def __init__(self, **settings):
  29                 self.collecty = daemon.Collecty(**settings)
  30
  31         @property
  32         def data_sources(self):
  33                 return self.collecty.data_sources
  34
  35         def get_data_source_by_id(self, id):
  36                 for ds in self.data_sources:
  37                         if not ds.id == id:
  38                                 continue
  39
  40                         return ds
  41
  42         def graph(self, id, filename, interval=None, **kwargs):
  43                 ds = self.get_data_source_by_id(id)
  44                 assert ds, "Could not find data source: %s" % id
  45
  46                 ds.graph(filename, interval=interval, **kwargs)