File: __init__.py

package info (click to toggle)
python-luftdaten 0.7.4-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 128 kB
  • sloc: python: 172; makefile: 6; sh: 5
file content (73 lines) | stat: -rw-r--r-- 2,714 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
"""Wrapper to get the measurings from a Luftdaten station."""
import logging

import httpx

from . import exceptions

_LOGGER = logging.getLogger(__name__)
_RESOURCE = "https://data.sensor.community/airrohr/v1"


class Luftdaten(object):
    """A class for handling the data retrieval."""

    def __init__(self, sensor_id):
        """Initialize the connection."""
        self.sensor_id = sensor_id
        self.data = None
        self.values = {}
        self.meta = {}
        self.url = "{}/{}".format(_RESOURCE, "sensor")

    async def get_data(self):
        """Retrieve the data."""
        url = "{}/{}/".format(self.url, self.sensor_id)

        try:
            async with httpx.AsyncClient() as client:
                response = await client.get(str(url))
        except httpx.ConnectError:
            raise exceptions.LuftdatenConnectionError(f"Connection to {url} failed")
        except httpx.ConnectTimeout:
            raise exceptions.LuftdatenConnectionError(f"Connection to {url} timed out")
        except httpx.ReadTimeout:
            raise exceptions.LuftdatenConnectionError(f"Read timeout from {url}")

        if response.status_code == httpx.codes.OK:
            try:
                _LOGGER.debug(response.json())
                self.data = response.json()
            except TypeError:
                _LOGGER.error("Can not load data from Luftdaten API")
                raise exceptions.LuftdatenConnectionError(
                    "Unable to get the data from Luftdaten API"
                )

        if not self.data:
            for measurement in self.values.keys():
                self.values[measurement] = None
            return

        try:
            sensor_data = sorted(
                self.data, key=lambda timestamp: timestamp["timestamp"], reverse=True
            )[0]

            for entry in sensor_data["sensordatavalues"]:
                if entry["value_type"] not in self.values.keys():
                    self.values[entry["value_type"]] = None
                for measurement in self.values.keys():
                    if measurement == entry["value_type"]:
                        self.values[measurement] = float(entry["value"])

            self.meta["sensor_id"] = self.sensor_id
            self.meta["longitude"] = float(sensor_data["location"]["longitude"])
            self.meta["latitude"] = float(sensor_data["location"]["latitude"])
            self.meta["altitude"] = float(sensor_data["location"]["altitude"])
        except (TypeError, IndexError):
            raise exceptions.LuftdatenError()

    async def validate_sensor(self):
        """Return True if the sensor ID is valid."""
        return True if self.data else False