HiLT/basic_backend.py at main · skberkeley/HiLT · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
import pandas as pd

# import coolNewLanguage.src as hilt
from coolNewLanguage import src as hilt

DESCRIPTION = "This is a Tool for testing basic backend functionality. Users can upload and name tables as CSV files, and pick and view the tables and columns they've uploaded."


tool = hilt.Tool('basic_backend', description=DESCRIPTION)

def csv_upload_and_name():
    """
    Stage to upload a csv file and give the resulting table a name. To test, upload a csv file, giving it a name,
    and then upload a non-csv file and make sure an error is shown. Finally, upload the original file and test that
    an error is shown regarding uploading a file with the same name.
    """
    table_name = hilt.UserInputComponent(str, label="Enter table name:")
    csv_file = hilt.FileUploadComponent('.csv', label="Upload a csv file:")

    if tool.user_input_received():
        df = pd.read_csv(csv_file.value)
        tool.tables[table_name.value] = df

        hilt.results.show_results((df, "Created table: "))

tool.add_stage('csv_upload_and_name', csv_upload_and_name)

def table_viewer():
    """
    Stage to select and view a table. To test, access this stage after uploading a csv, and view a table. Verify
    all uploaded tables are viewable, and no hidden ones are.
    """
    table = hilt.TableSelectorComponent()
    if tool.user_input_received():
        hilt.results.show_results((table, "Selected table: "))

tool.add_stage('table_viewer', table_viewer)

def column_viewer():
    """
    Stage to select and view a column of a table. To test, access this stage after uploading a csv, select a table,
    and then select a column. Verify that metadata columns (those beginning with '__') are not selectable.
    :return:
    """
    hilt.TextComponent("Select a table and then select a column:")
    column_selector1 = hilt.ColumnSelectorComponent("Select a column:")
    column_selector2 = hilt.ColumnSelectorComponent("Select another column:")
    if tool.user_input_received():
        hilt.results.show_results((column_selector1, "Selected column: "), (column_selector2, "Selected column: "))

tool.add_stage('column_viewer', column_viewer)

def column_viewer_with_table():
    """
    Stage to select and view a column of a table. To test, access this stage after uploading a csv, select a table,
    and then select a column. Verify that metadata columns (those beginning with '__') are not selectable.
    :return:
    """
    hilt.TextComponent("Select a table and then select a column:")
    table = hilt.TableSelectorComponent()
    column_selector = hilt.ColumnSelectorComponent("Select a column:")
    if tool.user_input_received():
        hilt.results.show_results((table, "Selected table: "), (column_selector, "Selected column: "))
tool.add_stage('column_viewer_with_table', column_viewer_with_table)


def copy_bug():
    table_name = hilt.UserInputComponent(str, label="Enter new table name: ")
    table = hilt.TableSelectorComponent()
    col = hilt.ColumnSelectorComponent()
    file = hilt.FileUploadComponent('csv')
    selector = hilt.SelectorComponent(['1', '2', '3'])
    if tool.user_input_received():
        print(table_name.value)
        print(table.table_name)
        print(col.table_name)
        print(file.value)
        print(selector.value)

tool.add_stage('copy_bug', copy_bug)

tool.run()