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

Split get users into chunks of 100 for Twitch sensors (#101211)

Browse Source 
* Split get users into chunks of 100

* Move to own function
This commit is contained in:
Aidan Timson 2023-10-02 14:20:19 +01:00 committed by GitHub
parent 15f945c47e
commit 9eaf326c0e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 19 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -52,6 +52,11 @@ STATE_OFFLINE = "offline"
STATE_STREAMING = "streaming"
def chunk_list(lst: list, chunk_size: int) -> list[list]:
"""Split a list into chunks of chunk_size."""
return [lst[i : i + chunk_size] for i in range(0, len(lst), chunk_size)]
async def async_setup_platform(
hass: HomeAssistant,
config: ConfigType,
@ -94,13 +99,20 @@ async def async_setup_entry(
"""Initialize entries."""
client = hass.data[DOMAIN][entry.entry_id]
async_add_entities(
[
TwitchSensor(channel, client)
async for channel in client.get_users(logins=entry.options[CONF_CHANNELS])
],
True,
)
channels = entry.options[CONF_CHANNELS]
entities: list[TwitchSensor] = []
# Split channels into chunks of 100 to avoid hitting the rate limit
for chunk in chunk_list(channels, 100):
entities.extend(
[
TwitchSensor(channel, client)
async for channel in client.get_users(logins=chunk)
]
)
async_add_entities(entities, True)
class TwitchSensor(SensorEntity):