fix: update get_device()

Sunbrye Ly · Sunbrye Ly · commit aef422468d8a · 2022-05-25T16:50:28.000-07:00
diff --git a/api/devices.py b/api/devices.py
@@ -28,31 +28,24 @@ def get_buckets():
     return buckets
 
 
-def get_device() -> {}:
+def get_device(device_id) -> {}:
     influxdb_client = InfluxDBClient(url=config.get('APP', 'INFLUX_URL'),
                                      token=config.get('APP', 'INFLUX_TOKEN'),
                                      org=config.get('APP', 'INFLUX_ORG'))
-
     # Queries must be formatted with single and double quotes correctly
     query_api = QueryApi(influxdb_client)
-    # query_api = influxdb_client.query_api()
-    # device_id = str(device_id)
-    device_filter = f'r._field != "token"'
+    device_id = str(device_id)
+    device_filter = f'r.deviceId == "{device_id}" and r._field != "token"'
     flux_query = f'from(bucket: "{config.get("APP", "INFLUX_BUCKET_AUTH")}") ' \
                  f'|> range(start: 0) ' \
                  f'|> filter(fn: (r) => r._measurement == "deviceauth" and {device_filter}) ' \
                  f'|> last()'
-    devices = {}
 
     response = query_api.query(flux_query)
-
-    # iterate through the result(s)
-    # TODO maybe change this to only show, device_id, auth_id, auth_token?
-    # iterate through the result(s)
     results = []
     for table in response:
-        results.append(table.records[0].values)
-
+        for record in table.records:
+            results.append((record.get_field(), record.get_value()))
     return results
 
 
