Add support for working with zoneinfo object field

axnsan12 · Dec 5, 2023 · 0c73a90 · 0c73a90
1 parent afea1bc
commit 0c73a90
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/drf_yasg/utils.py b/src/drf_yasg/utils.py
@@ -17,6 +17,15 @@
 
 from .app_settings import swagger_settings
 
+try:
+    import zoneinfo
+except ImportError:
+    try:
+        from backports import zoneinfo
+    except ImportError:
+        zoneinfo = None
+
+
 logger = logging.getLogger(__name__)
 
 
@@ -465,7 +474,10 @@ def field_value_to_representation(field, value):
         else:
             value = str(value)
 
-    if isinstance(value, pytz.BaseTzInfo):
+    elif isinstance(value, pytz.BaseTzInfo):
+        value = str(value)
+
+    elif zoneinfo is not None and isinstance(value, zoneinfo.ZoneInfo):
         value = str(value)
 
     # JSON roundtrip ensures that the value is valid JSON;