Skip to content

Commit a60db1d

Browse files
authored
Update the usage step size to follow Platform UI (#107)
1 parent e42c1f7 commit a60db1d

File tree

1 file changed

+17
-14
lines changed

1 file changed

+17
-14
lines changed

examples/sdk/get_deployment_usage.py

Lines changed: 17 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -11,24 +11,27 @@
1111

1212
def get_step_size(start_time_in_seconds: int, end_time_in_seconds: int) -> int:
1313
time_delta_in_seconds = end_time_in_seconds - start_time_in_seconds
14-
if time_delta_in_seconds <= 3 * HOUR_IN_SECONDS:
15-
return 15
16-
elif time_delta_in_seconds <= 6 * HOUR_IN_SECONDS:
17-
return 30
18-
elif time_delta_in_seconds <= 12 * HOUR_IN_SECONDS:
14+
# 0 seconds to 2 days: 60s
15+
if time_delta_in_seconds <= 2 * DAY_IN_SECONDS:
1916
return 60
20-
elif time_delta_in_seconds <= 24 * HOUR_IN_SECONDS:
21-
return 2 * 60
22-
elif time_delta_in_seconds <= 2 * DAY_IN_SECONDS:
23-
return 2 * 60
17+
# 2 days to 7 days: 5m
2418
elif time_delta_in_seconds <= 7 * DAY_IN_SECONDS:
19+
return 5 * 60
20+
# 7 days to 14 days: 10m
21+
elif time_delta_in_seconds <= 14 * DAY_IN_SECONDS:
2522
return 10 * 60
26-
elif time_delta_in_seconds <= 15 * DAY_IN_SECONDS:
27-
return 30 * 60
23+
# 14 days to 30 days: 30m
2824
elif time_delta_in_seconds <= 30 * DAY_IN_SECONDS:
29-
return 60 * 60
30-
return math.ceil(time_delta_in_seconds / MAX_DATA_POINTS)
31-
25+
return 30 * 60
26+
# 30 days to 60 days: 1 hour
27+
elif time_delta_in_seconds <= 60 * DAY_IN_SECONDS:
28+
return HOUR_IN_SECONDS
29+
# 60 days to 90 days: 2 hours
30+
elif time_delta_in_seconds <= 90 * DAY_IN_SECONDS:
31+
return 2 * HOUR_IN_SECONDS
32+
# 90+ days: 3 hours (This catches all ranges greater than 90 days)
33+
else:
34+
return 3 * HOUR_IN_SECONDS
3235

3336
def main():
3437
with get_centml_client() as cclient:

0 commit comments

Comments
 (0)