From 3ee6bce9a390adab5c95385726bdbd98b696fbf4 Mon Sep 17 00:00:00 2001 From: kenoel Date: Tue, 22 Aug 2023 11:57:32 -0400 Subject: [PATCH] Add constant for online state data batch size --- caracara/common/constants.py | 3 +++ caracara/modules/hosts/_online_state.py | 4 ++-- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/caracara/common/constants.py b/caracara/common/constants.py index ec780d6..48b8459 100644 --- a/caracara/common/constants.py +++ b/caracara/common/constants.py @@ -4,6 +4,9 @@ # Batch size of data downloaded via a multi-threaded data pull DEFAULT_DATA_BATCH_SIZE = 500 +# Batch size of data downloaded via a multi-threaded data pull from the online state endpoint +ONLINE_STATE_DATA_BATCH_SIZE = 100 + # Default pagination limit PAGINATION_LIMIT = 100 diff --git a/caracara/modules/hosts/_online_state.py b/caracara/modules/hosts/_online_state.py index 56d5631..d1f52a6 100644 --- a/caracara/modules/hosts/_online_state.py +++ b/caracara/modules/hosts/_online_state.py @@ -12,7 +12,7 @@ ) from caracara.common.batching import batch_get_data -from caracara.common.constants import OnlineState +from caracara.common.constants import OnlineState, ONLINE_STATE_DATA_BATCH_SIZE from caracara.common.exceptions import InvalidOnlineState if TYPE_CHECKING: @@ -59,7 +59,7 @@ def get_online_state( device_online_state_data = batch_get_data( device_ids, self.hosts_api.get_online_state, - data_batch_size=100, + data_batch_size=ONLINE_STATE_DATA_BATCH_SIZE, ) return device_online_state_data