updated data store util class

bdshare/util/store.py

@@ -1,10 +1,9 @@
 # -*- coding:utf-8 -*-
 
 import pandas as pd
-from pandas import compat
+from datetime import datetime
 import os
 
-
 class Store(object):
 
     def __init__(self, data=None, name=None, path=None):
@@ -15,20 +14,21 @@ def __init__(self, data=None, name=None, path=None):
         self.name = name
         self.path = path
 
-    def save_as(self, name, path, to='csv'):
-        if name is None:
-            name = self.name
-        if path is None:
-            path = self.path
+    def save(self, to='csv'):
+        if self.name is None:
+            self.name = f'{datetime.now().strftime("%Y%m%d-%H%M%S")}'
+
         file_path = '%s%s%s.%s'
-        if isinstance(name, compat.string_types) and name is not '':
-            if (path is None) or (path == ''):
-                file_path = '.'.join([name, to])
+        if isinstance(self.name, str) and self.name is not '':
+            if (self.path is None) or (self.path == ''):
+                file_path = '.'.join([self.name, to])
+                self.data.to_csv(file_path, index=False)
             else:
                 try:
-                    if os.path.exists(path) is False:
-                        os.mkdir(path) 
-                    file_path = file_path%(path, '/', name, to)
+                    if os.path.exists(self.path) is False:
+                        os.mkdir(self.path) 
+                    file_path = file_path%(self.path, '', self.name, to)
+                    self.data.to_csv(file_path, index=False)
                 except:
                     pass
 

tests/test_dsex.py

@@ -6,5 +6,6 @@
 from bdshare import get_dsex_data, Store
 
 df = get_dsex_data()
-Store(data=df)
-print(df.to_string())
+
+print(df.to_string())
+Store(data=df).save()