def download(*args, **kwargs): result = download_to_file.AsyncResult(args[0]['uuid']) state = result.state if state == states.PENDING: raise QueryExpired() elif state in states.EXCEPTION_STATES: result.maybe_reraise() info = result.wait() # TODO: Start returning data even if we're not done return export_csvxls.file_reader(open(info['file_path'], 'rb'))
def download(*args, **kwargs): notebook = args[0] result = download_to_file.AsyncResult(args[0]['uuid']) state = result.state if state == states.PENDING: raise QueryExpired() elif state in states.EXCEPTION_STATES: result.maybe_reraise() info = result.wait() # TODO: Start returning data even if we're not done return export_csvxls.file_reader(storage.open(_result_key(notebook), 'rb')) # TODO: Convert csv to excel if needed