Added camera functionality

custom_components/browser_mod/sensor.py

@@ -1,22 +1,29 @@
 from homeassistant.components.sensor import SensorEntity
 
 from .const import DOMAIN, DATA_ADDERS
-from .helpers import BrowserModEntity2
+from .helpers import BrowserModEntity
 
 
-async def async_setup_platform(hass, config_entry, async_add_entities, discoveryInfo = None):
+async def async_setup_platform(
+    hass, config_entry, async_add_entities, discoveryInfo=None
+):
     hass.data[DOMAIN][DATA_ADDERS]["sensor"] = async_add_entities
 
+
 async def async_setup_entry(hass, config_entry, async_add_entities):
     await async_setup_platform(hass, {}, async_add_entities)
 
 
-class BrowserSensor(BrowserModEntity2, SensorEntity):
-    def __init__(self, coordinator, deviceID, parameter,
-            name,
-            unit_of_measurement = None,
-            device_class = None,
-        ):
+class BrowserSensor(BrowserModEntity, SensorEntity):
+    def __init__(
+        self,
+        coordinator,
+        deviceID,
+        parameter,
+        name,
+        unit_of_measurement=None,
+        device_class=None,
+    ):
         super().__init__(coordinator, deviceID, name)
         self.parameter = parameter
         self._device_class = device_class
@@ -28,9 +35,11 @@ class BrowserSensor(BrowserModEntity2, SensorEntity):
         data = data.get("browser", {})
         data = data.get(self.parameter, None)
         return data
+
     @property
     def device_class(self):
         return self._device_class
+
     @property
     def native_unit_of_measurement(self):
         return self._unit_of_measurement
@@ -41,8 +50,10 @@ class BrowserSensor(BrowserModEntity2, SensorEntity):
         if self.parameter == "currentUser":
             retval["userData"] = self._data.get("browser", {}).get("userData")
         if self.parameter == "path":
-            retval["pathSegments"] = self._data.get("browser", {}).get("path", "").split("/")
+            retval["pathSegments"] = (
+                self._data.get("browser", {}).get("path", "").split("/")
+            )
         if self.parameter == "userAgent":
             retval["userAgent"] = self._data.get("browser", {}).get("userAgent")
 
-        return retval
+        return retval