dani/bite
1
0
Fork 0
mirror of https://github.com/daniviga/bite.git synced 2025-04-20 14:31:20 +02:00
Code Releases Activity

Do not crash mqtt-to-db if devices isn't proviosioned

Browse Source
This commit is contained in:
Daniele Viganò 2020-06-21 22:51:06 +02:00
parent 45d12b11a8
commit f5ff716ff3
Signed by: dani
GPG Key ID: DB49AFC03C40EE02
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
bite/telemetry/management/commands/mqtt-to-db.py
View File

@ -8,6 +8,8 @@ from asyncio_mqtt import Client
from django.conf import settings from django.conf import settings
from django.core.management.base import BaseCommand from django.core.management.base import BaseCommand
from django.core.exceptions import ObjectDoesNotExist
from api.models import Device from api.models import Device
from telemetry.models import Telemetry from telemetry.models import Telemetry
@ -20,7 +22,10 @@ class Command(BaseCommand):
@sync_to_async @sync_to_async
def get_device(self, serial): def get_device(self, serial):
return Device.objects.get(serial=serial) try:
return Device.objects.get(serial=serial)
except ObjectDoesNotExist:
return None
@sync_to_async @sync_to_async
def store_telemetry(self, device, payload): def store_telemetry(self, device, payload):
@ -39,7 +44,12 @@ class Command(BaseCommand):
async for message in messages: async for message in messages:
payload = json.loads(message.payload.decode('utf-8')) payload = json.loads(message.payload.decode('utf-8'))
device = await self.get_device(message.topic) device = await self.get_device(message.topic)
await self.store_telemetry(device, payload) if device is not None:
await self.store_telemetry(device, payload)
else:
self.stdout.write(
self.style.ERROR(
'DEBUG: message discarded'))
def handle(self, *args, **options): def handle(self, *args, **options):
client = mqtt.Client() client = mqtt.Client()