possibilita occurrences receber datetime ou date nos parametros de data

crossfire/clients/occurrences.py

@@ -1,6 +1,6 @@
 import re
 from asyncio import Semaphore, gather, sleep
-from datetime import datetime
+from datetime import date, datetime
 from urllib.parse import urlencode
 
 from httpx import ReadTimeout
@@ -30,12 +30,16 @@
 NOT_NUMBER = re.compile("\D")
 
 
-def date_formatter(date):
-    date_cleaned = re.sub(NOT_NUMBER, "", date)
+def date_formatter(date_parameter):
+    if isinstance(date_parameter, datetime):
+        return date_parameter.date()
+    elif isinstance(date_parameter, date):
+        return date_parameter
+    date_cleaned = re.sub(NOT_NUMBER, "", date_parameter)
     try:
         date_cleaned = datetime.strptime(date_cleaned, "%Y%m%d").date()
     except ValueError:
-        raise DateFormatError(date)
+        raise DateFormatError(date_parameter)
     return date_cleaned
 
 

tests/test_occurrences_client.py

@@ -213,3 +213,17 @@ def test_date_formatter_with_correct_date_format_doted():
     formated_date = date_formatter("2023.01.23")
     assert isinstance(formated_date, datetime.date)
     assert str(formated_date) == "2023-01-23"
+
+
+def test_date_formatter_with_python_date_format():
+    date = datetime.datetime(2023, 1, 23).date()
+    formated_date = date_formatter(date)
+    assert isinstance(formated_date, datetime.date)
+    assert str(formated_date) == "2023-01-23"
+
+
+def test_date_formatter_with_python_datetime_format():
+    date = datetime.datetime(2023, 1, 23)
+    formated_date = date_formatter(date)
+    assert isinstance(formated_date, datetime.date)
+    assert str(formated_date) == "2023-01-23"