Fix print_schema() bug

pandasticsearch/dataframe.py

@@ -441,9 +441,16 @@ def print_schema(self):
             return
 
         sys.stdout.write('{0}\n'.format(self._index))
-        index = list(self._mapping.values())[0]  # {'index': {}}
-        schema = self.resolve_schema(index['mappings']['properties'])
-        sys.stdout.write(schema)
+        index_name = list(self._mapping.keys())[0]
+        if self._compat >= 7:
+            json_obj = self._mapping[index_name]["mappings"]["properties"]
+            sys.stdout.write(self.resolve_schema(json_obj))
+        else:
+            if self._doc_type is not None:
+                json_obj = self._mapping[index_name]["mappings"][self._doc_type]["properties"]
+                sys.stdout.write(self.resolve_schema(json_obj))
+            else:
+                raise DataFrameException('Please specify mapping for ES version under 7')
 
     def resolve_schema(self, json_prop, res_schema="", depth=1):
         for field in json_prop:

setup.py

@@ -9,7 +9,7 @@
 
 setup(
     name='pandasticsearch',
-    version='0.5.2',
+    version='0.5.3',
     author='onesuper',
     author_email='[email protected]',
     packages=['pandasticsearch', 'pandasticsearch.operators'],

tests/test_dataframe.py

@@ -93,10 +93,18 @@ def test_columns(self):
         df = create_df_from_es()
         self.assertEqual(df.columns, ['a', 'b', 'c.d', 'c.e'])
 
+    def test_print_schema(self):
+        df = create_df_from_es()
+        df.print_schema()
+
     def test_columns_after_removal_mapping(self):
         df = create_df_from_es_after_removal_mapping()
         self.assertEqual(df.columns, ['a', 'b', 'c.d', 'c.e'])
 
+    def test_print_schema_after_removal_mapping(self):
+        df = create_df_from_es_after_removal_mapping()
+        df.print_schema()
+
     def test_init(self):
         df = create_df_from_es()
         self.assertEqual(df.to_dict(), {'size': 20})