Add new endpoints (alarms and allmetrics)

Author: fabaff

netdata/__init__.py

@@ -13,34 +13,43 @@
 from . import exceptions
 
 _LOGGER = logging.getLogger(__name__)
-_RESOURCE = 'http://{host}:{port}/api/v{api}/data?chart={resource}&{realtime}'
-_REALTIME = 'before=0&after=-1&options=seconds'
+_INSTANCE = 'http://{host}:{port}/api/v{api}/'
+_DATA_ENDPOINT = 'data?chart={resource}&before=0&after=-1&options=seconds'
+_ALARMS_ENDPOINT = 'alarms?all&format=json'
+_ALL_METRIC_ENDPOINT = ('allmetrics?format=json&help=no&types=no&'
+                        'timestamps=yes&names=yes&data=average')
 
 API_VERSION = 1
 
 
 class Netdata(object):
     """A class for handling connections with a Netdata instance."""
 
-    def __init__(self, host, resource, loop, session, port=19999):
+    def __init__(self, host, loop, session, port=19999, data=None):
         """Initialize the connection to the Netdata instance."""
         self._loop = loop
         self._session = session
         self.host = host
         self.port = port
-        self.resource = resource
-        self.values = None
+        self.values = self.alarms = self.metrics = None
+        self.base_url = _INSTANCE.format(host=host, port=port, api=API_VERSION)
+        if data is None:
+            self.endpoint = _ALL_METRIC_ENDPOINT
+        if data == 'alarms':
+            self.endpoint = _ALARMS_ENDPOINT
+        if data == 'data':
+            self.endpoint = _DATA_ENDPOINT
 
-    async def async_get_data(self):
-        url = _RESOURCE.format(
-            host=self.host, port=self.port, api=API_VERSION,
-            resource=self.resource, realtime=_REALTIME)
+    async def get_data(self, resource):
+        """Get detail for a resource from the data endpoint."""
+        url = '{}{}'.format(
+            self.base_url, self.endpoint.format(resource=resource))
 
         try:
             with async_timeout.timeout(5, loop=self._loop):
                 response = await self._session.get(url)
 
-            _LOGGER.debug(
+            _LOGGER.info(
                 "Response from Netdata: %s", response.status)
             data = await response.json()
             _LOGGER.debug(data)
@@ -50,3 +59,39 @@ async def async_get_data(self):
         except (asyncio.TimeoutError, aiohttp.ClientError, socket.gaierror):
             _LOGGER.error("Can not load data from Netdata")
             raise exceptions.NetdataConnectionError()
+
+    async def get_alarms(self):
+        """Get alarms for a Netdata instance."""
+        url = '{}{}'.format(self.base_url, self.endpoint)
+
+        try:
+            with async_timeout.timeout(5, loop=self._loop):
+                response = await self._session.get(url)
+
+            _LOGGER.debug(
+                "Response from Netdata: %s", response.status)
+            data = await response.text()
+            _LOGGER.debug(data)
+            self.alarms = data
+
+        except (asyncio.TimeoutError, aiohttp.ClientError, socket.gaierror):
+            _LOGGER.error("Can not load data from Netdata")
+            raise exceptions.NetdataConnectionError()
+
+    async def get_allmetrics(self):
+        """Get all available metrics from a Netdata instance."""
+        url = '{}{}'.format(self.base_url, self.endpoint)
+
+        try:
+            with async_timeout.timeout(5, loop=self._loop):
+                response = await self._session.get(url)
+
+            _LOGGER.debug(
+                "Response from Netdata: %s", response.status)
+            data = await response.json()
+            _LOGGER.debug(data)
+            self.metrics = data
+
+        except (asyncio.TimeoutError, aiohttp.ClientError, socket.gaierror):
+            _LOGGER.error("Can not load data from Netdata")
+            raise exceptions.NetdataConnectionError()