mirror of
https://git.sr.ht/~cadence/NewLeaf
synced 2024-11-22 07:37:29 +00:00
Implement search suggestions
This commit is contained in:
parent
c9e995473f
commit
080b00bc0c
31
index.py
31
index.py
@ -39,7 +39,7 @@ class Second(object):
|
|||||||
endpoints = [
|
endpoints = [
|
||||||
["channels", 1, 2],
|
["channels", 1, 2],
|
||||||
["videos", 1, 1],
|
["videos", 1, 1],
|
||||||
["search", 0, 0]
|
["search", 0, 1]
|
||||||
]
|
]
|
||||||
for e in endpoints:
|
for e in endpoints:
|
||||||
if vpath[2] == e[0] and len(vpath) >= e[1]+3 and len(vpath) <= e[2]+3:
|
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.expose
|
||||||
@cherrypy.tools.json_out()
|
@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)
|
info = ytdl.extract_info("ytsearchall:{}".format(q), download=False)
|
||||||
return list({
|
return list({
|
||||||
"type": "video",
|
"type": "video",
|
||||||
@ -411,6 +414,30 @@ class Second(object):
|
|||||||
"isUpcoming": None
|
"isUpcoming": None
|
||||||
} for video in info["entries"] if "title" in video)
|
} 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
|
@cherrypy.expose
|
||||||
def vi(self, id, file):
|
def vi(self, id, file):
|
||||||
with requests.get("https://i.ytimg.com/vi/{}/{}".format(id, file)) as r:
|
with requests.get("https://i.ytimg.com/vi/{}/{}".format(id, file)) as r:
|
||||||
|
Loading…
Reference in New Issue
Block a user