Make feth_info return an Index object instead of a dict

Author: curquiza

.code-samples.meilisearch.yaml

@@ -5,6 +5,9 @@
 ---
 get_one_index_1: |-
   index = client.get_index('movies')
+  # Or
+  index = client.index('movies').fetch_info()
+
   index.uid
   index.primary_key
 list_all_indexes_1: |-

meilisearch/client.py

@@ -80,8 +80,7 @@ def get_index(self, uid):
         index : Index
             An Index instance containing the information the fetched index.
         """
-        index_dict = Index(self.config, uid).fetch_info()
-        return Index(self.config, index_dict['uid'], index_dict['primaryKey'])
+        return Index(self.config, uid).fetch_info()
 
     def index(self, uid):
         """Create an Index instance.

meilisearch/index.py

@@ -71,12 +71,12 @@ def fetch_info(self):
 
         Returns
         -------
-        index: dict
-            Dictionary containing the index information.
+        index : Index
+            An instance of Index containing the information of the index.
         """
         index_dict = self.http.get('{}/{}'.format(self.config.paths.index, self.uid))
         self.primary_key = index_dict['primaryKey']
-        return index_dict
+        return self
 
     def get_primary_key(self):
         """Get the primary key.
@@ -86,8 +86,7 @@ def get_primary_key(self):
         primary_key: str
             String containing the primary key.
         """
-        self.primary_key = self.fetch_info()['primaryKey']
-        return self.primary_key
+        return self.fetch_info().primary_key
 
     @staticmethod
     def create(config, uid, options=None):

meilisearch/tests/index/test_index.py

@@ -111,20 +111,20 @@ def test_index_fetch_info(self):
         index = self.client.index(uid=self.index_uid)
         response = index.fetch_info()
         assert isinstance(response, object)
-        assert response['uid'] == self.index_uid
-        assert response['primaryKey'] is None
-        assert index.primary_key == response['primaryKey']
-        assert index.get_primary_key() == response['primaryKey']
+        assert response.uid == self.index_uid
+        assert response.primary_key is None
+        assert response.primary_key == index.primary_key
+        assert response.primary_key == index.get_primary_key()
 
     def test_index_fetch_info_containing_primary_key(self):
         """Tests getting the index info"""
         index = self.client.index(uid=self.index_uid3)
         response = index.fetch_info()
         assert isinstance(response, object)
-        assert response['uid'] == self.index_uid3
-        assert response['primaryKey'] == 'book_id'
-        assert index.primary_key == response['primaryKey']
-        assert index.get_primary_key() == response['primaryKey']
+        assert response.uid == self.index_uid3
+        assert response.primary_key == 'book_id'
+        assert response.primary_key == index.primary_key
+        assert response.primary_key == index.get_primary_key()
 
     def test_get_primary_key(self):
         """Tests getting the primary-key of an index"""