push latest changes

views/search/__init__.py

@@ -0,0 +1,62 @@
+from urllib.parse import unquote_plus
+
+from flask import Blueprint, json, render_template, request
+
+from api import Client
+from forms import SearchForm
+from utils import admin_required
+
+search_page = Blueprint("search", __name__, url_prefix="/search")
+
+
+@search_page.route("/details", methods=["GET", "POST"])
+@admin_required
+def details():
+    data = {
+        "info": json.loads(unquote_plus(request.form["data"])),
+        "type": request.form["type"],
+    }
+    return render_template("search/details.html", **data)
+
+
+@search_page.route("/", methods=["GET", "POST"])
+@admin_required
+def index():
+    c = Client()
+    results = {}
+    params = request.args
+    form = SearchForm()
+    form.indexer.choices = c.jackett.indexers()
+    if form.validate_on_submit():
+        query = form.query.data
+        if not (form.torrents.data or form.movies.data or form.tv_shows.data):
+            form.torrents.data = True
+            form.movies.data = True
+            form.tv_shows.data = True
+
+        if form.torrents.data:
+            results["torrents"] = c.jackett.search(
+                query, form.indexer.data or form.indexer.choices
+            )
+        if form.movies.data:
+            results["movies"] = c.radarr.search(query)
+        if form.tv_shows.data:
+            results["tv_shows"] = c.sonarr.search(query)
+        return render_template(
+            "search/index.html",
+            # form=form,
+            search_term=query,
+            results=results,
+            client=c,
+            group_by_tracker=form.group_by_tracker.data,
+        )
+    for name, field in form._fields.items():
+        field.default = params.get(name)
+    form.process()
+    return render_template(
+        "search/index.html",
+        form=form,
+        results={},
+        group_by_tracker=False,
+        sort_by="Gain",
+    )