feat: add enum of known and tested models to aws devices

Author: dahlb

src/blueair_api/__init__.py

@@ -4,3 +4,4 @@
 from .util_bootstrap import get_devices, get_aws_devices
 from .device import Device
 from .device_aws import DeviceAws
+from .model_enum import ModelEnum

src/blueair_api/const.py

@@ -1,7 +1,6 @@
 from collections.abc import Mapping
 from typing_extensions import TypedDict
 
-
 SENSITIVE_FIELD_NAMES = [
     "username",
     "password",
@@ -10,13 +9,6 @@
 # The BlueAir API uses a fixed API key.
 API_KEY = "eyJhbGciOiJIUzI1NiJ9.eyJncmFudGVlIjoiYmx1ZWFpciIsImlhdCI6MTQ1MzEyNTYzMiwidmFsaWRpdHkiOi0xLCJqdGkiOiJkNmY3OGE0Yi1iMWNkLTRkZDgtOTA2Yi1kN2JkNzM0MTQ2NzQiLCJwZXJtaXNzaW9ucyI6WyJhbGwiXSwicXVvdGEiOi0xLCJyYXRlTGltaXQiOi0xfQ.CJsfWVzFKKDDA6rWdh-hjVVVE9S3d6Hu9BzXG9htWFw"  # noqa: E501
 
-
-class MeasurementBundle(TypedDict):
-    sensors: list[str]
-    datapoints: list[list[int | float]]
-
-MeasurementList = list[Mapping[str, int | float]]
-
 AWS_APIKEYS = {
     "us": {
         "gigyaRegion": "us1",
@@ -31,3 +23,11 @@ class MeasurementBundle(TypedDict):
         "apiKey": "3_qRseYzrUJl1VyxvSJANalu_kNgQ83swB1B9uzgms58--5w1ClVNmrFdsDnWVQQCl",
     },
 }
+
+
+class MeasurementBundle(TypedDict):
+    sensors: list[str]
+    datapoints: list[list[int | float]]
+
+
+MeasurementList = list[Mapping[str, int | float]]

src/blueair_api/device_aws.py

@@ -2,6 +2,7 @@
 
 from .callbacks import CallbacksMixin
 from .http_aws_blueair import HttpAwsBlueair
+from .model_enum import ModelEnum
 from .util import convert_api_array_to_dict, safely_get_json_value
 
 _LOGGER = logging.getLogger(__name__)
@@ -132,6 +133,14 @@ async def set_wick_dry_mode(self, value: bool):
         await self.api.set_device_info(self.uuid, "wickdrys", "vb", value)
         self.publish_updates()
 
+    @property
+    def model(self) -> ModelEnum:
+        if self.sku == "111633":
+            return ModelEnum.HUMIDIFIER_I35
+        if self.sku == "105826":
+            return ModelEnum.PROTECT_7470I
+        return ModelEnum.UNKNOWN
+
     def __repr__(self):
         return {
             "uuid": self.uuid,

src/blueair_api/model_enum.py

@@ -0,0 +1,7 @@
+from enum import StrEnum
+
+
+class ModelEnum(StrEnum):
+    UNKNOWN = "Unknown"
+    HUMIDIFIER_I35 = "Blueair Humidifier i35"
+    PROTECT_7470I = "Blueair Protect 7470i"