diff options
Diffstat (limited to 'src/tablib/formats/_dbf.py')
| -rw-r--r-- | src/tablib/formats/_dbf.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/src/tablib/formats/_dbf.py b/src/tablib/formats/_dbf.py index fecb214..2a6df32 100644 --- a/src/tablib/formats/_dbf.py +++ b/src/tablib/formats/_dbf.py @@ -2,7 +2,6 @@ """ import io import os -import struct import tempfile from tablib.packages.dbfpy import dbf @@ -15,6 +14,7 @@ extensions = ('csv',) DEFAULT_ENCODING = 'utf-8' + def export_set(dataset): """Returns DBF representation of a Dataset""" new_dbf = dbfnew.dbf_new() @@ -45,6 +45,7 @@ def export_set(dataset): os.remove(temp_uri) return stream.getvalue() + def import_set(dset, in_stream, headers=True): """Returns a dataset from a DBF stream.""" @@ -55,13 +56,14 @@ def import_set(dset, in_stream, headers=True): row = [_dbf[record][f] for f in _dbf.fieldNames] dset.append(row) + def detect(stream): """Returns True if the given stream is valid DBF""" - #_dbf = dbf.Table(StringIO(stream)) + # _dbf = dbf.Table(StringIO(stream)) try: if type(stream) is not bytes: stream = bytes(stream, 'utf-8') - _dbf = dbf.Dbf(io.BytesIO(stream), readOnly=True) + dbf.Dbf(io.BytesIO(stream), readOnly=True) return True except Exception: return False |