def create_dataset_from_csv(csv_file):
"""
Create a dataset from the uploaded .csv file.
"""
dataset_id = uuid.uuid4().hex
dataset = Dataset.create(dataset_id)
# need to write out to a named tempfile in order
# to get a handle in order for pandas read_csv
with tempfile.NamedTemporaryFile() as tmpfile:
tmpfile.write(read_uploaded_file(csv_file))
import_dataset(tmpfile.name, dataset)
return {ID: dataset_id}
def create_dataset_from_url(url, allow_local_file=False):
"""
Load a URL, read from a CSV, create a dataset and return the unique ID.
"""
_file = None
try:
_file = open_data_file(url, allow_local_file)
except (IOError, urllib2.HTTPError):
# error reading file/url, return
pass
if not _file:
# could not get a file handle
return {ERROR: 'could not get a filehandle for: %s' % url}
dataset_id = uuid.uuid4().hex
dataset = Dataset.create(dataset_id)
import_dataset(_file, dataset)
return {ID: dataset_id}
