Create wcostream.py

anime_downloader/extractors/wcostream.py

@@ -0,0 +1,37 @@
+from anime_downloader.extractors.base_extractor import BaseExtractor
+from anime_downloader.sites import helpers
+import re
+
+
+class WcoStream(BaseExtractor):
+    def _get_data(self):
+        try:
+            if self.url.startswith('https://vidstream.pro/e'):
+                base_url = 'https://vidstream.pro'
+            elif self.url.startswith('https://mcloud.to/e/'):
+                base_url = 'https://mcloud.to'
+            else:
+                return []
+
+            html = helpers.get(self.url, referer='https://wcostream.cc/')
+            id_ = re.findall(r"/e/(.*?)\?domain", self.url)[0]
+            skey = re.findall(r"skey\s=\s['\"](.*?)['\"];", html.text)[0]
+
+            apiLink = f"{base_url}/info/{id_}?domain=wcostream.cc&skey={skey}"
+            referer = f"{base_url}/e/{id_}?domain=wcostream.cc"
+
+            response = helpers.get(apiLink, referer=referer).json()
+
+            if response['success'] is True:
+                sources = [
+                    {
+                        'stream_url': x['file']
+                    }
+                    for x in response['media']['sources']
+                ]
+                return sources
+            else:
+                return []
+
+        except Exception:
+            return {"stream_url": ''}