surrealdb-orm 0.5.3__py3-none-any.whl → 0.5.3.1__py3-none-any.whl

surreal_orm/model_base.py

@@ -1,6 +1,7 @@
-from typing import Any, Literal, Self, TYPE_CHECKING
+from datetime import datetime
+from typing import Any, Literal, Self, TYPE_CHECKING, get_args, get_origin
 
-from pydantic import BaseModel, ConfigDict, model_validator
+from pydantic import BaseModel, ConfigDict, PrivateAttr, model_validator
 
 from .connection_manager import SurrealDBConnectionManager
 from .types import SchemaMode, TableType
@@ -53,6 +54,36 @@ def _parse_record_id(record_id: Any) -> str | None:
     return record_str
 
 
+def _parse_datetime(value: Any) -> Any:
+    """
+    Parse a datetime value from SurrealDB.
+
+    SurrealDB returns datetime as ISO 8601 strings.
+    """
+    if value is None:
+        return None
+    if isinstance(value, datetime):
+        return value
+    if isinstance(value, str):
+        try:
+            # Try parsing ISO format (with or without timezone)
+            # SurrealDB returns format like "2026-02-02T13:21:23.641315924Z"
+            return datetime.fromisoformat(value.replace("Z", "+00:00"))
+        except ValueError:
+            pass
+    return value  # Return as-is if we can't parse
+
+
+def _is_datetime_field(field_type: Any) -> bool:
+    """Check if a field type is datetime or Optional[datetime]."""
+    # Handle Optional[datetime] which is Union[datetime, None]
+    origin = get_origin(field_type)
+    if origin is not None:
+        args = get_args(field_type)
+        return datetime in args
+    return field_type is datetime
+
+
 class SurrealConfigDict(ConfigDict):
     """
     SurrealConfigDict is a configuration dictionary for SurrealDB models.
@@ -109,6 +140,10 @@ class BaseSurrealModel(BaseModel):
             password: Encrypted
     """
 
+    # Private attribute to track if this instance has been persisted to the database.
+    # This helps distinguish between create (first save) and update (subsequent saves).
+    _db_persisted: bool = PrivateAttr(default=False)
+
     def __init_subclass__(cls, **kwargs: Any) -> None:
         """Register subclasses in the model registry for migration introspection."""
         super().__init_subclass__(**kwargs)
@@ -285,7 +320,9 @@ class BaseSurrealModel(BaseModel):
         if isinstance(record, list):
             return [cls.from_db(rs) for rs in record]  # type: ignore
 
-        return cls(**record)
+        instance = cls(**record)
+        instance._db_persisted = True
+        return instance
 
     @model_validator(mode="before")
     @classmethod
@@ -305,15 +342,26 @@ class BaseSurrealModel(BaseModel):
         This preserves the original __pydantic_fields_set__ so that exclude_unset=True
         continues to work correctly on subsequent saves.
 
+        Also handles type coercion for datetime fields (SurrealDB returns ISO strings).
+
         Args:
             record: Dictionary of field values from the database.
         """
         # Store original fields_set to preserve user-set tracking
         original_fields_set = self.__pydantic_fields_set__.copy()
 
+        # Get field type annotations for datetime parsing
+        field_types = self.__class__.model_fields
+
         for key, value in record.items():
             if key == "id":
                 value = _parse_record_id(value)
+            elif key in field_types:
+                # Check if this field is a datetime type and parse if needed
+                field_info = field_types[key]
+                if _is_datetime_field(field_info.annotation):
+                    value = _parse_datetime(value)
+
             if hasattr(self, key):
                 setattr(self, key, value)
 
@@ -321,6 +369,9 @@ class BaseSurrealModel(BaseModel):
         # DB-loaded fields should not be considered as "set" for exclude_unset
         object.__setattr__(self, "__pydantic_fields_set__", original_fields_set)
 
+        # Mark as persisted since we just loaded data from DB
+        self._db_persisted = True
+
     async def refresh(self) -> None:
         """
         Refresh the model instance from the database.
@@ -347,6 +398,11 @@ class BaseSurrealModel(BaseModel):
         """
         Save the model instance to the database.
 
+        For persisted records: uses merge() for partial update, only sending
+        explicitly set fields (preserving server-side values like timestamps).
+        For new records with ID: uses upsert() to create or fully replace.
+        For new records without ID: uses create() to auto-generate an ID.
+
         Args:
             tx: Optional transaction to use for this operation.
                 If provided, the operation will be part of the transaction.
@@ -361,37 +417,51 @@ class BaseSurrealModel(BaseModel):
         """
         # Build exclude set: always exclude 'id' and any server-generated fields
         exclude_fields = {"id"} | self.get_server_fields()
+        id = self.get_id()
+        table = self.get_table_name()
 
         if tx is not None:
             # Use transaction
             data = self.model_dump(exclude=exclude_fields, exclude_unset=True)
-            id = self.get_id()
-            table = self.get_table_name()
+
+            if self._db_persisted and id is not None:
+                # Already persisted: use merge for partial update
+                # Only sends explicitly set fields, preserving server-side values
+                thing = f"{table}:{id}"
+                await tx.merge(thing, data)
+                return self
 
             if id is not None:
-                # Use upsert for idempotent save (create or update)
+                # New record with user-provided ID: use upsert
                 thing = f"{table}:{id}"
                 await tx.upsert(thing, data)
-                return self
+            else:
+                # Auto-generate ID
+                await tx.create(table, data)
 
-            # Auto-generate ID - create without specific ID
-            await tx.create(table, data)
+            self._db_persisted = True
             return self
 
         # Original behavior without transaction
         client = await SurrealDBConnectionManager.get_client()
         data = self.model_dump(exclude=exclude_fields, exclude_unset=True)
-        id = self.get_id()
-        table = self.get_table_name()
+
+        if self._db_persisted and id is not None:
+            # Already persisted: use merge for partial update
+            # Only sends explicitly set fields, preserving server-side values
+            thing = f"{table}:{id}"
+            await client.merge(thing, data)
+            return self
 
         if id is not None:
-            # Use upsert for idempotent save (create or update)
+            # New record with user-provided ID: use upsert
             thing = f"{table}:{id}"
             await client.upsert(thing, data)
+            self._db_persisted = True
             return self
 
         # Auto-generate the ID
-        result = await client.create(table, data)  # pragma: no cover
+        result = await client.create(table, data)
 
         # SDK returns RecordResponse
         if not result.exists:
@@ -411,6 +481,9 @@ class BaseSurrealModel(BaseModel):
         """
         Update the model instance to the database.
 
+        Uses merge() to only update the specified fields, preserving
+        any fields that weren't explicitly set.
+
         Args:
             tx: Optional transaction to use for this operation.
         """
@@ -425,11 +498,13 @@ class BaseSurrealModel(BaseModel):
         thing = f"{self.__class__.__name__}:{id}"
 
         if tx is not None:
-            await tx.update(thing, data)
+            # Use merge for partial update - preserves unspecified fields
+            await tx.merge(thing, data)
             return None
 
         client = await SurrealDBConnectionManager.get_client()
-        result = await client.update(thing, data)
+        # Use merge for partial update - preserves unspecified fields
+        result = await client.merge(thing, data)
         return result.records
 
     @classmethod

{surrealdb_orm-0.5.3.dist-info → surrealdb_orm-0.5.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: surrealdb-orm
-Version: 0.5.3
+Version: 0.5.3.1
 Summary: SurrealDB ORM as 'DJango style' for Python with async support. Works with pydantic validation.
 Project-URL: Homepage, https://github.com/EulogySnowfall/SurrealDB-ORM
 Project-URL: Documentation, https://github.com/EulogySnowfall/SurrealDB-ORM
@@ -68,9 +68,15 @@ Description-Content-Type: text/markdown
 
 ## What's New in 0.5.x
 
+### v0.5.3.1 - Bug Fixes
+
+- **Partial updates for persisted records** - `save()` now uses `merge()` for already-persisted records, only sending modified fields
+- **datetime parsing** - `_update_from_db()` now parses ISO 8601 strings to `datetime` objects automatically
+- **`_db_persisted` flag** - Internal tracking to distinguish new vs persisted records
+
 ### v0.5.3 - ORM Improvements
 
-- **Upsert save behavior** - `save()` now uses `upsert` for existing records (idempotent, Django-like)
+- **Upsert save behavior** - `save()` now uses `upsert` for new records with ID (idempotent, Django-like)
 - **`server_fields` config** - Exclude server-generated fields (created_at, updated_at) from saves
 - **`merge()` returns self** - Now returns the updated model instance instead of None
 - **`save()` updates self** - Updates original instance attributes instead of returning new object

{surrealdb_orm-0.5.3.dist-info → surrealdb_orm-0.5.3.1.dist-info}/RECORD

@@ -3,7 +3,7 @@ surreal_orm/aggregations.py,sha256=5ERMHMWQfaW76OrMNazMjyg7dbf9bJ3GX_8QWz6tfxY,3
 surreal_orm/connection_manager.py,sha256=VJVfsuUXK5OEG0rtuiu3DKbyWxkkeFrCEtzjXFd6lAw,10496
 surreal_orm/constants.py,sha256=CLavEca1M6cLJLqVl4l4KoE-cBrgVQNsuGxW9zGJBmg,429
 surreal_orm/enum.py,sha256=kR-vzkHqnqy9YaYOvWTwAHdl2-WCzPcSEch-YTyJv1Y,158
-surreal_orm/model_base.py,sha256=yXFsCskiCkmqpWtpNpYWy-MIYsrQvovd71jHMgzUNG4,25652
+surreal_orm/model_base.py,sha256=mbU9zEDBN2Sty9c__JyVVFUJ_g_rkNrwxrKUZtDsiDg,28649
 surreal_orm/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 surreal_orm/query_set.py,sha256=3nmQ9A13g-1EJlRZsJQX8z7yFO3pjI_6MscFT3W4Lm0,38890
 surreal_orm/relations.py,sha256=pbsbKz2jw1h9FdFQowidUkx8krYv-l3OreQEViasN5w,21137
@@ -45,8 +45,8 @@ surreal_sdk/streaming/__init__.py,sha256=TljF9HFN-XOshK_1smmTanF68hcdNsTgdlHtfFo
 surreal_sdk/streaming/change_feed.py,sha256=lS6CGNinsMkzslf1y0aV0VgZ2gq01EDIEZrnvvYhv-E,8540
 surreal_sdk/streaming/live_query.py,sha256=QwPXmRIsH0j3jgGQY9ULJU7MB0eTk8dnOVbPzy4SeSo,7561
 surreal_sdk/streaming/live_select.py,sha256=mYg6NKMx3GPShg_hYz4SjjDwhG7baI3t_Gynv8YCBNE,12035
-surrealdb_orm-0.5.3.dist-info/METADATA,sha256=qCZAOpLAFD3y7PY_o8HJ2fgLldLm4wxE8E_G2nANl6k,17102
-surrealdb_orm-0.5.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-surrealdb_orm-0.5.3.dist-info/entry_points.txt,sha256=3zbp1VzVPSxFrxNuvKZAyt9U432QITdBNjp5QnMZ2o8,62
-surrealdb_orm-0.5.3.dist-info/licenses/LICENSE,sha256=OYVJQ7TKsjHAgVndePmKcMWOZkxwJxOHiXOG1TAnCk4,1079
-surrealdb_orm-0.5.3.dist-info/RECORD,,
+surrealdb_orm-0.5.3.1.dist-info/METADATA,sha256=PjZ0BlUAAnqXOJPmIX1spcFq8C_6IV0x35xHpcJH9eo,17464
+surrealdb_orm-0.5.3.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+surrealdb_orm-0.5.3.1.dist-info/entry_points.txt,sha256=3zbp1VzVPSxFrxNuvKZAyt9U432QITdBNjp5QnMZ2o8,62
+surrealdb_orm-0.5.3.1.dist-info/licenses/LICENSE,sha256=OYVJQ7TKsjHAgVndePmKcMWOZkxwJxOHiXOG1TAnCk4,1079
+surrealdb_orm-0.5.3.1.dist-info/RECORD,,