jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 21:27:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Twitch stream viewer as readable data for integration (#128787)

Browse Source
This commit is contained in:
LunaBytesBack 2024-10-20 16:07:43 +02:00 committed by GitHub
parent 0c6a640e50
commit 711c448242
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
homeassistant/components/twitch/coordinator.py
View File

@ -38,6 +38,7 @@ class TwitchUpdate:
subscription_gifted: bool | None subscription_gifted: bool | None
follows: bool follows: bool
following_since: datetime | None following_since: datetime | None
viewers: int | None
class TwitchCoordinator(DataUpdateCoordinator[dict[str, TwitchUpdate]]): class TwitchCoordinator(DataUpdateCoordinator[dict[str, TwitchUpdate]]):
@ -112,5 +113,6 @@ class TwitchCoordinator(DataUpdateCoordinator[dict[str, TwitchUpdate]]):
sub.is_gift if sub else None, sub.is_gift if sub else None,
follows is not None and follows.total > 0, follows is not None and follows.total > 0,
follows.data[0].followed_at if follows and follows.total else None, follows.data[0].followed_at if follows and follows.total else None,
stream.viewer_count if stream else None,
) )
return data return data

2
homeassistant/components/twitch/sensor.py
View File

@ -24,6 +24,7 @@ ATTR_FOLLOW = "following"
ATTR_FOLLOW_SINCE = "following_since" ATTR_FOLLOW_SINCE = "following_since"
ATTR_FOLLOWING = "followers" ATTR_FOLLOWING = "followers"
ATTR_VIEWS = "views" ATTR_VIEWS = "views"
ATTR_VIEWERS = "viewers"
ATTR_STARTED_AT = "started_at" ATTR_STARTED_AT = "started_at"
STATE_OFFLINE = "offline" STATE_OFFLINE = "offline"
@ -82,6 +83,7 @@ class TwitchSensor(CoordinatorEntity[TwitchCoordinator], SensorEntity):
ATTR_GAME: channel.game, ATTR_GAME: channel.game,
ATTR_TITLE: channel.title, ATTR_TITLE: channel.title,
ATTR_STARTED_AT: channel.started_at, ATTR_STARTED_AT: channel.started_at,
ATTR_VIEWERS: channel.viewers,
} }
resp[ATTR_SUBSCRIPTION] = False resp[ATTR_SUBSCRIPTION] = False
if channel.subscribed is not None: if channel.subscribed is not None:

3
tests/components/twitch/fixtures/get_streams.json
View File

@ -3,6 +3,7 @@
"game_name": "Good game", "game_name": "Good game",
"title": "Title", "title": "Title",
"thumbnail_url": "stream-medium.png", "thumbnail_url": "stream-medium.png",
"started_at": "2021-03-10T03:18:11Z" "started_at": "2021-03-10T03:18:11Z",
"viewer_count": 42
} }
] ]

1
tests/components/twitch/test_sensor.py
View File

@ -45,6 +45,7 @@ async def test_streaming(
assert sensor_state.attributes["started_at"] == datetime( assert sensor_state.attributes["started_at"] == datetime(
year=2021, month=3, day=10, hour=3, minute=18, second=11, tzinfo=tzutc() year=2021, month=3, day=10, hour=3, minute=18, second=11, tzinfo=tzutc()
) )
assert sensor_state.attributes["viewers"] == 42
async def test_oauth_without_sub_and_follow( async def test_oauth_without_sub_and_follow(