Rebase from upstream

custom_components/localtuya/number.py

@@ -5,13 +5,19 @@ from functools import partial
 import voluptuous as vol
 from homeassistant.components.number import DOMAIN, NumberEntity
 from homeassistant.const import CONF_DEVICE_CLASS, STATE_UNKNOWN
+from homeassistant.helpers.restore_state import RestoreEntity
 
 from .common import LocalTuyaEntity, async_setup_entry
 
 _LOGGER = logging.getLogger(__name__)
 
-CONF_MIN_VALUE = "min_value"
-CONF_MAX_VALUE = "max_value"
+from .const import (
+    CONF_DEFAULT_VALUE,
+    CONF_MIN_VALUE,
+    CONF_MAX_VALUE,
+    CONF_DEFAULT_VALUE,
+    CONF_RESTORE_ON_RECONNECT,
+)
 
 DEFAULT_MIN = 0
 DEFAULT_MAX = 100000
@@ -28,10 +34,12 @@ def flow_schema(dps):
             vol.Coerce(float),
             vol.Range(min=-1000000.0, max=1000000.0),
         ),
+        vol.Optional(CONF_DEFAULT_VALUE): str,
+        vol.Required(CONF_RESTORE_ON_RECONNECT): bool,
     }
 
 
-class LocaltuyaNumber(LocalTuyaEntity, NumberEntity):
+class LocaltuyaNumber(LocalTuyaEntity, NumberEntity, RestoreEntity):
     """Representation of a Tuya Number."""
 
     def __init__(
@@ -51,6 +59,11 @@ class LocaltuyaNumber(LocalTuyaEntity, NumberEntity):
 
         self._max_value = self._config.get(CONF_MAX_VALUE)
 
+        #Override standard default value handling to cast to a float
+        default_value = self._config.get(CONF_DEFAULT_VALUE)
+        if default_value is not None:
+            self._default_value = float(default_value)
+
     @property
     def value(self) -> float:
         """Return sensor state."""
@@ -75,10 +88,9 @@ class LocaltuyaNumber(LocalTuyaEntity, NumberEntity):
         """Update the current value."""
         await self._device.set_dp(value, self._dp_id)
 
-    def status_updated(self):
-        """Device status was updated."""
-        state = self.dps(self._dp_id)
-        self._state = state
+    # Default value is the minimum value
+    def entity_default_value(self):
+        return self._min_value
 
 
 async_setup_entry = partial(async_setup_entry, DOMAIN, LocaltuyaNumber, flow_schema)