Added todict method to Entities.

ejhumphrey · Mar 12, 2016 · 164216f · 164216f
1 parent 56f56f0
commit 164216f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/biggie/core.py b/biggie/core.py
@@ -184,6 +184,9 @@ def items(self):
         """Return the (key, value) items of the entity."""
         return [(k, getattr(self, k)) for k in self.keys()]
 
+    def todict(self):
+        return {k: v for k, v in self.items()}
+
     @classmethod
     def from_hdf5_group(cls, group):
         """writeme."""

diff --git a/biggie/sources.py b/biggie/sources.py
@@ -143,7 +143,7 @@ def add(self, key, entity, overwrite=False):
         key : str
             Key to write the entity under.
 
-        entity : Entity
+        entity : Entity or dict
             Data to write to file.
 
         overwrite : bool, default=False