Merge pull request #157 from fccn/152-end-date-inclusive

Shared revenue end date should be inclusive #152
fccn · Nov 2, 2023 · 56ef048 · 56ef048
2 parents 57b24d0 + fa74181
commit 56ef048
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/apps/shared_revenue/management/commands/export_split_revenue.py b/apps/shared_revenue/management/commands/export_split_revenue.py
@@ -1,5 +1,5 @@
 import time
-from datetime import datetime
+from datetime import datetime, timedelta
 
 from django.core.management.base import BaseCommand, CommandError, CommandParser
 
@@ -38,7 +38,9 @@ def handle(self, *args, **options) -> str | None:
             start = time.time()
             self.stdout.write("\nStarting file export...\n")
             start_date = datetime.strptime(options["start_date"], "%d/%m/%Y")
-            end_date = datetime.strptime(options["end_date"], "%d/%m/%Y")
+            end_date = datetime.strptime(options["end_date"], "%d/%m/%Y") + (
+                timedelta(days=1) - timedelta(milliseconds=1)
+            )
             product_id = options.get("product_id")
             organization_code = options.get("organization_code")
             kwargs = {