63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
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",
|
|
)
|