diff --git a/servers/wstream.py b/servers/wstream.py
index 15114a7f..5b0d4cd3 100644
--- a/servers/wstream.py
+++ b/servers/wstream.py
@@ -28,17 +28,19 @@ def get_video_url(page_url, premium=False, user="", password="", video_password=
vid = scrapertools.find_multiple_matches(data, 'download_video.*?>.*?<.*?
([^\,,\s]+)')
headers.append(['Referer', page_url])
- post_data = scrapertools.find_single_match(data,
- "\s*")
+ post_data = scrapertools.find_single_match(data,"\s*")
if post_data != "":
from lib import jsunpack
data = jsunpack.unpack(post_data)
- media_url = scrapertools.find_multiple_matches(data, '(http.*?\.mp4)')
+ block = scrapertools.find_single_match(data, 'sources:\s*\[[^\]]+\]')
+ if block: data = block
+
+ media_urls = scrapertools.find_multiple_matches(data, '(http.*?\.mp4)')
_headers = urllib.urlencode(dict(headers))
i = 0
- for media_url in media_url:
+ for media_url in media_urls:
video_urls.append([vid[i] + " mp4 [wstream] ", media_url + '|' + _headers])
i = i + 1
|