33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
import json
|
|
import requests
|
|
|
|
import engine.plugins
|
|
import engine.plugins.douyu
|
|
from engine.plugins import FFmpegdl
|
|
from common import logger
|
|
|
|
VALID_URL_BASE = r'(?:https?://)?(?:(?:www|m)\.)?huya\.com'
|
|
|
|
user_agent = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"
|
|
" (KHTML, like Gecko) Chrome/64.0.3282.140 Safari/537.36 Edge/17.17134"}
|
|
|
|
|
|
class Huya(FFmpegdl):
|
|
def __init__(self, fname, url, suffix='flv'):
|
|
super().__init__(fname, url, suffix)
|
|
|
|
def check_stream(self):
|
|
logger.debug(self.fname)
|
|
res = requests.get(self.url, timeout=5, headers=user_agent)
|
|
res.close()
|
|
data = res.text
|
|
huya = engine.plugins.match1(data, r'({"sCdnType":"TX".*?})')
|
|
if huya:
|
|
huyajson = json.loads(huya)
|
|
self.ydl_opts["absurl"] = huyajson["sFlvUrl"] + '/' + huyajson["sStreamName"] + '.' + \
|
|
huyajson["sFlvUrlSuffix"] + '?' + huyajson["sFlvAntiCode"] + "&ratio=2000"
|
|
return True
|
|
|
|
|
|
__plugin__ = Huya
|