Added device ID requesting
continuous-integration/drone/push Build is passing Details

This commit is contained in:
Pünkösd Marcell 2020-06-20 20:20:36 +02:00
parent 6606ec0699
commit 3afa6d7260
2 changed files with 16 additions and 3 deletions

View File

@ -1,3 +1,4 @@
aio-pika
sentry-sdk
aioinflux
aioinflux
requests

View File

@ -16,6 +16,7 @@ from aio_pika import connect, IncomingMessage, ExchangeType
import sentry_sdk
from aioinflux import InfluxDBClient
import config
import requests
if config.SENTRY_DSN:
sentry_sdk.init(
@ -33,12 +34,23 @@ async def on_message(message: IncomingMessage):
"""
msg_json = json.loads(message.body)
now = datetime.now()
r = requests.get(f"http://input-service/sample/{msg_json['tag']}")
r.raise_for_status()
point = {
'time': now,
'measurement': 'sturnus',
'tags': {'device': 'dummy'},
'fields': {'probability': msg_json['probability']}
'tags': {
'device': r.json()['device_id']
},
'fields': {
'probability': float(msg_json['probability']) # This is required because otherwise of the json serialization and de-serialization round values would interpret as ints and influxdb would fail
}
}
async with InfluxDBClient(username=config.INFLUXDB_USERNAME, password=config.INFLUXDB_PASSWORD,
host=config.INFLUXDB_HOST, port=config.INFLUXDB_PORT, ssl=False,
database=config.INFLUXDB_DB) as client: