Fix datetimes (#484)

project-koku · Feb 12, 2024 · 6595d0a · 6595d0a
1 parent 5c0b534
commit 6595d0a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/nise/__init__.py b/nise/__init__.py
@@ -1,3 +1,3 @@
-__version__ = "4.4.13"
+__version__ = "4.4.14"
 
 VERSION = __version__.split(".")
diff --git a/nise/__main__.py b/nise/__main__.py
@@ -741,6 +741,8 @@ def calculate_start_date(start_date):
         generated_start_date = today().replace(day=1, hour=0, minute=0, second=0) + relativedelta(months=-1)
     elif start_date == "today":
         generated_start_date = today().replace(hour=0, minute=0, second=0)
+    elif start_date and isinstance(start_date, datetime.datetime):
+        generated_start_date = start_date
     elif start_date and isinstance(start_date, datetime.date):
         generated_start_date = datetime.datetime(start_date.year, start_date.month, start_date.day)
     elif start_date:
@@ -759,6 +761,8 @@ def calculate_end_date(start_date, end_date):
             generated_end_date = today().replace(day=1, hour=0, minute=0, second=0) + relativedelta(months=-1)
         elif end_date == "today":
             generated_end_date = today().replace(hour=0, minute=0, second=0)
+        elif end_date and isinstance(end_date, datetime.datetime):
+            generated_end_date = end_date
         elif end_date and isinstance(end_date, datetime.date):
             generated_end_date = datetime.datetime(end_date.year, end_date.month, end_date.day)
         else: