string-schema 0.1.5__py3-none-any.whl → 0.1.6__py3-none-any.whl

string_schema/utilities.py

@@ -15,6 +15,7 @@ Key Functions:
 
 import functools
 import uuid
+from datetime import datetime, timezone
 from typing import Any, Dict, Type, Union, Callable, Optional, List
 import logging
 
@@ -30,6 +31,48 @@ except ImportError:
 logger = logging.getLogger(__name__)
 
 
+def _ensure_timezone_aware_dict(data: Dict[str, Any]) -> Dict[str, Any]:
+    """
+    Ensure all datetime values in a dictionary have timezone information.
+
+    This function recursively processes dictionaries and lists to add UTC timezone
+    information to naive datetime objects, ensuring consistent API responses.
+
+    Args:
+        data: Dictionary that may contain datetime values
+
+    Returns:
+        Dictionary with timezone-aware datetime values converted to ISO format
+    """
+    if not isinstance(data, dict):
+        return data
+
+    result = {}
+    for key, value in data.items():
+        if isinstance(value, datetime):
+            # Add UTC timezone to naive datetime objects
+            if value.tzinfo is None:
+                value = value.replace(tzinfo=timezone.utc)
+            # Convert to ISO format string for consistent API responses
+            result[key] = value.isoformat()
+        elif isinstance(value, dict):
+            # Recursively process nested dictionaries
+            result[key] = _ensure_timezone_aware_dict(value)
+        elif isinstance(value, list):
+            # Process lists that may contain dictionaries or datetime objects
+            result[key] = [
+                _ensure_timezone_aware_dict(item) if isinstance(item, dict)
+                else item.isoformat() if isinstance(item, datetime) and item.tzinfo is None
+                else item.replace(tzinfo=timezone.utc).isoformat() if isinstance(item, datetime)
+                else item
+                for item in value
+            ]
+        else:
+            result[key] = value
+
+    return result
+
+
 def string_to_model(schema_str: str, name: Optional[str] = None) -> Type[BaseModel]:
     """
     Create Pydantic model from string schema.
@@ -218,13 +261,21 @@ def validate_to_dict(data: Union[Dict[str, Any], Any], schema_str: str) -> Union
             try:
                 # Try Pydantic v2 RootModel style
                 validated_instance = TempModel(data)
-                # Return the validated array data
-                return validated_instance.model_dump() if hasattr(validated_instance, 'model_dump') else validated_instance.dict()
+                # Return the validated array data with timezone-aware conversion
+                result_data = validated_instance.model_dump() if hasattr(validated_instance, 'model_dump') else validated_instance.dict()
+                # Process array items for timezone-aware datetime conversion
+                if isinstance(result_data, list):
+                    return [_ensure_timezone_aware_dict(item) if isinstance(item, dict) else item for item in result_data]
+                return result_data
             except:
                 # Fallback to Pydantic v1 style
                 validated_instance = TempModel(__root__=data)
-                # Return the validated array data
-                return validated_instance.model_dump()['__root__'] if hasattr(validated_instance, 'model_dump') else validated_instance.dict()['__root__']
+                # Return the validated array data with timezone-aware conversion
+                result_data = validated_instance.model_dump()['__root__'] if hasattr(validated_instance, 'model_dump') else validated_instance.dict()['__root__']
+                # Process array items for timezone-aware datetime conversion
+                if isinstance(result_data, list):
+                    return [_ensure_timezone_aware_dict(item) if isinstance(item, dict) else item for item in result_data]
+                return result_data
         else:
             # Handle different input types for object schemas
             if isinstance(data, dict):
@@ -236,11 +287,14 @@ def validate_to_dict(data: Union[Dict[str, Any], Any], schema_str: str) -> Union
                 # Try direct validation
                 validated_instance = TempModel(data)
 
-            # Return as dictionary (use model_dump for Pydantic v2, fallback to dict for v1)
+            # Return as dictionary with timezone-aware datetime handling
             if hasattr(validated_instance, 'model_dump'):
-                return validated_instance.model_dump()
+                result_dict = validated_instance.model_dump()
             else:
-                return validated_instance.dict()
+                result_dict = validated_instance.dict()
+
+            # Ensure timezone-aware datetime conversion for consistent API responses
+            return _ensure_timezone_aware_dict(result_dict)
 
     except ValidationError as e:
         # Re-raise the original validation error

{string_schema-0.1.5.dist-info → string_schema-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: string-schema
-Version: 0.1.5
+Version: 0.1.6
 Summary: A simple, LLM-friendly schema definition library for converting string syntax to structured schemas
 Home-page: https://github.com/xychenmsn/string-schema
 Author: Michael Chen
@@ -151,6 +151,7 @@ String Schema takes human-readable text descriptions and converts them into stru
 - **🤖 LLM Data Extraction**: Define extraction schemas that LLMs can easily follow
 - **🔧 API Development**: Generate Pydantic models and OpenAPI docs from simple syntax
 - **✅ Data Validation**: Create robust validation schemas with minimal code
+- **🌍 Timezone-Aware APIs**: Automatic UTC conversion for consistent datetime handling
 - **📋 Configuration**: Define and validate application configuration schemas
 - **🔄 Data Transformation**: Convert between different schema formats
 
@@ -175,6 +176,12 @@ print(user_dict)  # {"name": "John Doe", "email": "john@example.com", "age": 25}
 user_model = validate_to_model(raw_data, "name:string, email:email, age:int?")
 print(user_model.name)  # "John Doe" - Full type safety
 print(user_model.age)   # 25 - Converted to int
+
+# 🌍 Automatic timezone handling for datetime fields
+from datetime import datetime
+event_data = {"name": "Meeting", "start_time": datetime(2025, 8, 13, 14, 30)}
+event_dict = validate_to_dict(event_data, "name:string, start_time:datetime")
+print(event_dict)  # {"name": "Meeting", "start_time": "2025-08-13T14:30:00+00:00"}
 ```
 
 ### 🎨 Function Decorators

{string_schema-0.1.5.dist-info → string_schema-0.1.6.dist-info}/RECORD

@@ -1,6 +1,6 @@
 string_schema/__init__.py,sha256=J0_B0TNO0AK_piEcT5gFFHllywjx16DTP59ps_Fp-WU,4052
 string_schema/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-string_schema/utilities.py,sha256=iLyqCKFd0qTL13xyZvt3Dia2-gjZCjHyDVVm59GxLK4,19918
+string_schema/utilities.py,sha256=FjWKNHPVMFxUc04ITtt2eC4bI6ayLDYwr44fnBgN0-o,22351
 string_schema/core/__init__.py,sha256=jeukkZ1WoCSXrBafumHVH2GkCJ6QjnkH_jAr3DESFio,481
 string_schema/core/builders.py,sha256=AHDTqtCnSe5ZN-DwedisqXqktjgl6nypd-mebucjI3k,7809
 string_schema/core/fields.py,sha256=iLUR-w3pZCr7OkQHhb2DFkVLtObnpyJ_mEUdc0qAqXY,5534
@@ -17,8 +17,8 @@ string_schema/parsing/__init__.py,sha256=dFW68DqJIwP4w_aYaZMjjVK15X7tCv8A0SPqmIg
 string_schema/parsing/optimizer.py,sha256=Ofte8Tb7sKmdEv7RrVIEBQ4Qqr-Whanp1vh8BakcOgw,8479
 string_schema/parsing/string_parser.py,sha256=-a_143fjNkxcRu24JdpqA6GltR7a_2AUXPNgHsbwYv4,24989
 string_schema/parsing/syntax.py,sha256=RO3BIAnWfDBupowOnoJocHtAe-kwE-SgRWXKknVwGdg,8900
-string_schema-0.1.5.dist-info/licenses/LICENSE,sha256=wCD2KBeSlVSkJy0apl6zmVj04uw97UJhRRi-en_3Qfk,1065
-string_schema-0.1.5.dist-info/METADATA,sha256=W8XGHUtUAz-SlPW1Mw1Y_SHy1SjuntWS9AolofnE64E,18246
-string_schema-0.1.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-string_schema-0.1.5.dist-info/top_level.txt,sha256=1uTmLPYIRrCDVxUW1fDFRMaWvGO48NRcGmbW4oq89I0,14
-string_schema-0.1.5.dist-info/RECORD,,
+string_schema-0.1.6.dist-info/licenses/LICENSE,sha256=wCD2KBeSlVSkJy0apl6zmVj04uw97UJhRRi-en_3Qfk,1065
+string_schema-0.1.6.dist-info/METADATA,sha256=L96Jgf9qVZ2K2LwLb5H7stwh684wy5vYFR2bze1yQII,18662
+string_schema-0.1.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+string_schema-0.1.6.dist-info/top_level.txt,sha256=1uTmLPYIRrCDVxUW1fDFRMaWvGO48NRcGmbW4oq89I0,14
+string_schema-0.1.6.dist-info/RECORD,,