Merge pull request #50 from PimDoos/dev

Convert to StrEnums

Author: PimDoos

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sessypy"
-version = "0.1.14"
+version = "0.1.15"
 authors = [
 	{ name="PimDoos" },
 ]

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = sessypy
-version = 0.1.14
+version = 0.1.15
 author = PimDoos
 url = https://github.com/PimDoos/sessypy
 long_description = file: README.md

src/sessypy/const.py

@@ -1,9 +1,9 @@
-from enum import Enum
+from enum import StrEnum
 
 API_VERSION_1 = "api/v1"
 API_VERSION_2 = "api/v2"
 
-class SessyApiCommand(str, Enum):
+class SessyApiCommand(StrEnum):
     CT_DETAILS = f"{API_VERSION_1}/ct/details"
     DYNAMIC_SCHEDULE = f"{API_VERSION_1}/dynamic/schedule"
 
@@ -31,15 +31,15 @@ class SessyApiCommand(str, Enum):
     WIFI_STA_CREDENTIALS = f"{API_VERSION_1}/wifi_sta/credentials"
 
 
-class SessyPowerStrategy(str, Enum):
+class SessyPowerStrategy(StrEnum):
     API = "POWER_STRATEGY_API"
     NOM = "POWER_STRATEGY_NOM"
     ROI = "POWER_STRATEGY_ROI"
     IDLE = "POWER_STRATEGY_IDLE"
     SESSY_CONNECT = "POWER_STRATEGY_SESSY_CONNECT"
     ECO = "POWER_STRATEGY_ECO"
 
-class SessySystemState(str, Enum):
+class SessySystemState(StrEnum):
     RUNNING_SAFE = "SYSTEM_STATE_RUNNING_SAFE"
     STANDBY = "SYSTEM_STATE_STANDBY"
     WAITING_FOR_SAFE = "SYSTEM_STATE_WAITING_FOR_SAFE_SITUATION"
@@ -55,19 +55,19 @@ class SessySystemState(str, Enum):
     BATTERY_EMPTY = "SYSTEM_STATE_BATTERY_EMPTY"
     OVERRIDE_BATTERY_UNDERVOLTAGE = "SYSTEM_STATE_OVERRIDE_BATTERY_UNDERVOLTAGE"
 
-class SessyP1State(str, Enum):
+class SessyP1State(StrEnum):
     NOT_CONNECTED = "P1_NOT_CONNECTED"
     DATA_VALIDITY_ERROR = "P1_DATAVALIDITY_ERR"
     VERSION_ERROR = "P1_VERSION_ERR"
     PARSE_ERROR = "P1_PARSE_ERR"
     OK = "P1_OK"
 
-class SessyOtaTarget(str, Enum):
+class SessyOtaTarget(StrEnum):
     SELF = "OTA_TARGET_SELF"
     SERIAL = "OTA_TARGET_SERIAL"
     ALL = "OTA_TARGET_ALL"  # available as of version 1.5.1
 
-class SessyOtaState(str, Enum):
+class SessyOtaState(StrEnum):
     FAILED = "OTA_UPDATE_FAILED"
     DISABLED = "OTA_DISABLED"
     INACTIVE = "OTA_INACTIVE"

src/sessypy/devices.py

@@ -76,7 +76,7 @@ async def get_power_strategy(self):
         return await self.api.get(SessyApiCommand.POWER_STRATEGY)
 
     async def set_power_strategy(self, strategy: SessyPowerStrategy):
-        return await self.api.post(SessyApiCommand.POWER_STRATEGY, {"strategy": strategy.value})
+        return await self.api.post(SessyApiCommand.POWER_STRATEGY, {"strategy": strategy})
 
     async def get_system_settings(self):
         return await self.api.get(SessyApiCommand.SYSTEM_SETTINGS)