Update add mode/battery

Author: Khole Jones

pyhiveapi/apyhiveapi/session.py

@@ -288,13 +288,10 @@ async def createDevices(self):
             for code in product_list:
                 eval("self." + code)
 
-            if self.data.products[aProduct]["type"] in HIVE_TYPES["Heating"]:
-                self.config.mode.append(p["id"])
-
-            if self.data.products[aProduct]["type"] in HIVE_TYPES["Switch"]:
-                self.config.mode.append(p["id"])
-
-            if self.data.products[aProduct]["type"] in HIVE_TYPES["Light"]:
+            hive_type = (
+                HIVE_TYPES["Heating"] + HIVE_TYPES["Switch"] + HIVE_TYPES["Light"]
+            )
+            if self.data.products[aProduct]["type"] in hive_type:
                 self.config.mode.append(p["id"])
 
         for aDevice in self.data["devices"]:
@@ -304,10 +301,8 @@ async def createDevices(self):
             for code in device_list:
                 eval("self." + code)
 
-            if (
-                self.data["devices"][aDevice]["type"] in HIVE_TYPES["Thermo"]
-                or self.data["devices"][aDevice]["type"] in HIVE_TYPES["Sensor"]
-            ):
+            hive_type = HIVE_TYPES["Thermo"] + HIVE_TYPES["Sensor"]
+            if self.data["devices"][aDevice]["type"] in hive_type:
                 self.config.battery.append(d["id"])
 
         if "action" in HIVE_TYPES["Switch"]:

setup.py

@@ -15,7 +15,7 @@ def requirements_from_file(filename="requirements.txt"):
 
 
 setup(
-    version="0.3.8",
+    version="0.3.9",
     package_data={"data": ["*.json"]},
     include_package_data=True,
     cmdclass={