# -*- 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