app/invidious/__init__.py

   1 import requests
   2 from flask import Blueprint, render_template, request
   3
   4 #from ..common.common import *
   5
   6 frontend = Blueprint('invidious', __name__,
   7     template_folder='templates',
   8     static_folder='static',
   9     static_url_path='/static/iv')
  10
  11 # TODO: minimal implementation (invidio.us hardcoded, videos only, no optional_params)
  12 @frontend.route('/search')
  13 def feed():
  14     #token = getattr(current_user, 'token', 'guest')
  15     q = request.args.get('q')
  16     page = int(request.args.get('page', 1))
  17     provider = request.args.get('provider', 'invidio.us:443')
  18
  19     """
  20         sort_by: "relevance", "rating", "upload_date", "view_count"
  21         date: "hour", "today", "week", "month", "year"
  22         duration: "short", "long"
  23         type: "video", "playlist", "channel", "all", (default: video)
  24         features: "hd", "subtitles", "creative_commons", "3d", "live",
  25                   "purchased", "4k", "360", "location", "hdr" (comma separated)
  26         region: ISO 3166 country code (default: "US")
  27     """
  28     optional_params = {
  29         p: request.args.get(p)
  30         for p in ['sort_by', 'date', 'duration', 'type', 'features', 'region']
  31     }
  32
  33     if not q:
  34         return "no search query", 400
  35
  36     r = requests.get(f"https://invidio.us/api/v1/search", {
  37         'q': q,
  38         'page': page,
  39         # **optional_params,
  40     })
  41     if not r.ok:
  42         return "error fetching search results", 502 # TODO: better
  43
  44     # XXX: should check r.json if it really are search results (if other provider)
  45     # XXX: should transform invidious-json to our naming scheme
  46     return render_template('search.html.j2', rows=r.json(), query=q, page=page,
  47             optional_params=optional_params)