feat: date range widget

src/unfold/admin.py

@@ -33,6 +33,7 @@
 from .widgets import (
     SELECT_CLASSES,
     UnfoldAdminBigIntegerFieldWidget,
+    UnfoldAdminDateTimeRangeWidget,
     UnfoldAdminDecimalFieldWidget,
     UnfoldAdminEmailInputWidget,
     UnfoldAdminFileFieldWidget,
@@ -57,7 +58,11 @@
 )
 
 try:
-    from django.contrib.postgres.fields import ArrayField, IntegerRangeField
+    from django.contrib.postgres.fields import (
+        ArrayField,
+        DateTimeRangeField,
+        IntegerRangeField,
+    )
     from django.contrib.postgres.search import SearchVectorField
 
     HAS_PSYCOPG = True
@@ -102,6 +107,7 @@
             ArrayField: {"widget": UnfoldAdminTextareaWidget},
             SearchVectorField: {"widget": UnfoldAdminTextareaWidget},
             IntegerRangeField: {"widget": UnfoldAdminIntegerRangeWidget},
+            DateTimeRangeField: {"widget": UnfoldAdminDateTimeRangeWidget},
         }
     )
 

src/unfold/widgets.py

@@ -1,3 +1,4 @@
+from datetime import datetime
 from typing import Any, Callable, Dict, Optional, Tuple, Union
 
 from django.contrib.admin.options import VERTICAL
@@ -22,6 +23,7 @@
 from django.forms import (
     CheckboxInput,
     CheckboxSelectMultiple,
+    DateTimeInput,
     MultiWidget,
     NullBooleanSelect,
     NumberInput,
@@ -297,6 +299,26 @@ def decompress(self, value: Union[str, None]) -> Tuple[Optional[Callable], ...]:
         return None, None
 
 
+class UnfoldAdminDateTimeRangeWidget(MultiWidget):
+    template_name = "unfold/widgets/range.html"
+
+    def __init__(self, attrs: Optional[Dict[str, Any]] = None) -> None:
+        if attrs is None:
+            attrs = {}
+
+        attrs["class"] = " ".join(INPUT_CLASSES)
+        attrs["type"] = "datetime-local"
+
+        _widgets = (DateTimeInput(attrs=attrs), DateTimeInput(attrs=attrs))
+
+        super().__init__(_widgets, attrs)
+
+    def decompress(self, value: Union[str, None]) -> Tuple[Optional[datetime], ...]:
+        if value:
+            return value.lower, value.upper
+        return None, None
+
+
 class UnfoldAdminEmailInputWidget(AdminEmailInputWidget):
     def __init__(self, attrs: Optional[Dict[str, Any]] = None) -> None:
         super().__init__(attrs={"class": " ".join(INPUT_CLASSES), **(attrs or {})})