Adapting method saving data to the file

SkillCorner · Mar 16, 2022 · d4ceded · d4ceded
1 parent 2776345
commit d4ceded
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/skillcorner/client.py b/skillcorner/client.py
@@ -552,6 +552,10 @@ def _get_and_write_data_with_id(self, id, filepath, *, url, paginated_request, p
         try:
             with open(filepath, 'w') as file:
                 json.dump(data, file, indent=4)
-        except TypeError:
-            with open(filepath, 'wb') as file:
-                file.write(data)
+        except TypeError as ex:
+            if 'DataFrame' in str(ex):
+                import pandas as pd
+                data.to_csv(filepath)
+            else:
+                with open(filepath, 'wb') as file:
+                    file.write(data)