Enable support for snoo sleepytime timeout levels with enum provide values

Author: lexiconzero

python_snoo/baby.py

@@ -1,8 +1,8 @@
 from datetime import datetime
 
-from python_snoo.containers import Activity, BabyData, BreastfeedingActivity, DiaperActivity, DiaperTypes
-from python_snoo.exceptions import SnooBabyError
-from python_snoo.snoo import Snoo
+from .containers import Activity, BabyData, BreastfeedingActivity, DiaperActivity, DiaperTypes
+from .exceptions import SnooBabyError
+from .snoo import Snoo
 
 
 class Baby:

python_snoo/containers.py

@@ -1,6 +1,9 @@
+#maybe we dont need this
+from __future__ import annotations
+
 import dataclasses
 import datetime
-from enum import StrEnum
+from enum import StrEnum, IntEnum
 from typing import Any, Union
 
 from mashumaro.mixins.json import DataClassJSONMixin
@@ -15,6 +18,44 @@ class SnooLevels(StrEnum):
     level4 = "LEVEL4"
     stop = "ONLINE"
 
+class SnooNoiseTimeoutLevels(IntEnum):
+    _5_minutes = 5
+    _10_minutes = 10
+    _15_minutes = 15
+    _20_minutes = 20
+    _25_minutes = 25
+    _30_minutes = 30
+    _35_minutes = 35
+    _40_minutes = 40
+    _45_minutes = 45
+    _50_minutes = 50
+    _55_minutes = 55
+    _60_minutes = 60
+    _65_minutes = 65
+    _70_minutes = 70
+    _75_minutes = 75
+    _80_minutes = 80
+    _85_minutes = 85
+    _90_minutes = 90
+    _95_minutes = 95
+    _100_minutes = 100
+    _105_minutes = 105
+    _110_minutes = 110
+    _115_minutes = 115
+    _120_minutes = 120
+    _125_minutes = 125
+    _130_minutes = 130
+    _135_minutes = 135
+    _140_minutes = 140
+    _145_minutes = 145
+    _150_minutes = 150
+    _155_minutes = 155
+    _160_minutes = 160
+    _165_minutes = 165
+    _170_minutes = 170
+    _175_minutes = 175
+    _180_minutes = 180
+
 
 class SnooStates(StrEnum):
     baseline = "BASELINE"

python_snoo/snoo.py

@@ -19,6 +19,7 @@
     SnooData,
     SnooDevice,
     SnooStates,
+    SnooNoiseTimeoutLevels
 )
 from .exceptions import InvalidSnooAuth, SnooAuthException, SnooBabyError, SnooCommandException, SnooDeviceError
 from .pubnub_async import SnooPubNub
@@ -233,11 +234,11 @@ async def set_level(self, device: SnooDevice, level: SnooStates, hold: bool = Fa
 
         await self.send_command("go_to_state", device, **{"state": level.value, "hold": hold})
 
-    async def set_sticky_white_noise(self, device: SnooDevice, on: bool):
+    async def set_sticky_white_noise(self, device: SnooDevice, on: bool, timeout_value: SnooNoiseTimeoutLevels = SnooNoiseTimeoutLevels._15_minutes):
         await self.send_command(
             "set_sticky_white_noise",
             device,
-            **{"state": "on" if on else "off", "timeout_min": 15},
+            **{"state": "on" if on else "off", "timeout_min": timeout_value},
         )
 
     async def get_status(self, device: SnooDevice):