live stream test

stream.py

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