Cleanup for class meta failing with Table(query=Query(...)) provided (#632)

Author: jlubcke

iommi/declarative/util.py

@@ -44,21 +44,16 @@ def inject_args(args, kwargs, extra_args, pos_arg_names, merge_namespaces):
     else:
         new_args = args
 
-    for k, v in kwargs.items():
-        new_value = new_kwargs.get(k, None)
-        if merge_namespaces and isinstance(new_value, Namespace):
-            if v is not None:
-                new_kwargs[k] = Namespace(new_value, v)
-            else:
-                new_kwargs[k] = None
-        else:
-            new_kwargs[k] = v
+    if merge_namespaces:
+        new_kwargs = Namespace(new_kwargs, kwargs)
+    else:
+        new_kwargs.update(kwargs)
 
     return new_args, new_kwargs
 
 
 def strip_prefix(s, *, prefix, strict=False):
     if s.startswith(prefix):
-        return s[len(prefix):]
+        return s[len(prefix) :]
     assert strict is False, f"String '{s}' does not start with prefix '{prefix}'"
     return s

iommi/table.py

@@ -1636,6 +1636,13 @@ class TableAutoConfig(AutoConfig):
 
     rows = Refinable()
 
+def endpoint__tbody(table, **_):
+    return {
+        'html': table.container.__html__(
+            render=lambda fragment, context: fragment.render_text_or_children(context=context)
+        )
+    }
+
 
 def endpoint__csv(table, **_):
     from datetime import timezone
@@ -1780,8 +1787,6 @@ class Meta:
     columns: Dict[str, Column] = RefinableMembers()
 
     class Meta:
-        query__advanced__assets__query_form_toggle_script__template = "iommi/query/form_toggle_script.html"
-        assets__table_js_select_all__template = "iommi/table/js_select_all.html"
         member_class = Column
         form_class = Form
         query_class = Query
@@ -1790,24 +1795,6 @@ class Meta:
         cells_class = Cells
         row_group_class = RowGroup
 
-        @staticmethod
-        def endpoints__tbody__func(table, **_):
-            return {
-                'html': table.container.__html__(
-                    render=lambda fragment, context: fragment.render_text_or_children(context=context)
-                )
-            }
-
-        endpoints__csv__func = endpoint__csv
-
-        container__attrs = Namespace(
-            {
-                'data-endpoint': lambda table, **_: DISPATCH_PREFIX + table.endpoints.tbody.iommi_path,
-                'data-iommi-id': lambda table, **_: table.iommi_path,
-            }
-        )
-
-        query__form__attrs = {'data-iommi-id-of-table': lambda table, **_: table.iommi_path}
         columns = EMPTY
         parts = EMPTY
         row__attrs__class = EMPTY
@@ -1875,6 +1862,17 @@ def post_bulk_edit(table, queryset, updates, **_):
         # style
         query__form__actions__submit__call_target=Action.button,
         title=MISSING,
+        endpoints__tbody__func=endpoint__tbody,
+        endpoints__csv__func=endpoint__csv,
+        query__advanced__assets__query_form_toggle_script__template = "iommi/query/form_toggle_script.html",
+        query__form__attrs = {
+            'data-iommi-id-of-table': lambda table, **_: table.iommi_path,
+        },
+        assets__table_js_select_all__template = "iommi/table/js_select_all.html",
+        container__attrs ={
+            'data-endpoint': lambda table, **_: DISPATCH_PREFIX + table.endpoints.tbody.iommi_path,
+            'data-iommi-id': lambda table, **_: table.iommi_path,
+        },
     )
     def __init__(
         self,

iommi/table__tests.py

@@ -4561,3 +4561,20 @@ class TestTable(NoSortTable):
     t = t.bind(request=req('get'))
     verify_table_html(table=t, expected_html=expected_html)
     verify_table_html(table=t, expected_html=expected_html)
+
+
+def test_table_explicit_query():
+    Table(
+        rows=[],
+        query=Query(
+            filter=lambda rows, **_: rows,
+        ),
+    ).bind(request=req('get'))
+
+
+def test_table_list_custom_query():
+    Table(
+        rows=[],
+        query__filter=lambda rows, **_: rows,
+        query__filters__foo=Filter(),
+    ).bind(request=req('get'))