fix one off error in until

enerBit · Feb 27, 2024 · 6b20c3a · 6b20c3a
1 parent 529784a
commit 6b20c3a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/enerbitdso/VERSION b/src/enerbitdso/VERSION
@@ -1 +1 @@
-0.1.7
+0.1.8
diff --git a/src/enerbitdso/enerbit.py b/src/enerbitdso/enerbit.py
@@ -116,12 +116,12 @@ def get_schedule_measurement_records(
 def fetch_schedule_usage_records_large_interval(
     client: httpx.Client, frt_code: str, since: dt.datetime, until: dt.datetime
 ) -> list[ScheduleUsageRecord]:
-    number_of_requests = math.floor((until - since) / MAX_REQUEST_RANGE)
+    number_of_requests = math.ceil((until - since) / MAX_REQUEST_RANGE)
     logger.debug(f"Fetching usages in {number_of_requests} requests")
     usage_records = []
     for i in range(0, number_of_requests):
         fi = since + i * MAX_REQUEST_RANGE
-        ff = fi + MAX_REQUEST_RANGE
+        ff = min(fi + MAX_REQUEST_RANGE, until)
         this_usage_records = get_schedule_usage_records(
             client, frt_code, since=fi, until=ff
         )