-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
117 lines (81 loc) · 3.1 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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
import os
import pandas as pd
from dotenv import load_dotenv
from flask import after_this_request, render_template, request, send_file, send_from_directory
from flask import Flask
from flask_session import Session
import tempfile
import traceback
import exceptions
import logging
import actions
from mapping import google, maps
load_dotenv()
app = Flask(__name__)
app.secret_key = os.urandom(28)
app.config["SESSION_TYPE"] = "filesystem"
sess = Session()
sess.init_app(app)
GA_TRACKING_ID = os.environ.get("GA_TRACKING_ID")
logging.basicConfig(level=logging.INFO)
@app.route("/")
def upload_page():
logging.info("Visitor to homepage")
return render_template("file_upload.html", GA_TRACKING_ID=GA_TRACKING_ID)
@app.errorhandler(exceptions.InvalidFile)
def handle_invalid_usage(e):
return render_template("file_upload.html", error=e.to_dict(), GA_TRACKING_ID=GA_TRACKING_ID)
@app.errorhandler(Exception)
def handle_unknown_error(e):
app.logger.warning(traceback.format_exc())
error = exceptions.UnknownError(str(e), e.args)
return render_template("file_upload.html", error=error.to_dict(), GA_TRACKING_ID=GA_TRACKING_ID)
@app.route("/base")
def base():
return render_template("base.html", GA_TRACKING_ID=GA_TRACKING_ID)
@app.route("/api/handle-upload", methods=["GET", "POST"])
def handle_upload():
try:
data = request.files["data"]
original_df = actions.parse_uploaded_file(data)
assert "count" in original_df
df = google.add_trip_data_to_dataframe(original_df.copy())
df = google.combine_with_original_dataframe(original_df, df)
df = google.multiply_measures_by_count(df)
assert len(df) == len(original_df)
except Exception as e:
raise e
logging.info("Adding coords from Google")
df = maps.add_coords_to_df(df)
logging.info("Coords finished")
actions.store_output_in_session(df.to_json(orient="records"))
total_co2 = df["total emissions (kg CO2)"].sum()
logging.info(f"Uploaded file successfully handled, total CO2 {total_co2:.2f}")
return render_template(
"results.html",
total_co2=f"{total_co2:.2f}",
map=maps.clean_html(
maps.plot_3d_map(df).to_html(as_string=True, iframe_width=800, iframe_height=800, notebook_display=False)
),
GA_TRACKING_ID=GA_TRACKING_ID,
)
@app.route("/api/download-results", methods=["GET"])
def download_results():
data = actions.retrieve_output_from_session()
df = pd.read_json(data, dtype=False)
df = actions.format_data_for_download(df)
temp = tempfile.NamedTemporaryFile(suffix=".xls")
df.to_excel(temp.name, index=False)
logging.info("File download successful")
@after_this_request
def teardown(response):
temp.close()
return response
return send_file(temp.name, as_attachment=True, attachment_filename="processed.xls")
@app.route("/favicon.ico")
def favicon():
return send_from_directory(
os.path.join(app.root_path, "static"), "favicon.ico", mimetype="image/vnd.microsoft.icon"
)
if __name__ == "__main__":
app.run(debug=True, use_reloader=False)