From 4c8816117bc2174f7c4e6ebd7612e01a0b557062 Mon Sep 17 00:00:00 2001 From: sepro Date: Thu, 30 Nov 2017 10:44:18 +0100 Subject: [PATCH] fixed bug when attempting to create an empty heatmap --- conekt/controllers/heatmap.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/conekt/controllers/heatmap.py b/conekt/controllers/heatmap.py index 19c7211..2e073c6 100644 --- a/conekt/controllers/heatmap.py +++ b/conekt/controllers/heatmap.py @@ -1,4 +1,4 @@ -from flask import Blueprint, request, render_template, Response +from flask import Blueprint, request, render_template, Response, redirect, flash, url_for import json @@ -77,15 +77,17 @@ def heatmap_custom_default(): form.populate_species() form.populate_options() - terms = request.form.get('probes').split() + probes = request.form.get('probes').split() species_id = request.form.get('species_id') option = request.form.get('options') - probes = terms + if len(probes) == 0: + flash("No genes selected!", "warning") + return redirect(url_for('heatmap.heatmap_main')) # also do search by gene ID - sequences = Sequence.query.filter(Sequence.name.in_(terms)).all() + sequences = Sequence.query.filter(Sequence.name.in_(probes)).all() for s in sequences: for ep in s.expression_profiles: @@ -113,6 +115,10 @@ def heatmap_custom_comparable(): option = request.form.get('options') + if len(terms) == 0: + flash("No genes selected!", "warning") + return redirect(url_for('heatmap.heatmap_main')) + sequences = Sequence.query.filter(Sequence.name.in_(terms)).all() sequence_ids = [s.id for s in sequences]