summaryrefslogtreecommitdiff
path: root/tablib
diff options
context:
space:
mode:
Diffstat (limited to 'tablib')
-rw-r--r--tablib/__init__.py2
-rw-r--r--tablib/core.py13
2 files changed, 14 insertions, 1 deletions
diff --git a/tablib/__init__.py b/tablib/__init__.py
index ec8467e..65d1e13 100644
--- a/tablib/__init__.py
+++ b/tablib/__init__.py
@@ -1,7 +1,7 @@
""" Tablib. """
from tablib.core import (
- Databook, Dataset, detect, import_set,
+ Databook, Dataset, detect, import_set, import_book,
InvalidDatasetType, InvalidDimensions, UnsupportedFormat,
__version__
)
diff --git a/tablib/core.py b/tablib/core.py
index c52671e..1fdc77c 100644
--- a/tablib/core.py
+++ b/tablib/core.py
@@ -1008,6 +1008,19 @@ def import_set(stream):
return None
+def import_book(stream):
+ """Return dataset of given stream."""
+ (format, stream) = detect(stream)
+
+ try:
+ databook = Databook()
+ format.import_book(databook, stream)
+ return databook
+
+ except AttributeError:
+ return None
+
+
class InvalidDatasetType(Exception):
"Only Datasets can be added to a DataBook"