Merge pull request #5 from myoung34/datadog

Add datadog statsd emitter

README.md

@@ -18,6 +18,7 @@ The Tilt supports writing to a google doc which you could use with something lik
 
 * Webhooks
 * InfluxDB
+* Datadog (dogstatsd)
 
 
 ## Usage ##
@@ -40,6 +41,11 @@ port = 80
 database = tilty
 gravity_payload_template = {"measurement": "gravity", "tags": {"color": "{{ color }}"}, "fields": {"value": {{ gravity }}}}
 temperature_payload_template = {"measurement": "temperature", "tags": {"color": "{{ color }}"}, "fields": {"value": {{ temp }}}}
+
+[datadog]
+# Note: make sure that the dd agent has DD_DOGSTATSD_NON_LOCAL_TRAFFIC=true
+host = statsdhost.corp.com
+port = 8125
 EOF
 ```
 

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "Tilty"
-version = "0.0.1"
+version = "0.3.0"
 description = "A pluggable system to receive and transmit bluetooth events from the Tilt Hydrometer"
 authors = ["Marcus Young <[email protected]>"]
 license = "MIT"
@@ -12,6 +12,7 @@ pybluez = "^0.22.0"
 requests = "^2.22"
 jinja2 = "^2.11.1"
 influxdb = "^5.2.3"
+datadog = "^0.34.1"
 
 [tool.poetry.dev-dependencies]
 flake8 = "^3.7"

requirements.txt

@@ -4,6 +4,8 @@ certifi==2019.11.28
 chardet==3.0.4
 Click==7.0
 coverage==5.0
+datadog==0.34.1
+decorator==4.4.1
 entrypoints==0.3
 filelock==3.0.12
 flake8==3.7.9

setup.py

@@ -13,10 +13,11 @@
     long_description=long_description,
     long_description_content_type="text/markdown",
     py_modules=['tilty', 'blescan'],
-    version='0.2.2',
+    version='0.3.0',
     packages=find_packages(exclude=['tests*']),
     install_requires=[
         'Click',
+        'datadog',
         'influxdb',
         'Jinja2',
         'pybluez',

tests/mock_config_parser.py

@@ -18,6 +18,11 @@ def __getitem__(self, key):
                 'gravity_payload_template': 'gravity,color={{ color }} value={{ gravity }} {{timestamp}}',  # noqa
                 'temperature_payload_template': 'temperature,scale=fahrenheit,color={{ color }} value={{ temp }} {{timestamp}}',  # noqa
             }
+        if self.section == 'datadog':
+            return {
+                'host': 'http://api.datadog.com',
+                'port': '8120',
+            }
         return None
 
     def has_section(self, *args, **kwargs):

tests/test_datadog.py

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+from unittest import mock
+
+from tilty.emitters import datadog
+
+
+@mock.patch('tilty.emitters.datadog.statsd')
+@mock.patch('tilty.emitters.datadog.initialize')
+def test_datadog(
+    mock_statsd_init,
+    mock_statsd_client,
+):
+    config = {
+        'host': 'http://statsd.google.com',
+        'port': '8130',
+        'temperature': '55',
+        'gravity': '1054',
+        'color': 'black',
+    }
+    datadog.Datadog(config=config).emit()
+    mock_statsd_init.mock_calls == [
+        mock.call(statsd_host='http://statsd.google.com', statsd_port='8130')
+    ]
+    assert mock_statsd_client.mock_calls == [
+        mock.call.gauge('tilty.temperature', '55', tags=['color:black']),
+        mock.call.gauge('tilty.gravity', '1054', tags=['color:black']),
+    ]

tests/test_tilty.py

@@ -56,3 +56,18 @@ def test_scan_for_tilt_data_parse_influxdb(
         ),
         mock.call().emit()
     ]
+
+
+@mock.patch('tilty.emitters.datadog.Datadog')
+def test_scan_for_tilt_data_parse_datadog(
+    mock_dd,
+):
+    config = MockConfigParser('datadog')
+    tilty.emit(
+        config,
+        {'color': 'black', 'gravity': 1, 'temp': 32, 'timestamp': 155558888}
+    )
+    assert mock_dd.mock_calls == [
+        mock.call(config={'host': 'http://api.datadog.com', 'port': '8120', 'gravity': 1, 'temperature': 32, 'color': 'black'}),  # noqa
+        mock.call().emit()
+    ]

tilty/emitters/datadog.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+""" DataDog emitter """
+import logging
+
+from datadog import initialize, statsd
+
+from tilty.common import safe_get_key
+
+LOGGER = logging.getLogger()
+
+
+class Datadog:  # pylint: disable=too-few-public-methods
+    """ Class to represent the actual device """
+    def __init__(self, config):
+        """ Initializer
+
+        Args:
+            config: (dict) represents the configuration for the emitter
+        """
+        self.temperature = config['temperature']
+        self.gravity = config['gravity']
+        self.color = config['color']
+        options = {
+            'statsd_host': config['host'],
+            'statsd_port': safe_get_key(config, 'port', 8125),
+        }
+        initialize(**options)
+
+    def emit(self, **kwargs):  # pylint: disable=no-self-use,unused-argument
+        """ Initializer
+
+        Args:
+        """
+        LOGGER.info('[datadog] posting temperature data')
+        statsd.gauge(
+            'tilty.temperature',
+            self.temperature,
+            tags=[f"color:{self.color}"]
+        )
+        LOGGER.info('[datadog] posting gravity data')
+        statsd.gauge(
+            'tilty.gravity',
+            self.gravity,
+            tags=[f"color:{self.color}"]
+        )

tilty/tilty.py

@@ -4,7 +4,7 @@
 
 from jinja2 import Template
 
-from tilty.emitters import influxdb, webhook
+from tilty.emitters import datadog, influxdb, webhook
 
 
 def emit(config, tilt_data):
@@ -65,3 +65,18 @@ def emit(config, tilt_data):
         }
         _influxdb = influxdb.InfluxDB(config=_config)
         _influxdb.emit()
+
+    # <start config sample>
+    # [datadog]
+    # host = 'host'
+    # port = 'port'
+    if config.has_section('datadog'):
+        _config = {
+            'host': config['datadog']['host'],
+            'port': config['datadog']['port'],
+            'gravity': tilt_data['gravity'],
+            'temperature': tilt_data['temp'],
+            'color': tilt_data['color'],
+        }
+        _datadog = datadog.Datadog(config=_config)
+        _datadog.emit()