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