Example #1
0
def generateTestFile(id, option='csv', fname=None):
    """
    Generates a test file & uploads it to Document
    OPTIONS: (csv|not-csv|bad_data)
    :return file object
    """
    file_name = None
    output = io.BytesIO()
    wr = csv.writer(output, delimiter=",")
    if option == 'csv':
        file_name = "test1.csv"
        wr.writerow(['a', 'b', 'c', 'd', 'e'])
        for _ in range(1, 5):
            wr.writerow([1, 2, 3, 4, 5])
    elif option == 'not-csv':
        file_name = 'not-csv'
        wr.writerow('hello world\n1\n2.5\nlol\n');
    elif option == 'bad_data':
        file_name = 'bad_data.csv'
        for _ in range(1, 4):
            wr.writerow([1, 2, 3, 4, 5])
            wr.writerow([6.0, 6.0, 6.0, 6.0, 6.0])

    if fname: # overrides any default file name set
        file_name = fname

    user_file = SimpleUploadedFile(file_name, output.getvalue())
    Document.upload_file(id, file_name, user_file, FilesTest.username)
Example #2
0
def generateBadDoc(id):
    """
    Generates a 'bad' document (causes data-processing to crash)
    """
    new_files = []
    file_name = "test1"
    new_files.append(file_name)
    with open(os.path.join(settings.MEDIA_ROOT, file_name), 'wb') as csvfile:
            wr = csv.writer(csvfile, delimiter=",")
            wr.writerow(['c{}'.format(k) for k in range(4)])
            # uneven row/column lengths!
            wr.writerow([k for k in range(4)])
            wr.writerow([k for k in range(3)])
            wr.writerow([k for k in range(2)])
    Document.upload_file(id, file_name, file_name, PlotTest.username) # not sure about path correctness
    return new_files
Example #3
0
def list_panes(request):
    """
    Entry handler for user's home page!
    """
    # redirect to login page if not signed in
    if not request.user.is_authenticated():
        return redirect("top")

    if request.method == "POST":
        form = DocumentForm(request.POST, request.FILES)
        if form.is_valid():
            owner = Account.objects.get(id=request.user.id).username
            Document.upload_file(request.user.id, request.FILES["csv_file"].name, request.FILES["csv_file"], owner)
        # Redirect to the document list after POST
        return HttpResponseRedirect(reverse("app.views.list_panes"))

    elif request.method == "GET":
        return render_to_response(
            "home.html", get_panes_data(request.user.id), context_instance=RequestContext(request)
        )
Example #4
0
def generateDocs(id, n, r, c, name=None):
    """
    Helper to create test docs according to the parameters below

    :param id: account id
    :param n: number of files to generate
    :param r: number of rows per file
    :param c: number of cols per file
    :return: [files names of new files generated]
    """
    new_files = []
    for i in range(1, n+1):
        if not name:
            file_name = "test{}".format(i)
        else:
            file_name = name + str(i)
        new_files.append(file_name)
        with open(os.path.join(settings.MEDIA_ROOT, file_name), 'wb') as csvfile:
            wr = csv.writer(csvfile, delimiter=",")
            wr.writerow(['c{}'.format(k) for k in range(c)])
            for j in range(r):
                wr.writerow([i*k for k in range(1, c+1)])
        Document.upload_file(id, file_name, file_name, PlotTest.username) # not sure about path correctness
    return new_files