sqlmodel 0.0.45__tar.gz → 0.0.46__tar.gz

{sqlmodel-0.0.45 → sqlmodel-0.0.46}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlmodel
-Version: 0.0.45
+Version: 0.0.46
 Summary: SQLModel, SQL databases in Python, designed for simplicity, compatibility, and robustness.
 Author-Email: =?utf-8?q?Sebasti=C3=A1n_Ram=C3=ADrez?= <tiangolo@gmail.com>
 License-Expression: MIT

{sqlmodel-0.0.45 → sqlmodel-0.0.46}/pyproject.toml

@@ -41,7 +41,7 @@ dependencies = [
     "pydantic>=2.11.0",
     "typing-extensions>=4.5.0",
 ]
-version = "0.0.45"
+version = "0.0.46"
 
 [project.urls]
 Homepage = "https://github.com/fastapi/sqlmodel"

{sqlmodel-0.0.45 → sqlmodel-0.0.46}/sqlmodel/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.0.45"
+__version__ = "0.0.46"
 
 # Re-export from Pydantic
 from pydantic import Discriminator as Discriminator

{sqlmodel-0.0.45 → sqlmodel-0.0.46}/sqlmodel/sql/sqltypes.py

@@ -1,8 +1,9 @@
-from datetime import datetime, timezone
+from datetime import datetime, timedelta, timezone
 from typing import Any, cast
 
 from sqlalchemy import types
 from sqlalchemy.engine.interfaces import Dialect
+from sqlalchemy.sql.operators import OperatorType
 
 
 class UTCDateTime(types.TypeDecorator[datetime]):
@@ -17,6 +18,13 @@ class UTCDateTime(types.TypeDecorator[datetime]):
     def __repr__(self) -> str:
         return "UTCDateTime()"
 
+    def coerce_compared_value(
+        self, op: OperatorType | None, value: Any
+    ) -> types.TypeEngine[Any]:
+        if isinstance(value, timedelta):
+            return types.Interval()
+        return self
+
     def process_bind_param(
         self, value: datetime | None, dialect: Dialect
     ) -> datetime | None:

{sqlmodel-0.0.45 → sqlmodel-0.0.46}/tests/test_datetime.py

@@ -1,5 +1,7 @@
+from collections.abc import Callable
 from datetime import date, datetime, time, timedelta, timezone, tzinfo
 from io import StringIO
+from operator import add, sub
 from typing import Annotated, Any
 
 import pytest
@@ -264,6 +266,34 @@ def test_postgresql_datetime_processing() -> None:
     assert result(None) is None
 
 
+@pytest.mark.parametrize(
+    "operation, reverse",
+    [(add, False), (sub, False), (add, True)],
+)
+def test_postgresql_datetime_interval_arithmetic(
+    operation: Callable[[Any, Any], Any], reverse: bool
+) -> None:
+    class Event(SQLModel, table=True):
+        id: int | None = Field(default=None, primary_key=True)
+        occurred_at: datetime
+
+    occurred_at = SQLModel.metadata.tables["event"].c.occurred_at
+    duration = timedelta(days=30)
+    expression = (
+        operation(duration, occurred_at)
+        if reverse
+        else operation(occurred_at, duration)
+    )
+    # Adding or subtracting a duration still produces a UTC datetime.
+    assert isinstance(expression.type, UTCDateTime)
+    # Exercise the actual parameter processor used for database execution.
+    parameter = expression.left if reverse else expression.right
+    dialect = postgresql.dialect()
+    bind = parameter.type.dialect_impl(dialect).bind_processor(dialect)
+    bound = bind(parameter.value) if bind is not None else parameter.value
+    assert bound == duration
+
+
 def test_existing_sqlite_utc_data() -> None:
     class Event(SQLModel, table=True):
         id: int | None = Field(default=None, primary_key=True)