-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
98 lines (69 loc) · 2.41 KB
/
app.py
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
import json
import os
from flask import Flask, flash, redirect, render_template, request, url_for
from werkzeug.utils import secure_filename
from src.handlers import handle_analyze, handle_upload, ANALYSES_FOLDER, UPLOAD_FOLDER
app = Flask(__name__)
app.secret_key = b"6b3c3553ae7c43e5a6d4bf069b496ed6e29db4b94044be78b3bdd95f2dd75a5d"
@app.route("/")
def index():
existing_files = get_files()
return render_template("index.html", existing_files=existing_files)
def get_files():
return [
f
for f in os.listdir(UPLOAD_FOLDER)
if os.path.isfile(os.path.join(UPLOAD_FOLDER, f))
]
@app.post("/upload")
def upload_file():
if "upload_file" not in request.files:
flash("File is required.")
return redirect(url_for("index"))
delimiter_str = request.form["delimiter"]
if not delimiter_str:
flash("Please select delimiter")
return redirect(url_for("index"))
delimiter = get_delimiter(delimiter_str)
f = request.files["upload_file"]
filename = secure_filename(f.filename)
save_path = os.path.join(UPLOAD_FOLDER, filename)
f.save(save_path)
csv_cols = handle_upload(save_path, delimiter)
return render_template(
"column_selector.html", cols=csv_cols, filename=filename, delimiter=delimiter
)
@app.post("/analyze")
def analyze():
column = request.form.get("col")
filename = request.form.get("filename")
delimiter = request.form.get("delimiter")
if not column or not filename or not delimiter:
flash("An unexpected error occurred.")
return redirect(url_for("index"))
count_tuples, errors = handle_analyze(filename, column, delimiter)
return render_template(
"analysis.html",
filename=filename,
count_tuples=count_tuples,
errors=errors,
)
@app.get("/view/<filename>")
def view(filename):
analysis_path = os.path.join(ANALYSES_FOLDER, filename + ".json")
with open(analysis_path, "rt") as fin:
analysis = json.load(fin)
count_tuples = analysis["count_tuples"]
errors = analysis["errors"]
return render_template(
"analysis.html",
filename=filename,
count_tuples=count_tuples,
errors=errors,
)
def get_delimiter(delimiter_str: str) -> str:
if delimiter_str == "comma":
return ","
if delimiter_str == "tab":
return "\t"
raise ValueError("Unsupported delimiter.")