Use pressure units in events

Author: jason0x43

custom_components/hubitat/sensor.py

@@ -75,6 +75,7 @@ def __init__(
             else attribute.replace("_", " ")
         )
 
+        self.default_unit = unit
         self._attribute = attribute
         self._attr_name = f"{super(HubitatEntity, self).name} {attr_name}".title()
         self._attr_native_unit_of_measurement = unit
@@ -102,6 +103,27 @@ def _get_native_value(self) -> StateType | date | datetime | Decimal:
         """Return this sensor's current value."""
         return self.get_attr(self._attribute)
 
+    def _get_native_unit_of_measurement(self) -> str | None:
+        """Return this sensor's current value."""
+        unit = self.default_unit
+        units = {
+            "pa": UnitOfPressure.PA,
+            "hpa": UnitOfPressure.HPA,
+            "kpa": UnitOfPressure.KPA,
+            "bar": UnitOfPressure.BAR,
+            "cbar": UnitOfPressure.CBAR,
+            "mbar": UnitOfPressure.MBAR,
+            "mmhg": UnitOfPressure.MMHG,
+            "inhg": UnitOfPressure.INHG,
+            "psi": UnitOfPressure.PSI,
+        }
+        attr_unit = self.get_attr_unit(self._attribute)
+        if attr_unit is not None:
+            lower_unit = attr_unit.lower()
+            if lower_unit in units:
+                return units[lower_unit]
+        return unit
+
 
 class HubitatBatterySensor(HubitatSensor):
     """A battery sensor."""