mirror of https://github.com/streamlink/streamlink
Merge pull request #774 from back-to/chaturbate
[chaturbate] Fixed plugin - New API
This commit is contained in:
commit
51b3efa45d
|
@ -1,37 +1,53 @@
|
||||||
import re
|
import re
|
||||||
|
import uuid
|
||||||
|
|
||||||
from streamlink.plugin import Plugin
|
from streamlink.plugin import Plugin
|
||||||
from streamlink.plugin.api import http, validate
|
from streamlink.plugin.api import http
|
||||||
|
from streamlink.plugin.api import validate
|
||||||
from streamlink.stream import HLSStream
|
from streamlink.stream import HLSStream
|
||||||
|
|
||||||
_url_re = re.compile(r"http(s)?://(\w+.)?chaturbate.com/[^/?&]+")
|
API_HLS = "https://chaturbate.com/get_edge_hls_url_ajax/"
|
||||||
_playlist_url_re = re.compile(r"var hlsSource\w+ = '(?P<url>[^']+)';")
|
|
||||||
_schema = validate.Schema(
|
_url_re = re.compile(r"https?://(\w+\.)?chaturbate\.com/(?P<username>\w+)")
|
||||||
validate.transform(_playlist_url_re.search),
|
|
||||||
validate.any(
|
_post_schema = validate.Schema(
|
||||||
None,
|
{
|
||||||
validate.all(
|
"url": validate.text,
|
||||||
validate.get("url"),
|
"room_status": validate.text,
|
||||||
validate.url(
|
"success": int
|
||||||
scheme="http",
|
}
|
||||||
path=validate.endswith(".m3u8")
|
|
||||||
)
|
|
||||||
)
|
|
||||||
)
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
class Chaturbate(Plugin):
|
class Chaturbate(Plugin):
|
||||||
@classmethod
|
@classmethod
|
||||||
def can_handle_url(self, url):
|
def can_handle_url(cls, url):
|
||||||
return _url_re.match(url)
|
return _url_re.match(url)
|
||||||
|
|
||||||
def _get_streams(self):
|
def _get_streams(self):
|
||||||
playlist_url = http.get(self.url, schema=_schema)
|
match = _url_re.match(self.url)
|
||||||
if not playlist_url:
|
username = match.group("username")
|
||||||
return
|
|
||||||
|
|
||||||
return HLSStream.parse_variant_playlist(self.session, playlist_url)
|
CSRFToken = str(uuid.uuid4().hex.upper()[0:32])
|
||||||
|
|
||||||
|
headers = {
|
||||||
|
"Content-Type": "application/x-www-form-urlencoded",
|
||||||
|
"X-CSRFToken": CSRFToken,
|
||||||
|
"X-Requested-With": "XMLHttpRequest",
|
||||||
|
"Referer": self.url,
|
||||||
|
}
|
||||||
|
|
||||||
|
cookies = {
|
||||||
|
"csrftoken": CSRFToken,
|
||||||
|
}
|
||||||
|
|
||||||
|
post_data = "room_slug={0}&bandwidth=high".format(username)
|
||||||
|
|
||||||
|
res = http.post(API_HLS, headers=headers, cookies=cookies, data=post_data)
|
||||||
|
data = http.json(res, schema=_post_schema)
|
||||||
|
|
||||||
|
if data["success"] is True and data["room_status"] == "public":
|
||||||
|
for s in HLSStream.parse_variant_playlist(self.session, data["url"]).items():
|
||||||
|
yield s
|
||||||
|
|
||||||
__plugin__ = Chaturbate
|
__plugin__ = Chaturbate
|
||||||
|
|
|
@ -0,0 +1,15 @@
|
||||||
|
import unittest
|
||||||
|
|
||||||
|
from streamlink.plugins.chaturbate import Chaturbate
|
||||||
|
|
||||||
|
|
||||||
|
class TestPluginChaturbate(unittest.TestCase):
|
||||||
|
def test_can_handle_url(self):
|
||||||
|
# should match
|
||||||
|
self.assertTrue(Chaturbate.can_handle_url("https://chaturbate.com/username"))
|
||||||
|
self.assertTrue(Chaturbate.can_handle_url("https://m.chaturbate.com/username"))
|
||||||
|
self.assertTrue(Chaturbate.can_handle_url("https://www.chaturbate.com/username"))
|
||||||
|
|
||||||
|
# shouldn't match
|
||||||
|
self.assertFalse(Chaturbate.can_handle_url("http://local.local/"))
|
||||||
|
self.assertFalse(Chaturbate.can_handle_url("http://localhost.localhost/"))
|
Loading…
Reference in New Issue