blob: 632c54aa2badfbaa272964fbc5a9c7d3029d8079 (
plain)
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
|
"""
Module to hold the entrypoint functions that generate heatmaps
"""
import io
from flask import jsonify
from flask import request
from flask import Blueprint, current_app
from gn3.heatmaps import build_heatmap
from gn3.db_utils import database_connection
heatmaps = Blueprint("heatmaps", __name__)
@heatmaps.route("/clustered", methods=("POST",))
def clustered_heatmaps():
"""
Parses the incoming data and responds with the JSON-serialized plotly figure
representing the clustered heatmap.
"""
heatmap_request = request.get_json()
traits_names = heatmap_request.get("traits_names", tuple())
vertical = heatmap_request.get("vertical", False)
if len(traits_names) < 2:
return jsonify({
"message": "You need to provide at least two trait names."
}), 400
with database_connection(current_app.config["SQL_URI"]) as conn:
def parse_trait_fullname(trait):
name_parts = trait.split(":")
return f"{name_parts[1]}::{name_parts[0]}"
traits_fullnames = [parse_trait_fullname(trait) for trait in traits_names]
with io.StringIO() as io_str:
figure = build_heatmap(conn,
traits_fullnames,
current_app.config["GENOTYPE_FILES"],
vertical=vertical,
tmpdir=current_app.config["TMPDIR"])
figure.write_json(io_str)
fig_json = io_str.getvalue()
return fig_json, 200
|
