Replace IndexStats with CamelBase class

Author: ellnix

meilisearch/index.py

@@ -269,7 +269,7 @@ def get_stats(self) -> IndexStats:
             An error containing details about why Meilisearch can't process your request. Meilisearch error codes are described here: https://www.meilisearch.com/docs/reference/errors/error_codes#meilisearch-errors
         """
         stats = self.http.get(f"{self.config.paths.index}/{self.uid}/{self.config.paths.stat}")
-        return IndexStats(stats)
+        return IndexStats(**stats)
 
     @version_error_hint_message
     def search(self, query: str, opt_params: Optional[Mapping[str, Any]] = None) -> Dict[str, Any]:

meilisearch/models/index.py

@@ -1,31 +1,15 @@
 from __future__ import annotations
 
 from enum import Enum
-from typing import Any, Dict, Iterator, List, Optional, Union
+from typing import Dict, List, Optional, Union
 
-from camel_converter import to_snake
 from camel_converter.pydantic_base import CamelBase
 
 
-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())
+class IndexStats(CamelBase):
+    number_of_documents: int
+    is_indexing: bool
+    field_distribution: Dict[str, int]
 
 
 class Faceting(CamelBase):

tests/index/test_index_stats_meilisearch.py

@@ -13,5 +13,5 @@ def test_get_stats_default(index_with_documents):
     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
+    assert "genre" in response.field_distribution
+    assert response.field_distribution["genre"] == 11