Saverr/svtplay-dl
1
0
Fork 0
mirror of https://github.com/spaam/svtplay-dl.git synced 2024-11-24 04:05:39 +01:00
Code Issues Projects Releases Wiki Activity

tv4play: Support downloading all premium videos

Browse Source 
this fixes #275
This commit is contained in:
Johan Andersson 2015-10-05 19:43:57 +02:00
parent c7373854f9
commit e6baf3e4da
1 changed files with 31 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
lib/svtplay_dl/service/tv4play.py
View File

@ -35,19 +35,11 @@ class Tv4play(Service, OpenGraphThumbMixin):
return
if options.username and options.password:
data = self.http.request("get", "https://www.tv4play.se/session/new?https=")
auth_token = re.search('name="authenticity_token" ([a-z]+="[^"]+" )?value="([^"]+)"', data.text)
if not auth_token:
yield ServiceError("Can't find authenticity_token needed for user / password")
return
url = "https://www.tv4play.se/session"
postdata = {"user_name" : options.username, "password": options.password, "authenticity_token":auth_token.group(2), "https": ""}
data = self.http.request("post", url, data=postdata, cookies=self.cookies)
self.cookies = data.cookies
fail = re.search("<p class='failed-login'>([^<]+)</p>", data.text)
if fail:
yield ServiceError(fail.group(1))
work = self._login(options.username, options.password)
if isinstance(work, Exception):
yield work
return
url = "http://premium.tv4play.se/api/web/asset/%s/play" % vid
data = self.http.request("get", url, cookies=self.cookies)
if data.status_code == 401:
@ -124,6 +116,13 @@ class Tv4play(Service, OpenGraphThumbMixin):
yield streams[n]
def find_all_episodes(self, options):
premium = False
if options.username and options.password:
premium = self._login(options.username, options.password)
if isinstance(premium, Exception):
log.error(premium.message)
return None
parse = urlparse(self.url)
show = parse.path[parse.path.find("/", 1)+1:]
if not re.search("%", show):
@ -133,8 +132,13 @@ class Tv4play(Service, OpenGraphThumbMixin):
episodes = []
n = 1
for i in jsondata["results"]:
if premium:
text = "availability_group_premium"
else:
text = "availability_group_free"
try:
days = int(i["availability"]["availability_group_free"])
days = int(i["availability"][text])
except (ValueError, TypeError):
days = 999
if days > 0:
@ -148,6 +152,20 @@ class Tv4play(Service, OpenGraphThumbMixin):
return episodes
def _login(self, username, password):
data = self.http.request("get", "https://www.tv4play.se/session/new?https=")
auth_token = re.search('name="authenticity_token" ([a-z]+="[^"]+" )?value="([^"]+)"', data.text)
if not auth_token:
return ServiceError("Can't find authenticity_token needed for user / password")
url = "https://www.tv4play.se/session"
postdata = {"user_name" : username, "password": password, "authenticity_token":auth_token.group(2), "https": ""}
data = self.http.request("post", url, data=postdata, cookies=self.cookies)
self.cookies = data.cookies
fail = re.search("<p class='failed-login'>([^<]+)</p>", data.text)
if fail:
return ServiceError(fail.group(1))
return True
def findvid(url, data):
parse = urlparse(url)