svtplay-dl/lib/svtplay_dl/service/solidtango.py

# ex:ts=4:sw=4:sts=4:et
# -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
from __future__ import absolute_import
import re

from svtplay_dl.service import Service
from svtplay_dl.fetcher.hls import hlsparse
from svtplay_dl.error import ServiceError


class Solidtango(Service):
    supported_domains = ['skkplay.se', 'skkplay.solidtango.com']

    def get(self, options):
        data = self.get_urldata()

        if self.exclude(options):
            yield ServiceError("Excluding video")
            return

        match = re.search(r'<title>(http[^<]+)</title>', data)
        if match:
            data = self.http.request("get", match.group(1)).text

        match = re.search('html5_source: "([^"]+)"', data)
        if match:
            streams = hlsparse(options, self.http.request("get", match.group(1)), match.group(1))
            for n in list(streams.keys()):
                yield streams[n]
        else:
            yield ServiceError("Can't find video info. if there is a video on the page. its a bug.")
            return