mirror of https://git.sr.ht/~cadence/NewLeaf
Stream responses on /vi and /ggpht endpoints
The chunk_size=None parameter to iter_content lets us consume data as soon as it arrives https://docs.python-requests.org/en/master/api/#requests.Response.iter_content
This commit is contained in:
parent
66b7d1bec8
commit
0a13ab88cb
8
index.py
8
index.py
|
@ -123,17 +123,17 @@ class NewLeaf(object):
|
|||
|
||||
@cherrypy.expose
|
||||
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), stream=True) as r:
|
||||
r.raise_for_status()
|
||||
cherrypy.response.headers["content-type"] = r.headers["content-type"]
|
||||
return r # no idea if this is a good way to do it, but it definitely works! :D
|
||||
return next(r.iter_content(chunk_size=None))
|
||||
|
||||
@cherrypy.expose
|
||||
def ggpht(self, *path):
|
||||
with requests.get("https://yt3.ggpht.com/{}".format("/".join(path))) as r:
|
||||
with requests.get("https://yt3.ggpht.com/{}".format("/".join(path)), stream=True) as r:
|
||||
r.raise_for_status()
|
||||
cherrypy.response.headers["content-type"] = r.headers["content-type"]
|
||||
return r
|
||||
return next(r.iter_content(chunk_size=None))
|
||||
|
||||
bind_port = getattr(configuration, "bind_port", 3000)
|
||||
bind_host = getattr(configuration, "bind_host", "0.0.0.0")
|
||||
|
|
Loading…
Reference in New Issue