PyNyaaTa/pynyaata2/__init__.py

import asyncio
from datetime import datetime

from flask import Flask, redirect, render_template, request, url_for

from pynyaata2.bridge import search_all
from pynyaata2.translations import current_lang, i18n


app = Flask(__name__)


@app.context_processor
def utility_processor():
    current_date = datetime.now()
    apocalypse_left = (
        datetime(current_date.year + 5 - (current_date.year % 5), 6, 4, 0, 1)
        - current_date
    )

    return dict(
        _=i18n,
        current_lang=current_lang(),
        apocalypse_left=apocalypse_left,
    )


@app.route("/")
def index():
    return render_template("index.html.j2")


@app.route("/search")
def search():
    query = request.args.get("q")
    if not query:
        return redirect(url_for("index"))

    torrents = asyncio.run(search_all(query))

    return render_template("search.html.j2", torrents=torrents)


def run():
    app.run("0.0.0.0")


if __name__ == "__main__":
    run()