index stats dot attributes + tests

Author: elamc-2

meilisearch/index.py

@@ -227,7 +227,7 @@ def get_stats(self) -> IndexStats:
         stats = self.http.get(
             f'{self.config.paths.index}/{self.uid}/{self.config.paths.stat}'
         )
-        return IndexStats(**stats)
+        return IndexStats(stats)
 
     def search(self, query: str, opt_params: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
         """Search in the index.

meilisearch/models/index.py

@@ -1,7 +1,22 @@
-from typing import Any, Dict
-from camel_converter.pydantic_base import CamelBase
+from typing import Any, Dict, Iterator
+from camel_converter import to_snake
 
-class IndexStats(CamelBase):
-    number_of_documents: int
-    is_indexing: bool
-    field_distribution: Dict[str, Any]
+class IndexStats:
+    __dict: dict
+
+    def __init__(self, doc: Dict[str, Any]) -> None:
+        self.__dict = doc
+        for key, val in doc.items():
+            key = to_snake(key)
+            if isinstance(val, dict):
+                setattr(self, key, IndexStats(val))
+            else:
+                setattr(self, key, val)
+
+    def __getattr__(self, attr: str) -> Any:
+        if attr in self.__dict.keys():
+            return attr
+        raise AttributeError(f"{self.__class__.__name__} object has no attribute {attr}")
+
+    def __iter__(self) -> Iterator:
+        return iter(self.__dict__.items())  # type: ignore

tests/index/test_index_stats_meilisearch.py

@@ -5,3 +5,11 @@ def test_get_stats(empty_index):
     response = empty_index().get_stats()
     assert isinstance(response, IndexStats)
     assert response.number_of_documents == 0
+
+def test_get_stats_default(index_with_documents):
+    """Tests getting stats of a non-empty index."""
+    response = index_with_documents().get_stats()
+    assert isinstance(response, IndexStats)
+    assert response.number_of_documents == 31
+    assert hasattr(response.field_distribution, 'genre')
+    assert response.field_distribution.genre == 11