Add support for Solarbank 2

custom_components/solix_ble/__init__.py

@@ -21,6 +21,7 @@
     Generic,
     PrimeCharger160w,
     PrimeCharger250w,
+    Solarbank2,
     SolixBLEDevice,
 )
 
@@ -31,12 +32,12 @@
 type SolixBLEConfigEntry = ConfigEntry[SolixBLEDevice]
 
 
 def get_power_station_class(model: Models) -> SolixBLEDevice:
     """Return correct class for power station from model."""
 
     if model is Models.C300:
         return C300
     elif model is Models.C300DC:
         return C300DC
     elif model is Models.C800:
         return C800
@@ -52,16 +53,18 @@
         return PrimeCharger160w
     elif model is Models.PRIME_CHARGER_250:
         return PrimeCharger250w
+    elif model is Models.SOLARBANK_2:
+        return Solarbank2
     elif model is Models.UNKNOWN:
         return Generic
     else:
         raise NotImplementedError(f"Unexpected model. Got: '{type(model)}'!")
 
 
 async def async_setup_entry(hass: HomeAssistant, entry: SolixBLEConfigEntry) -> bool:
     """Set up the integration from a config entry."""
 
     assert entry.unique_id is not None
     address = entry.unique_id.upper()
     model = Models(entry.data["model"])
 
@@ -72,10 +75,10 @@
         _LOGGER.debug("Count of BLE scanners: %i", count_scanners)
 
         if count_scanners < 1:
             raise ConfigEntryNotReady(
                 "No Bluetooth scanners are available to search for the device."
             )
         raise ConfigEntryNotReady("The device was not found.")
 
     DeviceClass = get_power_station_class(model)
     if model is Models.UNKNOWN:

custom_components/solix_ble/const.py

@@ -4,8 +4,18 @@
 
 DOMAIN = "solix_ble"
 
+
+###############
+# Port status #
+###############
+
 PORT_STATUS_STRINGS = ["Unknown", "Not connected", "Output", "Input"]
 
+
+###################
+# Charging status #
+###################
+
 CHARGING_STATUS_C300_STRINGS = ["Unknown", "Idle", "Discharging", "Charging"]
 
 CHARGING_STATUS_F3800_STRINGS = [
@@ -16,11 +26,65 @@
     "Charging (Both)",
 ]
 
+
+################
+# Light status #
+################
+
 LIGHT_STATUS_STRINGS = ["Unknown", "Off", "Low", "Medium", "High"]
 
+LIGHT_STATUS_SB2_STRINGS = ["Unknown", "Normal", "Mood"]
+
+
+###################
+# Overload status #
+###################
+
 OVERLOAD_STATUS_C300DC_STRINGS = ["Unknown", "None", "USB C1", "USB C2", "USB C3"]
 
 
+###################
+# Max load status #
+###################
+
+MAX_LOAD_SB2_STRINGS = ["Unknown", "350w", "600w", "800w", "1000w"]
+
+
+###############
+# Grid status #
+###############
+
+GRID_STATUS_STRINGS = ["Unknown", "Ok", "Ok?", "Connecting", "No grid"]
+
+
+##############
+# Usage mode #
+##############
+
+USAGE_MODE_SB2_STRINGS = [
+    "Unknown",
+    "Manual",
+    "Smart meter",
+    "Smart plugs",
+    "Backup",
+    "Use Time",
+    "Smart",
+    "Time slot",
+]
+
+
+##################
+# Battery cutoff #
+##################
+
+CUT_OFF_SB2_STRINGS = ["Unknown", "5%", "10%"]
+
+
+####################
+# Supported models #
+####################
+
+
 class Models(Enum):
     C300 = "C300(X)"
     C300DC = "C300(X) DC"
@@ -31,4 +95,5 @@ class Models(Enum):
     F3800 = "F3800"
     PRIME_CHARGER_160 = "Prime Charger (160w)"
     PRIME_CHARGER_250 = "Prime Charger (250w)"
+    SOLARBANK_2 = "Solarbank 2"
     UNKNOWN = "Unknown"