about summary refs log tree commit diff
path: root/uploader/publications/views.py
diff options
context:
space:
mode:
Diffstat (limited to 'uploader/publications/views.py')
-rw-r--r--uploader/publications/views.py29
1 files changed, 20 insertions, 9 deletions
diff --git a/uploader/publications/views.py b/uploader/publications/views.py
index 4ec832f..f0ec95a 100644
--- a/uploader/publications/views.py
+++ b/uploader/publications/views.py
@@ -10,6 +10,8 @@ from flask import (
     render_template,
     current_app as app)
 
+from uploader.sui import sui_template
+
 from uploader.flask_extensions import url_for
 from uploader.authorisation import require_login
 from uploader.route_utils import redirect_to_next
@@ -30,7 +32,7 @@ pubbp = Blueprint("publications", __name__)
 @require_login
 def index():
     """Index page for publications."""
-    return render_template("publications/index.html")
+    return render_template(sui_template("publications/index.html"))
 
 
 @pubbp.route("/list", methods=["GET"])
@@ -72,7 +74,7 @@ def view_publication(publication_id: int):
             return redirect(url_for('publications.index'))
 
         return render_template(
-            "publications/view-publication.html",
+            sui_template("publications/view-publication.html"),
             publication=publication,
             linked_phenotypes=tuple(fetch_publication_phenotypes(
                 conn, publication_id)))
@@ -82,13 +84,21 @@ def view_publication(publication_id: int):
 @require_login
 def create_publication():
     """Create a new publication."""
+    _get_args = {
+        key: request.args[key]
+        for key in ("species_id", "population_id", "dataset_id", "return_to")
+        if bool(request.args.get(key))
+    }
+
     if request.method == "GET":
-        return render_template("publications/create-publication.html")
+        return render_template(
+            sui_template("publications/create-publication.html"),
+            get_args=_get_args)
     form = request.form
     authors = form.get("publication-authors").encode("utf8")
     if authors is None or authors == "":
         flash("The publication's author(s) MUST be provided!", "alert alert-danger")
-        return redirect(url_for("publications.create", **request.args))
+        return redirect(url_for("publications.create"))
 
     with database_connection(app.config["SQL_URI"]) as conn:
         publications = create_new_publications(conn, ({
@@ -106,7 +116,7 @@ def create_publication():
         return redirect(url_for(
             request.args.get("return_to") or "publications.view_publication",
             publication_id=publications[0]["publication_id"],
-            **request.args))
+            **_get_args))
 
     flash("Publication creation failed!", "alert alert-danger")
     app.logger.debug("Failed to create the new publication.", exc_info=True)
@@ -120,7 +130,7 @@ def edit_publication(publication_id: int):
     with database_connection(app.config["SQL_URI"]) as conn:
         if request.method == "GET":
             return render_template(
-                "publications/edit-publication.html",
+                sui_template("publications/edit-publication.html"),
                 publication=fetch_publication_by_id(conn, publication_id),
                 linked_phenotypes=tuple(fetch_publication_phenotypes(
                     conn, publication_id)),
@@ -171,15 +181,16 @@ def delete_publication(publication_id: int):
             flash("Cannot delete publication with linked phenotypes!",
                   "alert-warning")
             return redirect(url_for(
-                "publications.view_publication", publication_id=publication_id))
+                sui_template("publications.view_publication"),
+                publication_id=publication_id))
 
         if request.method == "GET":
             return render_template(
-                "publications/delete-publication.html",
+                sui_template("publications/delete-publication.html"),
                 publication=publication,
                 linked_phenotypes=linked_phenotypes,
                 publication_id=publication_id)
 
         delete_publications(conn, (publication,))
         flash("Deleted the publication successfully.", "alert-success")
-        return render_template("publications/delete-publication-success.html")
+        return redirect(url_for("publications.index"))