PYTHON-4834 Add __repr__ to IndexModel

- Special case for `key` kwarg, all other document keys used as-is

Author: aclark4life

pymongo/operations.py

@@ -774,11 +774,16 @@ def document(self) -> dict[str, Any]:
         return self.__document
 
     def __repr__(self) -> str:
-        return "%s(%s%s%s)" % (
-            self.__class__.__name__,
-            "" if not self.document.get("name") else "name='%s'" % self.document["name"],
-            "" if not self.document.get("key") else ", keys=%s" % self.document["key"],
-            "" if not self.document.get("unique") else ", unique='%s'" % self.document["unique"],
+        return "%s(%s%s)" % (
+            self.__class__.__name__,                                               
+            "" if not self.document.get("key") else "keys=%s" % self.document["key"],
+            "".join(
+                [
+                    ", %s=%r" % (key, value)
+                    for key, value in self.document.items()
+                    if not key == "key"
+                ]
+            ),
         )
 
 

test/test_collection.py

@@ -2259,10 +2259,23 @@ def assertRepr(self, obj):
         self.assertEqual(repr(new_obj), repr(obj))
 
     def test_index_model_repr(self):
-        self.assertRepr(IndexModel("hello", unique=False))
-        self.assertRepr(IndexModel("world", unique=True))
-        self.assertRepr(IndexModel([("hello", DESCENDING)], unique=True))
-        self.assertRepr(IndexModel([("world", DESCENDING)], name="world", unique=True))
+        self.assertRepr(IndexModel("hello"))
+        self.assertRepr(IndexModel([("hello", DESCENDING), ("world", ASCENDING)]))
+        self.assertRepr(IndexModel([("hello", DESCENDING), ("world", ASCENDING)], name="hello_world"))
+        self.assertRepr(IndexModel("a", unique=True))
+        # self.assertRepr(name
+        # self.assertRepr(unique
+        # self.assertRepr(background
+        # self.assertRepr(sparse
+        # self.assertRepr(bucketSize
+        # self.assertRepr(min
+        # self.assertRepr(max
+        # self.assertRepr(expireAfterSeconds
+        # self.assertRepr(partialFilterExpression
+        # self.assertRepr(collation
+        # self.assertRepr(wildcardProjection
+        # self.assertRepr(hidden
+
 
 if __name__ == "__main__":
     unittest.main()