Expose device types supported by Bond as constants

bond_api/__init__.py

@@ -2,3 +2,4 @@
 
 from .bond import Bond
 from .action import Action, Direction
+from .device_type import DeviceType

bond_api/device_type.py

@@ -0,0 +1,29 @@
+"""Bond Device type enumeration."""
+
+
+class DeviceType:
+    """Bond Device type enumeration."""
+    CEILING_FAN = "CF"
+    MOTORIZED_SHADES = "MS"
+    FIREPLACE = "FP"
+    GENERIC_DEVICE = "GX"
+
+    @staticmethod
+    def is_fan(device_type: str) -> bool:
+        """Checks if specified device type is a fan."""
+        return device_type == DeviceType.CEILING_FAN
+
+    @staticmethod
+    def is_shades(device_type: str) -> bool:
+        """Checks if specified device type is shades."""
+        return device_type == DeviceType.MOTORIZED_SHADES
+
+    @staticmethod
+    def is_fireplace(device_type: str) -> bool:
+        """Checks if specified device type is fireplace."""
+        return device_type == DeviceType.FIREPLACE
+
+    @staticmethod
+    def is_generic(device_type: str) -> bool:
+        """Checks if specified device type is generic."""
+        return device_type == DeviceType.GENERIC_DEVICE

setup.py

@@ -12,7 +12,7 @@
 
 setup(
     name="bond_api",
-    version="0.1.0",
+    version="0.1.2",
     packages=find_packages(),
 
     author="Eugene Prystupa",

tests/test_device_type.py

@@ -0,0 +1,15 @@
+from bond_api.device_type import DeviceType
+
+
+def test_compare_device_types():
+    assert DeviceType.CEILING_FAN == "CF"
+    assert DeviceType.is_fan("CF")
+
+    assert DeviceType.MOTORIZED_SHADES == "MS"
+    assert DeviceType.is_shades("MS")
+
+    assert DeviceType.FIREPLACE == "FP"
+    assert DeviceType.is_fireplace("FP")
+
+    assert DeviceType.GENERIC_DEVICE == "GX"
+    assert DeviceType.is_generic("GX")