Add support for adding mac address into payloads

README.md

@@ -39,6 +39,8 @@ sleep_interval = 1
 url = http://www.foo.com
 headers = {"Content-Type": "application/json"}
 payload_template = {"color": "{{ color }}", "gravity": {{ gravity }}, "temp": {{ temp }}, "timestamp": "{{ timestamp }}"}
+# optional payload to include mac address
+# payload_template = {"color": "{{ color }}", "gravity": {{ gravity }}, "mac": {{ mac }}, "temp": {{ temp }}, "timestamp": "{{ timestamp }}"}
 method = POST
 
 # Brewstat.us example
@@ -54,6 +56,10 @@ 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 }}}}
+# optional payload to include mac address
+# gravity_payload_template = {"measurement": "gravity", "tags": {"color": "{{ color }}", "mac": "{{ mac }}"}, "fields": {"value": {{ gravity }}}}
+# temperature_payload_template = {"measurement": "temperature", "tags": {"color": "{{ color }}", "mac": "{{ mac }}"}, "fields": {"value": {{ temp }}}}
+# `% curl  -s -G 'http://influxdb.service.consul:8086/query?pretty=true' --data-urlencode "db=tilty" --data-urlencode 'q=SELECT "value", "mac", "color" FROM "autogen"."gravity"' | jq '.results[].series[].values[0]'`
 
 [datadog]
 # Note: make sure that the dd agent has DD_DOGSTATSD_NON_LOCAL_TRAFFIC=true

tests/mock_config_parser_mac.py

@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+class MockConfigParserMac:
+    def __init__(self, section):
+        self.section = section
+
+    def __getitem__(self, key):
+        if self.section == 'webhook':
+            return {
+                'url': 'http://www.google.com',
+                'headers': '{"Content-Type": "application/json"}',
+                'payload_template': '{"color": "{{ color }}", "gravity": {{ gravity }}, "temp": {{ temp }}, "mac": "{{ mac }}", "timestamp": "{{ timestamp }}"}',  # noqa
+                'method': 'GET'
+            }
+        if self.section == 'influxdb':
+            return {
+                'url': 'http://www.google.com',
+                'database': 'foo',
+                'gravity_payload_template': 'gravity,mac={{ mac }} color={{ color }} value={{ gravity }} {{timestamp}}',  # noqa
+                'temperature_payload_template': 'temperature,scale=fahrenheit,mac={{ mac }} 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):
+        return self.section in args

tests/test_cli.py

@@ -53,7 +53,7 @@ def test_cli_no_params_no_data(
     assert result.exit_code == 0
     assert result.output == 'Scanning for Tilt data...\n' # noqa
 
-@mock.patch('tilty.blescan.get_events', return_value=[{'uuid': 'a495bb30c5b14b44b5121370f02d74de', 'major': 60, 'minor': 1053}]) # noqa
+@mock.patch('tilty.blescan.get_events', return_value=[{'mac': '00:0a:95:9d:68:16', 'uuid': 'a495bb30c5b14b44b5121370f02d74de', 'major': 60, 'minor': 1053}]) # noqa
 @mock.patch('tilty.blescan.hci_le_set_scan_parameters') # noqa
 @mock.patch('tilty.blescan.hci_enable_le_scan') # noqa
 def test_cli_no_params_success(
@@ -64,4 +64,4 @@ def test_cli_no_params_success(
     runner = CliRunner()
     result = runner.invoke(cli.run, [])
     assert result.exit_code == 0
-    assert "Scanning for Tilt data...\n{'color': 'Black', 'gravity': 1.053, 'temp': 60, 'timestamp'" in result.output# noqa
+    assert "Scanning for Tilt data...\n{'color': 'Black', 'gravity': 1.053, 'temp': 60, 'mac': '00:0a:95:9d:68:16', 'timestamp'" in result.output# noqa

tests/test_datadog.py

@@ -16,12 +16,21 @@ def test_datadog(
         'temperature': '55',
         'gravity': '1054',
         'color': 'black',
+        'mac': '00:0a:95:9d:68:16',
     }
     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']),
+        mock.call.gauge(
+            'tilty.temperature',
+            '55',
+            tags=['color:black', 'mac:00:0a:95:9d:68:16']
+        ),
+        mock.call.gauge(
+            'tilty.gravity',
+            '1054',
+            tags=['color:black', 'mac:00:0a:95:9d:68:16']
+        ),
     ]

tests/test_tilty.py

@@ -2,6 +2,7 @@
 from unittest import mock
 
 from mock_config_parser import MockConfigParser
+from mock_config_parser_mac import MockConfigParserMac
 from tilty import tilt_device, tilty
 
 
@@ -21,7 +22,13 @@ def test_scan_for_tilt_data_parse_webhook(
     config = MockConfigParser('webhook')
     tilty.emit(
         config,
-        {'color': 'black', 'gravity': 1, 'temp': 32, 'timestamp': 155558888}
+        {
+            'color': 'black',
+            'gravity': 1,
+            'temp': 32,
+            'mac': '00:0a:95:9d:68:16',
+            'timestamp': 155558888
+        }
     )
     assert mock_webhook.mock_calls == [
         mock.call(
@@ -36,14 +43,48 @@ def test_scan_for_tilt_data_parse_webhook(
     ]
 
 
+@mock.patch('tilty.emitters.webhook.Webhook')
+def test_scan_for_tilt_data_parse_webhook_with_mac(
+    mock_webhook,
+):
+    config = MockConfigParserMac('webhook')
+    tilty.emit(
+        config,
+        {
+            'color': 'black',
+            'gravity': 1,
+            'temp': 32,
+            'mac': '00:0a:95:9d:68:16',
+            'timestamp': 155558888
+        }
+    )
+    assert mock_webhook.mock_calls == [
+        mock.call(
+            config={
+                'url': 'http://www.google.com',
+                'headers': {'Content-Type': 'application/json'},
+                'method': 'GET',
+                'payload': {'color': 'black', 'gravity': 1, 'temp': 32, 'mac': '00:0a:95:9d:68:16', 'timestamp': '155558888'}  # noqa
+            }
+        ),
+        mock.call().emit()
+    ]
+
+
 @mock.patch('tilty.emitters.influxdb.InfluxDB')
 def test_scan_for_tilt_data_parse_influxdb(
     mock_influxdb,
 ):
     config = MockConfigParser('influxdb')
     tilty.emit(
         config,
-        {'color': 'black', 'gravity': 1, 'temp': 32, 'timestamp': 155558888}
+        {
+            'color': 'black',
+            'gravity': 1,
+            'temp': 32,
+            'mac': '00:0a:95:9d:68:16',
+            'timestamp': 155558888
+        }
     )
     assert mock_influxdb.mock_calls == [
         mock.call(
@@ -58,16 +99,50 @@ def test_scan_for_tilt_data_parse_influxdb(
     ]
 
 
+@mock.patch('tilty.emitters.influxdb.InfluxDB')
+def test_scan_for_tilt_data_parse_influxdb_with_mac(
+    mock_influxdb,
+):
+    config = MockConfigParserMac('influxdb')
+    tilty.emit(
+        config,
+        {
+            'color': 'black',
+            'gravity': 1,
+            'temp': 32,
+            'mac': '00:0a:95:9d:68:16',
+            'timestamp': 155558888
+        }
+    )
+    assert mock_influxdb.mock_calls == [
+        mock.call(
+            config={
+                'url': 'http://www.google.com',
+                'database': 'foo',
+                'temperature_payload': 'temperature,scale=fahrenheit,mac=00:0a:95:9d:68:16 color=black value=32 155558888',  # noqa
+                'gravity_payload': 'gravity,mac=00:0a:95:9d:68:16 color=black value=1 155558888'  # noqa
+            }
+        ),
+        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}
+        {
+            'color': 'black',
+            'gravity': 1,
+            'temp': 32,
+            'mac': '00:0a:95:9d:68:16',
+            '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(config={'host': 'http://api.datadog.com', 'port': '8120', 'gravity': 1, 'mac': '00:0a:95:9d:68:16', 'temperature': 32, 'color': 'black'}),  # noqa
         mock.call().emit()
     ]

tilty/emitters/datadog.py

@@ -20,6 +20,7 @@ def __init__(self, config):
         self.temperature = config['temperature']
         self.gravity = config['gravity']
         self.color = config['color']
+        self.mac = config['mac']
         options = {
             'statsd_host': config['host'],
             'statsd_port': safe_get_key(config, 'port', 8125),
@@ -32,14 +33,20 @@ def emit(self, **kwargs):  # pylint: disable=no-self-use,unused-argument
         Args:
         """
         LOGGER.info('[datadog] posting temperature data')
+        tags = [f"color:{self.color}"]
+        if self.mac:
+            tags = [
+                f"color:{self.color}",
+                f"mac:{self.mac}",
+            ]
         statsd.gauge(
             'tilty.temperature',
             self.temperature,
-            tags=[f"color:{self.color}"]
+            tags=tags,
         )
         LOGGER.info('[datadog] posting gravity data')
         statsd.gauge(
             'tilty.gravity',
             self.gravity,
-            tags=[f"color:{self.color}"]
+            tags=tags,
         )

tilty/tilt_device.py

@@ -44,6 +44,7 @@ def scan_for_tilt_data(self):
                     'color': constants.TILT_DEVICES[beacon['uuid']],
                     'gravity': float(beacon['minor']/1000),
                     'temp': beacon['major'],
+                    'mac': beacon['mac'],
                     'timestamp': datetime.now().isoformat(),
                 }
 

tilty/tilty.py

@@ -30,6 +30,7 @@ def emit(config, tilt_data):
             'payload': json.loads(_template.render(
                 color=tilt_data['color'],
                 gravity=tilt_data['gravity'],
+                mac=tilt_data['mac'],
                 temp=tilt_data['temp'],
                 timestamp=tilt_data['timestamp'],
             )),
@@ -53,12 +54,14 @@ def emit(config, tilt_data):
             'temperature_payload': _temperature_template.render(
                 color=tilt_data['color'],
                 gravity=tilt_data['gravity'],
+                mac=tilt_data['mac'],
                 temp=tilt_data['temp'],
                 timestamp=tilt_data['timestamp'],
             ),
             'gravity_payload': _gravity_template.render(
                 color=tilt_data['color'],
                 gravity=tilt_data['gravity'],
+                mac=tilt_data['mac'],
                 temp=tilt_data['temp'],
                 timestamp=tilt_data['timestamp'],
             ),
@@ -75,6 +78,7 @@ def emit(config, tilt_data):
             'host': config['datadog']['host'],
             'port': config['datadog']['port'],
             'gravity': tilt_data['gravity'],
+            'mac': tilt_data['mac'],
             'temperature': tilt_data['temp'],
             'color': tilt_data['color'],
         }