Removing DictObject class.

1 files changed, 6 insertions, 26 deletions

diff --git a/src/argaze/DataStructures.py b/src/argaze/DataStructures.py
index 8517378..ead5c48 100644
--- a/src/argaze/DataStructures.py
+++ b/src/argaze/DataStructures.py
@@ -5,31 +5,6 @@ import json
 import bisect
 import pandas
 
-class DictObject():
-    """Convert dictionnary into object"""
-
-    def __init__(self, object_type = str, **dictionnary):
-        self.__dict__.update(dictionnary)
-        self.__type = object_type
-
-    def __getitem__(self, key):
-        return self.__dict__[key]
-
-    def __str__(self):
-        return json.dumps({key: self.__dict__[key] for key in self.__dict__.keys()}, default = vars)
-
-    def get_type(self):
-        return self.__type
-
-    def keys(self):
-        return list(self.__dict__.keys())[:-1]
-
-    def append(self, key, value):
-        __type_key = list(self.__dict__.keys())[-1]
-        __type_value = self.__dict__.pop(__type_key)
-        self.__dict__.update({key:value})
-        self.__dict__[__type_key] = __type_value
-
 class TimeStampedBuffer(collections.OrderedDict):
     """Ordered dictionary to handle timestamped data.
     ```
@@ -107,11 +82,16 @@ class TimeStampedBuffer(collections.OrderedDict):
         except:
             raise RuntimeError(f'Can\' write {filepath}')
 
-    def as_dataframe(self, exclude=[]):
+    def as_dataframe(self, exclude=[], split={}):
         """Convert buffer as pandas dataframe."""
 
         df = pandas.DataFrame.from_dict(self.values())
         df.drop(exclude, inplace=True, axis=True)
+
+        for key, columns in split.items():
+            df[columns] = pandas.DataFrame(df[key].tolist(), index=df.index)
+            df.drop(key, inplace=True, axis=True)
+
         df['timestamp'] = self.keys()
         df.set_index('timestamp', inplace=True)