Skip to content

Feat: Enable support for snoo sleepytime timeout levels #46

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Lash-L merged 10 commits into from
Jan 29, 2026
Merged

Feat: Enable support for snoo sleepytime timeout levels #46

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
1a9b50f
Enable support for snoo sleepytime timeout levels with enum provide v…
lexiconzero Jan 21, 2026
d9e51f1
Update python_snoo/snoo.py
lexiconzero Jan 21, 2026
a84c8ba
Update containers.py
lexiconzero Jan 22, 2026
3e263b5
Fix up linting as suggested
lexiconzero Jan 22, 2026
8a74f0c
Fix ordering of imports
lexiconzero Jan 22, 2026
04f2612
Update snoo.py
lexiconzero Jan 22, 2026
2e100ec
Update python_snoo/snoo.py
lexiconzero Jan 23, 2026
fa2232f
Update snoo.py for linter
lexiconzero Jan 29, 2026
e8f7697
trailing comma for linter
lexiconzero Jan 29, 2026
27d4a75
chore: lint
Lash-L Jan 29, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions python_snoo/baby.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -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:
Expand Down
41 changes: 40 additions & 1 deletion python_snoo/containers.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import dataclasses
import datetime
from enum import StrEnum
from enum import IntEnum, StrEnum
from typing import Any, Union

from mashumaro.mixins.json import DataClassJSONMixin
Expand All @@ -16,6 +16,45 @@ class SnooLevels(StrEnum):
stop = "ONLINE"


class SnooNoiseTimeoutLevels(IntEnum):
Copy link
Owner

@Lash-L Lash-L Jan 22, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

So walk me through it - what is the benefit of this IntEnum opposed to just passing in a int? Does Snoo only accept specific values?

Copy link
Contributor Author

@lexiconzero lexiconzero Jan 22, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Readability and "app feeling" cloning was the idea - I basically just replicated the setup for the snoo levels. I believe the API just wants an integer, however the app only provides it in 5 minute increments. I don't like passing magic numbers around hence the enum class

Copy link
Owner

@Lash-L Lash-L Jan 22, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

okay not completely against it - but you will have to see how this feels when you are adding it on the core side. It may be good to accept the IntEnum | int just to make your life easier in the function just in case you need to pivot.

Copy link
Contributor Author

@lexiconzero lexiconzero Jan 22, 2026 •
edited
Loading

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

When writing the HA core code, I had to replicate the structure over to the strings file to get it to look good for the UI - so realistically that code is now duplicated. I thought that I could just let the HA UI handle it, but then it would have that logic only in HA, leaving python_snoo with a potential foot-gun should someone decide to pass 190 minutes through to the API.

I'm happy either way

Copy link
Owner

@Lash-L Lash-L Jan 22, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

this may simplify things?

https://developers.home-assistant.io/blog/2024/01/19/entity-translations-placeholders/

Copy link
Owner

@Lash-L Lash-L Jan 23, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Realized this does not allow for translation placeholders in options...

Will approve for now, but we will see what the actual core implementation looks like and what the HA devs think

Copy link
Contributor Author

@lexiconzero lexiconzero Jan 23, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Will have a crack at it! Thanks!

_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"
level1 = "LEVEL1"
Expand Down
26 changes: 17 additions & 9 deletions python_snoo/snoo.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,13 +13,7 @@
from pubnub.pnconfiguration import PNConfiguration
from pubnub.pubnub_asyncio import PubNubAsyncio

from .containers import (
AuthorizationInfo,
BabyData,
SnooData,
SnooDevice,
SnooStates,
)
from .containers import AuthorizationInfo, BabyData, SnooData, SnooDevice, SnooNoiseTimeoutLevels, SnooStates
from .exceptions import InvalidSnooAuth, SnooAuthException, SnooBabyError, SnooCommandException, SnooDeviceError
from .pubnub_async import SnooPubNub

Expand Down Expand Up @@ -233,11 +227,25 @@ 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 | int = SnooNoiseTimeoutLevels._15_minutes,
):
"""Enable or disable sticky white noise for a device.

Args:
device: The SnooDevice to control.
on: True to turn sticky white noise on, False to turn it off.
timeout_value: How long the white noise should remain on before timing out.
Must be a member of the SnooNoiseTimeoutLevels enum. The default is
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):
Expand Down
Loading