add provider test
This commit is contained in:
		
							
								
								
									
										12
									
								
								stream.py
									
									
									
									
									
								
							
							
						
						
									
										12
									
								
								stream.py
									
									
									
									
									
								
							| @@ -113,10 +113,13 @@ class UpstreamHandler(): | ||||
|     async def setup(self, handler): | ||||
|         self.provider = handler.get_query_argument("provider", None) | ||||
|         render_str = handler.get_query_argument("render", "false") | ||||
|         if render_str.lower() == "true": | ||||
|             self.render = True | ||||
|         self.render = False | ||||
|         self.stream = False | ||||
|         if self.provider in providers.keys(): | ||||
|             if render_str.lower() == "true": | ||||
|                 self.render = True | ||||
|             else: | ||||
|                 self.stream = True | ||||
|             path = handler.request.path | ||||
|             if self.provider == "youtube": | ||||
|                 path = path.strip("/") | ||||
| @@ -316,9 +319,8 @@ class MainHandler(tornado.web.RequestHandler): | ||||
|  | ||||
|     async def handle_stream(self, handler, redir): | ||||
|         upstream = None | ||||
|         if handler.proxy is not None: | ||||
|         try: | ||||
|                 logger.info(proxy) | ||||
|             logger.info(handler.proxy) | ||||
|             streams = handler.proxy.stream().streams(handler.upstream) | ||||
|         except Exception as e: | ||||
|             logger.info(e) | ||||
| @@ -345,7 +347,7 @@ class MainHandler(tornado.web.RequestHandler): | ||||
|             logger.info(upstream) | ||||
|             data = None | ||||
|             if "mpegurl" in ctype.lower(): | ||||
|                 data = await rewrite(upstream, proxy) | ||||
|                 data = await rewrite(upstream, handler.proxy) | ||||
|                 logger.info(data) | ||||
|                 if isinstance(data, str): | ||||
|                     self.set_header("Content-Type", "application/vnd.apple.mpegurl") | ||||
|   | ||||
		Reference in New Issue
	
	Block a user