roypur/stream-api
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

retry content type

Browse Source
This commit is contained in:
Roy Olav Purser 2021-05-15 09:45:30 +02:00
parent 97563dec60
commit df153d5a12
Signed by: roypur
GPG Key ID: E14D26A036F21656
1 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
stream.py
View File

@ -48,14 +48,18 @@ class ProxyElem():
def __repr__(self):
return str(self.proxy)
async def content_type(self, url):
ctype = "binary/octet-stream"
try:
async with self.session() as session:
resp = await session.head(url)
ctype = resp.headers.get("Content-Type", "binary/octet-stream")
except Exception as e:
logger.info(e)
return ctype
ctype = None
async with self.session() as session:
for i in range(5):
try:
resp = await session.head(url)
ctype = resp.headers.get("Content-Type", None)
except Exception as e:
logger.info(e)
else:
if isinstance(ctype, str):
return ctype
return "binary/octet-type"
async def proxy_url(self, current, path):
jdata = None
data = {}
@ -348,7 +352,6 @@ class MainHandler(tornado.web.RequestHandler):
else:
upstream_proxy = await handler.proxy.proxy_url(upstream, None)
ctype = await handler.proxy.content_type(upstream_proxy)
logger.info(upstream_proxy)
data = None
if "mpegurl" in ctype.lower():
data = await rewrite(upstream, handler.proxy)