TypeDAL 4.6.2__tar.gz → 4.6.4__tar.gz

{typedal-4.6.2 → typedal-4.6.4}/CHANGELOG.md

@@ -2,6 +2,18 @@
 
 <!--next-version-placeholder-->
 
+## v4.6.4 (2026-03-30)
+
+### Fix
+
+* Add collect_into to allow instantiating results into different model classes + fix render ([#8](https://github.com/trialandsuccess/TypeDAL/issues/8)) ([`bc5b5b7`](https://github.com/trialandsuccess/TypeDAL/commit/bc5b5b78ce033b068c21f468932a4b2f3adfeeac))
+
+## v4.6.3 (2026-03-19)
+
+### Fix
+
+* `as_dict()` and similar changed behavior after doing `.render()` ([`8241503`](https://github.com/trialandsuccess/TypeDAL/commit/8241503073c5a9fd76cc898048f6c7796de3ce7a))
+
 ## v4.6.2 (2026-03-13)
 
 ### Fix

{typedal-4.6.2 → typedal-4.6.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: TypeDAL
-Version: 4.6.2
+Version: 4.6.4
 Summary: Typing support for PyDAL
 Project-URL: Documentation, https://typedal.readthedocs.io/
 Project-URL: Issues, https://github.com/trialandsuccess/TypeDAL/issues

{typedal-4.6.2 → typedal-4.6.4}/src/typedal/__about__.py

@@ -5,4 +5,4 @@ This file contains the Version info for this package.
 # SPDX-FileCopyrightText: 2023-present Robin van der Noord <robinvandernoord@gmail.com>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "4.6.2"
+__version__ = "4.6.4"

{typedal-4.6.2 → typedal-4.6.4}/src/typedal/query_builder.py

@@ -33,6 +33,7 @@ from .types import (
     OnQuery,
     OrderBy,
     Query,
+    Row,
     Rows,
     SelectKwargs,
     T_MetaInstance,
@@ -517,7 +518,11 @@ class QueryBuilder[T_MetaInstance: _TypedTable]:
         """
         return self.to_sql()
 
-    def _collect_cached(self, metadata: Metadata) -> "TypedRows[T_MetaInstance] | None":
+    def _collect_cached(
+        self,
+        metadata: Metadata,
+        into: t.Type[_TypedTable],
+    ) -> "TypedRows[T_MetaInstance] | None":
         expires_at = metadata["cache"].get("expires_at")
         metadata["cache"] |= {
             # key is partly dependant on cache metadata but not these:
@@ -529,6 +534,7 @@ class QueryBuilder[T_MetaInstance: _TypedTable]:
 
         _, key = create_and_hash_cache_key(
             self.model,
+            f"{into.__module__}.{into.__qualname__}",
             metadata,
             self.query,
             self.select_args,
@@ -566,12 +572,15 @@ class QueryBuilder[T_MetaInstance: _TypedTable]:
         verbose: bool = False,
         _to: t.Type["TypedRows[t.Any]"] = None,
         add_id: bool = True,
+        _into: t.Type[_TypedTable] | None = None,
+        _init: t.Callable[[_TypedTable, Row], None] | None = None,
     ) -> "TypedRows[T_MetaInstance]":
         """
         Execute the built query and turn it into model instances, while handling relationships.
         """
         if _to is None:
             _to = TypedRows
+        into = _into or self.model
 
         if not isinstance(self.model, TableMeta):
             # tried to use querybuilder with a non-typedal table,
@@ -585,7 +594,7 @@ class QueryBuilder[T_MetaInstance: _TypedTable]:
 
         metadata: Metadata = self.metadata.copy()
 
-        if metadata.get("cache", {}).get("enabled") and (result := self._collect_cached(metadata)):
+        if metadata.get("cache", {}).get("enabled") and (result := self._collect_cached(metadata, into)):
             return result
 
         query, select_args, select_kwargs = self._before_query(metadata, add_id=add_id)
@@ -609,12 +618,12 @@ class QueryBuilder[T_MetaInstance: _TypedTable]:
 
         if not self.relationships:
             # easy
-            typed_rows = _to.from_rows(rows, self.model, metadata=metadata)
+            typed_rows = _to.from_rows(rows, self.model, metadata=metadata, into=into, init=_init)
         else:
             # harder: try to match rows to the belonging objects
             # assume structure of {'table': <data>} per row.
             # if that's not the case, return default behavior again
-            typed_rows = self._collect_with_relationships(rows, metadata=metadata, _to=_to)
+            typed_rows = self._collect_with_relationships(rows, metadata=metadata, _to=_to, _into=into, _init=_init)
 
         for fn_after in db._after_collect:
             fn_after(self, typed_rows, rows)
@@ -622,6 +631,35 @@ class QueryBuilder[T_MetaInstance: _TypedTable]:
         # only saves if requested in metadata:
         return save_to_cache(typed_rows, rows)
 
+    def collect_into[T_Into: _TypedTable](
+        self,
+        into: t.Type[T_Into],
+        verbose: bool = False,
+        add_id: bool = True,
+        init: t.Callable[[T_Into, Row], None] | None = None,
+    ) -> "TypedRows[T_Into]":
+        """
+        Execute the built query and instantiate root records as another model class.
+        """
+        self._validate_collect_into_model(into)
+        _init = t.cast(t.Callable[[_TypedTable, Row], None] | None, init)
+        rows = self.collect(verbose=verbose, add_id=add_id, _into=into, _init=_init)
+        return t.cast("TypedRows[T_Into]", rows)
+
+    def _validate_collect_into_model(self, into: t.Type[t.Any]) -> None:
+        if not isinstance(into, TableMeta):
+            raise TypeError("collect_into expects a TypedTable class")
+
+        source = self.model._ensure_table_defined()
+        target = into._ensure_table_defined()
+
+        if source is target or str(source) == str(target):
+            return
+
+        raise ValueError(
+            f"collect_into target '{into.__name__}' must be bound to table '{source}', got '{target}'",
+        )
+
     @t.overload
     def column[T: t.Any](self, field: TypedField[T], **options: t.Unpack[SelectKwargs]) -> list[T]:
         """
@@ -843,12 +881,15 @@ class QueryBuilder[T_MetaInstance: _TypedTable]:
         rows: Rows,
         metadata: Metadata,
         _to: t.Type["TypedRows[T_MetaInstance]"],
+        _into: t.Type[_TypedTable] | None = None,
+        _init: t.Callable[[_TypedTable, Row], None] | None = None,
     ) -> "TypedRows[T_MetaInstance]":
         """
         Transform the raw rows into Typed Table model instances with nested relationships.
         """
         db = self._get_db()
         main_table = self._ensure_table_defined()
+        into = _into or self.model
 
         # id: Model
         records: dict[t.Any, T_MetaInstance] = {}
@@ -866,7 +907,9 @@ class QueryBuilder[T_MetaInstance: _TypedTable]:
             raw_per_id[main_id].append(normalize_table_keys(row))
 
             if main_id not in records:
-                records[main_id] = self.model(main)
+                records[main_id] = t.cast(T_MetaInstance, into(main))
+                if _init:
+                    _init(t.cast(_TypedTable, records[main_id]), row)
                 records[main_id]._with = list(self.relationships.keys())
 
                 # Setup all relationship defaults (once)

{typedal-4.6.2 → typedal-4.6.4}/src/typedal/rows.py

@@ -62,21 +62,7 @@ class TypedRows(t.Collection[T_MetaInstance], Rows):
         `model` is a Typed Table class
         """
 
-        def _get_id(row: Row) -> int:
-            """
-            Try to find the id field in a row.
-
-            If _extra exists, the row changes:
-            <Row {'test_relationship': {'id': 1}, '_extra': {'COUNT("test_relationship"."querytable")': 8}}>
-            """
-            if idx := getattr(row, "id", None):
-                return t.cast(int, idx)
-            elif main := getattr(row, str(model), None):
-                return t.cast(int, main.id)
-            else:  # pragma: no cover
-                raise NotImplementedError(f"`id` could not be found for {row}")
-
-        records = records or {_get_id(row): model(row) for row in rows}
+        records = records or {self._get_id(row, model): model(row) for row in rows}
         raw = raw or {}
 
         for idx, entity in records.items():
@@ -87,6 +73,21 @@ class TypedRows(t.Collection[T_MetaInstance], Rows):
         self.metadata = metadata or {}
         self.colnames = rows.colnames
 
+    @staticmethod
+    def _get_id(row: Row, model: t.Type[t.Any]) -> int:
+        """
+        Try to find the id field in a row.
+
+        If _extra exists, the row changes:
+        <Row {'test_relationship': {'id': 1}, '_extra': {'COUNT("test_relationship"."querytable")': 8}}>
+        """
+        if idx := getattr(row, "id", None):
+            return t.cast(int, idx)
+        elif main := getattr(row, str(model), None):
+            return t.cast(int, main.id)
+        else:  # pragma: no cover
+            raise NotImplementedError(f"`id` could not be found for {row}")
+
     def __len__(self) -> int:
         """
         Return the count of rows.
@@ -374,11 +375,22 @@ class TypedRows(t.Collection[T_MetaInstance], Rows):
         rows: Rows,
         model: t.Type[T_MetaInstance],
         metadata: Metadata = None,
+        into: t.Type[_TypedTable] | None = None,
+        init: t.Callable[[_TypedTable, Row], None] | None = None,
     ) -> "TypedRows[T_MetaInstance]":
         """
         Internal method to convert a Rows object to a TypedRows.
         """
-        return cls(rows, model, metadata=metadata)
+        target_model = into or model
+
+        def build(row: Row) -> T_MetaInstance:
+            instance = t.cast(T_MetaInstance, target_model(row))
+            if init:
+                init(instance, row)
+            return instance
+
+        records = {cls._get_id(row, model): build(row) for row in rows}
+        return cls(rows, model, records=records, metadata=metadata)
 
     def __getstate__(self) -> AnyDict:
         """

{typedal-4.6.2 → typedal-4.6.4}/src/typedal/tables.py

@@ -386,6 +386,17 @@ class TableMeta(type):
         """
         return QueryBuilder(self).collect(verbose=verbose)
 
+    def collect_into[T_Into: _TypedTable](
+        self: t.Type[_TypedTable],
+        into: t.Type[T_Into],
+        verbose: bool = False,
+        init: t.Callable[[T_Into, Row], None] | None = None,
+    ) -> "TypedRows[T_Into]":
+        """
+        See QueryBuilder.collect_into!
+        """
+        return QueryBuilder(self).collect_into(into=into, verbose=verbose, init=init)
+
     @property
     def ALL(cls) -> pydal.objects.SQLALL:
         """
@@ -1104,6 +1115,7 @@ class TypedTable(_TypedTable, metaclass=TableMeta):
         # then create a new (more empty) row object:
         state["_row"] = Row(json.loads(state["_row"]))
         self.__dict__ |= state
+        self._setup_instance_methods()
 
     @classmethod
     def _sql(cls) -> str:
@@ -1152,7 +1164,9 @@ class TypedTable(_TypedTable, metaclass=TableMeta):
 
                 relation_row = row[relation_name]
 
-                if isinstance(relation_row, list):
+                if relation_row is None:
+                    row[relation_name] = None
+                elif isinstance(relation_row, list):
                     # list of rows
                     combined = []
 

{typedal-4.6.2 → typedal-4.6.4}/tests/test_query_builder.py

@@ -28,6 +28,10 @@ class TestRelationship(TypedTable):
     querytable: TestQueryTable
 
 
+class TestQueryTableBound(TestQueryTable):
+    pass
+
+
 class Undefined(TypedTable):
     value: int
 
@@ -210,6 +214,60 @@ def test_select():
     assert other.value
 
 
+def test_collect_into():
+    _setup_data()
+
+    rows = TestQueryTable.where(lambda row: row.number < 3).collect_into(TestQueryTableBound)
+    first = rows.first()
+
+    assert first
+    assert isinstance(first, TestQueryTableBound)
+    assert rows.model is TestQueryTable
+
+    joined = TestQueryTable.join("relations").where(id=1).collect_into(TestQueryTableBound)
+    joined_first = joined.first()
+
+    assert joined_first
+    assert isinstance(joined_first, TestQueryTableBound)
+    assert isinstance(joined_first.relations[0], TestRelationship)
+
+    marker = object()
+
+    def bind(sticker: TestQueryTableBound, _row):
+        sticker.item = marker
+
+    bound_rows = TestQueryTable.where(lambda row: row.number < 3).collect_into(TestQueryTableBound, init=bind)
+    assert all(getattr(row, "item", None) is marker for row in bound_rows)
+
+    calls: list[int] = []
+
+    def bind_once_per_root(sticker: TestQueryTableBound, _row):
+        calls.append(sticker.id)
+
+    TestQueryTable.join("relations").where(id=1).collect_into(TestQueryTableBound, init=bind_once_per_root)
+    assert calls == [1]
+
+    with pytest.raises(TypeError):
+        TestQueryTable.collect_into(dict)  # type: ignore[arg-type]
+
+    with pytest.raises(ValueError):
+        TestQueryTable.collect_into(TestRelationship)
+
+
+def test_collect_into_cache_isolation():
+    _setup_data()
+
+    regular = TestQueryTable.where(id=1).cache().collect()
+    remapped_fresh = TestQueryTable.where(id=1).cache().collect_into(TestQueryTableBound)
+    remapped_cached = TestQueryTable.where(id=1).cache().collect_into(TestQueryTableBound)
+
+    assert regular.metadata["cache"]["status"] == "fresh"
+    assert remapped_fresh.metadata["cache"]["status"] == "fresh"
+    assert remapped_cached.metadata["cache"]["status"] == "cached"
+
+    assert isinstance(remapped_cached.first(), TestQueryTableBound)
+
+
 def test_paginate():
     _setup_data()
 

{typedal-4.6.2 → typedal-4.6.4}/tests/test_row.py

@@ -266,6 +266,15 @@ def test_render():
     assert rendered_two.normal == "123"
     assert rendered_two.list_field == "abc, def"
     assert rendered_two.related.also_normal == "321"
+    assert json.loads(rendered_two.as_json()) == {
+        "id": rendered_two.id,
+        "normal": "123",
+        "list_field": "abc, def",
+        "related": {
+            "id": rendered_two.related.id,
+            "also_normal": "321",
+        },
+    }
 
     # test list:
 
@@ -287,3 +296,39 @@ def test_render():
     assert rendered_four.normal == "123"
     assert rendered_four.list_field == "abc, def"
     assert rendered_four.related_list[0].also_normal == "321"
+    assert json.loads(rendered_four.as_json()) == {
+        "id": rendered_four.id,
+        "normal": "123",
+        "list_field": "abc, def",
+        "related_list": [
+            {
+                "id": rendered_four.related_list[0].id,
+                "also_normal": "321",
+            }
+        ],
+    }
+
+
+def test_render_with_none_single_relationship_row():
+    @db.define()
+    class RelatedTableNone(TypedTable):
+        value: str
+
+    @db.define()
+    class RenderTableNone(TypedTable):
+        normal: str
+        related = relationship(
+            RelatedTableNone,
+            condition=lambda this, that: this.normal == that.value,
+        )
+
+    RenderTableNone.insert(normal="no-match")
+
+    row = RenderTableNone.select().join("related").first()
+
+    assert row
+    assert row.related is None
+
+    rendered = row.render()
+
+    assert rendered.related is None

typedal-4.6.2/.crush/.gitignore

@@ -1 +0,0 @@
-*

typedal-4.6.2/.crush/logs/crush.log

@@ -1,19 +0,0 @@
-{"time":"2026-03-13T20:12:28.275088774+01:00","level":"INFO","source":{"function":"github.com/charmbracelet/crush/internal/config.(*catwalkSync).Get.func1","file":"github.com/charmbracelet/crush/internal/config/catwalk.go","line":55},"msg":"Fetching providers from Catwalk"}
-{"time":"2026-03-13T20:12:28.486572096+01:00","level":"INFO","source":{"function":"github.com/charmbracelet/crush/internal/config.(*catwalkSync).Get.func1","file":"github.com/charmbracelet/crush/internal/config/catwalk.go","line":63},"msg":"Catwalk providers not modified"}
-{"time":"2026-03-13T20:12:28.486667656+01:00","level":"WARN","source":{"function":"github.com/charmbracelet/crush/internal/config.(*Config).configureProviders","file":"github.com/charmbracelet/crush/internal/config/load.go","line":296},"msg":"Skipping provider due to missing API key","provider":"anthropic"}
-{"time":"2026-03-13T20:12:28.497862335+01:00","level":"INFO","msg":"OK   20250424200609_initial.sql (1.55ms)"}
-{"time":"2026-03-13T20:12:28.498184106+01:00","level":"INFO","msg":"OK   20250515105448_add_summary_message_id.sql (300.3µs)"}
-{"time":"2026-03-13T20:12:28.498510046+01:00","level":"INFO","msg":"OK   20250624000000_add_created_at_indexes.sql (298.08µs)"}
-{"time":"2026-03-13T20:12:28.498811912+01:00","level":"INFO","msg":"OK   20250627000000_add_provider_to_messages.sql (289.51µs)"}
-{"time":"2026-03-13T20:12:28.499108738+01:00","level":"INFO","msg":"OK   20250810000000_add_is_summary_message.sql (245.48µs)"}
-{"time":"2026-03-13T20:12:28.499362065+01:00","level":"INFO","msg":"OK   20250812000000_add_todos_to_sessions.sql (241.95µs)"}
-{"time":"2026-03-13T20:12:28.499726903+01:00","level":"INFO","msg":"OK   20260127000000_add_read_files_table.sql (351.05µs)"}
-{"time":"2026-03-13T20:12:28.499731494+01:00","level":"INFO","msg":"goose: successfully migrated database to version: 20260127000000"}
-{"time":"2026-03-13T20:12:28.50106094+01:00","level":"INFO","source":{"function":"github.com/charmbracelet/crush/internal/agent/tools/mcp.Initialize","file":"github.com/charmbracelet/crush/internal/agent/tools/mcp/init.go","line":167},"msg":"Initializing MCP clients"}
-{"time":"2026-03-13T20:12:31.915239786+01:00","level":"INFO","source":{"function":"github.com/charmbracelet/crush/internal/config.(*catwalkSync).Get.func1","file":"github.com/charmbracelet/crush/internal/config/catwalk.go","line":55},"msg":"Fetching providers from Catwalk"}
-{"time":"2026-03-13T20:12:32.150078123+01:00","level":"INFO","source":{"function":"github.com/charmbracelet/crush/internal/config.(*catwalkSync).Get.func1","file":"github.com/charmbracelet/crush/internal/config/catwalk.go","line":63},"msg":"Catwalk providers not modified"}
-{"time":"2026-03-13T20:12:32.150303604+01:00","level":"WARN","source":{"function":"github.com/charmbracelet/crush/internal/config.(*Config).configureProviders","file":"github.com/charmbracelet/crush/internal/config/load.go","line":296},"msg":"Skipping provider due to missing API key","provider":"anthropic"}
-{"time":"2026-03-13T20:12:32.15507598+01:00","level":"INFO","msg":"goose: no migrations to run. current version: 20260127000000"}
-{"time":"2026-03-13T20:12:32.161646966+01:00","level":"INFO","source":{"function":"github.com/charmbracelet/crush/internal/agent/tools/mcp.Initialize","file":"github.com/charmbracelet/crush/internal/agent/tools/mcp/init.go","line":167},"msg":"Initializing MCP clients"}
-{"time":"2026-03-13T20:15:20.65020982+01:00","level":"ERROR","source":{"function":"github.com/charmbracelet/x/powernap/pkg/lsp.startServerProcess.func1","file":"github.com/charmbracelet/x/powernap@v0.1.3/pkg/lsp/client.go","line":650},"msg":"Language server stderr","command":"ty","output":"2026-03-13 20:15:20.650149482  INFO Version: 0.0.20\n"}
-{"time":"2026-03-13T20:15:20.651255948+01:00","level":"ERROR","source":{"function":"github.com/charmbracelet/x/powernap/pkg/lsp.startServerProcess.func1","file":"github.com/charmbracelet/x/powernap@v0.1.3/pkg/lsp/client.go","line":650},"msg":"Language server stderr","command":"ty","output":"2026-03-13 20:15:20.651183033 ERROR Invalid client response: did not contain a result or error (method=workspace/configuration)\n"}