aboutsummaryrefslogtreecommitdiff
path: root/qc_app/upload
diff options
context:
space:
mode:
Diffstat (limited to 'qc_app/upload')
-rw-r--r--qc_app/upload/rqtl2.py56
1 files changed, 56 insertions, 0 deletions
diff --git a/qc_app/upload/rqtl2.py b/qc_app/upload/rqtl2.py
index 681e54c..53e7f3f 100644
--- a/qc_app/upload/rqtl2.py
+++ b/qc_app/upload/rqtl2.py
@@ -1,4 +1,6 @@
"""Module to handle uploading of R/qtl2 bundles."""
+from pathlib import Path
+from zipfile import ZipFile, is_zipfile
from flask import (
flash,
@@ -9,6 +11,10 @@ from flask import (
render_template,
current_app as app)
+from r_qtl import r_qtl2
+from r_qtl.errors import InvalidFormat
+
+from qc_app.files import save_file
from qc_app.dbinsert import species as all_species
from qc_app.db_utils import with_db_connection, database_connection
from qc_app.db import (
@@ -96,3 +102,53 @@ def create_population(species_id: int):
population_id=new_population["population_id"],
pgsrc="create-population"),
code=307)
+
+@rqtl2.route(("/upload/species/<int:species_id>/population/<int:population_id>"
+ "/rqtl2-bundle"),
+ methods=["GET", "POST"])
+def upload_rqtl2_bundle(species_id: int, population_id: int):
+ """Allow upload of R/qtl2 bundle."""
+ this_page_with_errors = redirect(url_for("upload.rqtl2.upload_rqtl2_bundle",
+ species_id=species_id,
+ population_id=population_id,
+ pgsrc="error"),
+ code=307)
+
+ with database_connection(app.config["SQL_URI"]) as conn:
+ species = species_by_id(conn, species_id)
+ population = population_by_species_and_id(
+ conn, species["SpeciesId"], population_id)
+ if not bool(species):
+ flash("Invalid species!", "alert-error error-rqtl2")
+ return redirect(url_for("upload.rqtl2.select_species"))
+ if not bool(population):
+ flash("Invalid Population!", "alert-error error-rqtl2")
+ return redirect(
+ url_for("upload.rqtl2.select_population", pgsrc="error"),
+ code=307)
+ if request.method == "GET" or (
+ request.method == "POST"
+ and bool(request.args.get("pgsrc"))):
+ return render_template("rqtl2/upload-rqtl2-bundle.html",
+ species=species,
+ population=population)
+
+ the_file = save_file(
+ request.files.get("rqtl2_bundle"), Path(app.config["UPLOAD_FOLDER"]))
+ if not bool(the_file):
+ flash("Please provide a valid R/qtl2 zip bundle.",
+ "alert-error alert-danger error-rqtl2")
+ return this_page_with_errors
+
+ if not is_zipfile(the_file):
+ flash("Invalid file! Expected a zip file.",
+ "alert-error alert-danger error-rqtl2")
+ return this_page_with_errors
+
+ try:
+ with ZipFile(the_file, "r") as zfile:
+ r_qtl2.validate_bundle(zfile)
+ return "WOULD PROCESS THE BUNDLE..."
+ except InvalidFormat as invf:
+ flash("".join(invf.args), "alert-error alert-danger error-rqtl2")
+ return this_page_with_errors