From c95617f707fb33183cda4f80fd6445972dd7b82c Mon Sep 17 00:00:00 2001
From: Alhaziel <tomas.massarenti@gmail.comm>
Date: Fri, 31 May 2019 18:16:36 +0200
Subject: [PATCH] Little Fix to Wstream

---
 servers/wstream.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

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.*?>.*?<.*?<td>([^\,,\s]+)')
 
     headers.append(['Referer', page_url])
-    post_data = scrapertools.find_single_match(data,
-                                               "</div>\s*<script type='text/javascript'>(eval.function.p,a,c,k,e,.*?)\s*</script>")
+    post_data = scrapertools.find_single_match(data,"</div>\s*<script type='text/javascript'>(eval.function.p,a,c,k,e,.*?)\s*</script>")
     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