Use Python descriptors to access entity properties

CHANGELOG.md

@@ -1,5 +1,6 @@
 ## 0.2.0 ()
 
+- Use Python descriptors to access entity properties.
 - Add validation for `id` field of PathElement.
 - Add validation for `path` field of Key.
 

aiodatastore/entity.py

@@ -24,6 +24,15 @@ def __eq__(self, other: Any) -> bool:
             and self.properties == other.properties
         )
 
+    def __getitem__(self, key):
+        return self.properties[key]
+
+    def __setitem__(self, key, value):
+        self.properties[key] = value
+
+    def __delitem__(self, key):
+        del self.properties[key]
+
     @classmethod
     def from_ds(cls, data: Dict[str, Any]) -> "Entity":
         properties = {}

tests/unit/test_entity.py

@@ -19,6 +19,25 @@ def test__init(self):
         assert e.key is None
         assert e.properties == {}
 
+    def test__getitem__(self):
+        entity = Entity(None, {"field1": StringValue("str1")})
+        assert entity["field1"] == entity.properties["field1"]
+
+        with self.assertRaises(KeyError):
+            entity["some-key123"]
+
+    def test__setitem__(self):
+        entity = Entity(None, {})
+        entity["field2"] = StringValue("str2")
+        assert entity["field2"] == entity.properties["field2"]
+
+    def test__delitem__(self):
+        entity = Entity(None, {"field3": StringValue("str3")})
+        assert len(entity.properties) == 1
+        del entity["field3"]
+        assert len(entity.properties) == 0
+        assert "field3" not in entity.properties
+
     def test_eq(self):
         key1 = Key(PartitionId("project1"), [PathElement("kind1")])
         key2 = Key(PartitionId("project1"), [PathElement("kind1")])