1
0
mirror of https://git.sr.ht/~cadence/NewLeaf synced 2024-12-22 21:16:58 +00:00

Implement search suggestions

This commit is contained in:
Cadence Ember 2020-08-10 19:25:15 +12:00
parent c9e995473f
commit 080b00bc0c
No known key found for this signature in database
GPG Key ID: 128B99B1B74A6412

View File

@ -39,7 +39,7 @@ class Second(object):
endpoints = [
["channels", 1, 2],
["videos", 1, 1],
["search", 0, 0]
["search", 0, 1]
]
for e in endpoints:
if vpath[2] == e[0] and len(vpath) >= e[1]+3 and len(vpath) <= e[2]+3:
@ -389,7 +389,10 @@ class Second(object):
@cherrypy.expose
@cherrypy.tools.json_out()
def search(self, *, q, **kwargs):
def search(self, *suffix, q, **kwargs):
if suffix == ("suggestions",):
return self.suggestions(q=q)
info = ytdl.extract_info("ytsearchall:{}".format(q), download=False)
return list({
"type": "video",
@ -411,6 +414,30 @@ class Second(object):
"isUpcoming": None
} for video in info["entries"] if "title" in video)
@cherrypy.expose
@cherrypy.tools.json_out()
def suggestions(self, *, q, **kwargs):
params = {
"client": "youtube",
"hl": "en",
"gl": "us",
"gs_rn": "64",
"gs_ri": "youtube",
"ds": "yt",
"cp": "3",
"gs_id": "k",
"q": q,
"xhr": "t",
# "xssi": "t"
}
with requests.get("https://clients1.google.com/complete/search", params=params) as r:
r.raise_for_status()
response = r.json()
return {
"query": q,
"suggestions": [s[0] for s in response[1]]
}
@cherrypy.expose
def vi(self, id, file):
with requests.get("https://i.ytimg.com/vi/{}/{}".format(id, file)) as r: