From 80aa8933e6efd2b85d8ea077903f506f3797be0a Mon Sep 17 00:00:00 2001
From: Simon <fortier.simon@hotmail.com>
Date: Thu, 21 Jun 2018 13:02:16 -0400
Subject: [PATCH] Added rescan button

---
 app.py                 | 21 +++++++++++++++++++++
 templates/website.html |  1 +
 2 files changed, 22 insertions(+)

diff --git a/app.py b/app.py
index c19482d..97fa422 100644
--- a/app.py
+++ b/app.py
@@ -162,6 +162,27 @@ def admin_delete_website(website_id):
         abort(403)
 
 
+@app.route("/website/<int:website_id>/rescan")
+def admin_rescan_website(website_id):
+
+    if "username" in session:
+
+        website = db.get_website_by_id(website_id)
+
+        if website:
+            priority = request.args.get("priority") if "priority" in request.args else 1
+            task = Task(website_id, website.url, priority)
+            taskDispatcher.dispatch_task(task)
+
+            flash("Enqueued rescan task", "success")
+        else:
+            flash("Website does not exist", "danger")
+        return redirect("/website/" + str(website_id))
+
+    else:
+        abort(403)
+
+
 @app.route("/search")
 def search():
 
diff --git a/templates/website.html b/templates/website.html
index 89c7297..499e70b 100644
--- a/templates/website.html
+++ b/templates/website.html
@@ -43,6 +43,7 @@
                 {% if "username" in session %}
                     <a href="/website/{{ website.id }}/clear" class="btn btn-danger"><i class="fas fa-exclamation"></i> Clear</a>
                     <a href="/website/{{ website.id }}/delete" class="btn btn-danger"><i class="fas fa-trash"></i> Delete</a>
+                    <a href="/website/{{ website.id }}/rescan" class="btn btn-secondary"><i class="fas fa-redo"></i> rescan</a>
                 {% endif %}
             </div>
         </div>