mirror of
https://github.com/kennethreitz/tablib.git
synced 2026-06-05 23:10:17 +00:00
Kenneth Reitz
52 lines
923 B
Python
52 lines
923 B
Python
# -*- coding: utf-8 -*-
|
|
|
|
""" Tablib - TSV (Tab Separated Values) Support.
|
|
"""
|
|
|
|
import cStringIO
|
|
import csv
|
|
import os
|
|
|
|
import tablib
|
|
|
|
|
|
title = 'tsv'
|
|
extentions = ('tsv',)
|
|
|
|
|
|
|
|
def export_set(dataset):
|
|
"""Returns a TSV representation of Dataset."""
|
|
stream = cStringIO.StringIO()
|
|
_tsv = csv.writer(stream, delimiter='\t')
|
|
|
|
for row in dataset._package(dicts=False):
|
|
_tsv.writerow(row)
|
|
|
|
return stream.getvalue()
|
|
|
|
|
|
def import_set(dset, in_stream, headers=True):
|
|
"""Returns dataset from TSV stream."""
|
|
dset.wipe()
|
|
|
|
rows = csv.reader(in_stream.split('\r\n'), delimiter='\t')
|
|
for i, row in enumerate(rows):
|
|
# Skip empty rows
|
|
if not row:
|
|
continue
|
|
|
|
if (i == 0) and (headers):
|
|
dset.headers = row
|
|
else:
|
|
dset.append(row)
|
|
|
|
|
|
def detect(stream):
|
|
"""Returns True if given stream is valid TSV."""
|
|
try:
|
|
rows = dialect = csv.Sniffer().sniff(stream, delimiters='\t')
|
|
return True
|
|
except csv.Error:
|
|
return False
|