Making search document subscriptable

redis/commands/search/document.py

@@ -11,3 +11,7 @@ def __init__(self, id, payload=None, **fields):
 
     def __repr__(self):
         return f"Document {self.__dict__}"
+
+    def __getitem__(self, item):
+        value = getattr(self, item)
+        return value

tests/test_search.py

@@ -140,7 +140,9 @@ def test_client(client):
 
     for doc in res.docs:
         assert doc.id
+        assert doc["id"]
         assert doc.play == "Henry IV"
+        assert doc["play"] == "Henry IV"
         assert len(doc.txt) > 0
 
     # test no content