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

live stream test

Browse Source
This commit is contained in:
Roy Olav Purser
2021-05-14 16:26:31 +02:00
parent 677648ecdd
commit c8d51d2e3d
1 changed files with 20 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
stream.py
View File

@@ -220,10 +220,12 @@ try:
except Exception as e:
logger.info(e)
def rewrite(current, provider, proxy):
resp = requests.get(current, proxies=proxy.req)
async def rewrite(current, provider, proxy):
async with proxy.session() as session:
resp = await session.get(current)
text = await resp.text()
ndata = None
if resp.text is not None:
if text is not None:
links = []
for line in resp.text.splitlines():
if line.startswith("#EXT-X-KEY:METHOD="):
@@ -244,19 +246,21 @@ def rewrite(current, provider, proxy):
links.append(ldata)
if isinstance(proxy_server, str):
ndata = ""
presp = requests.post(proxy_server, json=links)
if isinstance(presp.text, str):
links = json.loads(presp.text)
for line in resp.text.splitlines():
if line.startswith("#EXT-X-KEY:METHOD="):
matches = re.findall(r'(?<=URI=").+(?=")', line)
if len(matches) == 1:
new_url = links.pop(0)
ndata += re.sub(r'URI=".+"', f'URI="{new_url}"', line)
elif line.startswith("#"):
ndata += line
else:
ndata += links.pop(0)
async with proxy.session() as session:
resp = await session.post(proxy_server, json=links)
link_text = await resp.text()
if isinstance(link_text, str):
links = json.loads(link_text)
for line in text.splitlines():
if line.startswith("#EXT-X-KEY:METHOD="):
matches = re.findall(r'(?<=URI=").+(?=")', line)
if len(matches) == 1:
new_url = links.pop(0)
ndata += re.sub(r'URI=".+"', f'URI="{new_url}"', line)
elif line.startswith("#"):
ndata += line
else:
ndata += links.pop(0)
ndata += "\n"
return ndata
class MainHandler(tornado.web.RequestHandler):