Implement keys() and items() for AttrDict

Author: miguelgrinberg

elasticsearch_dsl/utils.py

@@ -184,6 +184,12 @@ def __iter__(self):
     def to_dict(self):
         return self._d_
 
+    def keys(self):
+        return self._d_.keys()
+
+    def items(self):
+        return self._d_.items()
+
 
 class DslMeta(type):
     """

tests/test_integration/_async/test_document.py

@@ -219,7 +219,7 @@ async def test_update_object_field(async_write_client):
     )
     await w.save()
 
-    assert "updated" == await w.update(owner=[{"name": "Honza"}, {"name": "Nick"}])
+    assert "updated" == await w.update(owner=[{"name": "Honza"}, User(name="Nick")])
     assert w.owner[0].name == "Honza"
     assert w.owner[1].name == "Nick"
 

tests/test_integration/_sync/test_document.py

@@ -219,7 +219,7 @@ def test_update_object_field(write_client):
     )
     w.save()
 
-    assert "updated" == w.update(owner=[{"name": "Honza"}, {"name": "Nick"}])
+    assert "updated" == w.update(owner=[{"name": "Honza"}, User(name="Nick")])
     assert w.owner[0].name == "Honza"
     assert w.owner[1].name == "Nick"
 

tests/test_utils.py

@@ -44,6 +44,12 @@ class MyAttrDict(utils.AttrDict):
     assert isinstance(l[:][0], MyAttrDict)
 
 
+def test_attrdict_keys_items():
+    a = utils.AttrDict({"a": {"b": 42, "c": 47}, "d": "e"})
+    assert list(a.keys()) == ["a", "d"]
+    assert list(a.items()) == [("a", {"b": 42, "c": 47}), ("d", "e")]
+
+
 def test_merge():
     a = utils.AttrDict({"a": {"b": 42, "c": 47}})
     b = {"a": {"b": 123, "d": -12}, "e": [1, 2, 3]}