35 lines
857 B
Python
35 lines
857 B
Python
import glob
|
|
from io import BytesIO
|
|
from flask import Blueprint, request, send_file
|
|
from matplotlib import pyplot as plt
|
|
|
|
from tidi.staged.plot import tidi_render
|
|
|
|
|
|
tidi_module = Blueprint("Tidi", __name__)
|
|
|
|
@tidi_module.route('/metadata')
|
|
def get_all_years():
|
|
res = glob.glob("./tidi/data/**/**.txt", recursive=True)
|
|
# search for the folder name that is year
|
|
|
|
return {
|
|
"path": res
|
|
}
|
|
|
|
@tidi_module.route('/render/wave')
|
|
def render_wave():
|
|
mode = request.args.get('mode')
|
|
year = request.args.get('year')
|
|
k = request.args.get('k')
|
|
T = request.args.get('T')
|
|
year = int(year)
|
|
k = int(k)
|
|
T = int(T)
|
|
|
|
tidi_render(mode, year, k, T)
|
|
buffer = BytesIO()
|
|
plt.savefig(buffer, format="png")
|
|
buffer.seek(0)
|
|
|
|
return send_file(buffer, mimetype="image/png") |