[YadiSk] Cleanup

1 files changed, 85 insertions, 0 deletions

diff --git a/module/plugins/hoster/YadiSk.py b/module/plugins/hoster/YadiSk.py
new file mode 100644
index 000000000..c71d16cf2
--- /dev/null
+++ b/module/plugins/hoster/YadiSk.py
@@ -0,0 +1,85 @@
+# -*- coding: utf-8 -*-
+
+import re
+import pycurl
+import random
+
+from module.common.json_layer import json_loads
+from module.plugins.internal.SimpleHoster import SimpleHoster, create_getInfo
+
+
+class YadiSk(SimpleHoster):
+    __name__    = "YadiSk"
+    __type__    = "hoster"
+    __version__ = "0.02"
+
+    __pattern__ = r'https?://yadi\.sk/d/\w+'
+
+    __description__ = """Yadi.sk hoster plugin"""
+    __license__     = "GPLv3"
+    __authors__     = [("GammaC0de", "nomail@fakemailbox.com")]
+
+
+    OFFLINE_PATTERN = r'Nothing found'
+
+
+    def setup(self):
+        self.resumeDownload = False
+        self.multiDL        = False
+        self.chunkLimit     = 1
+
+
+    def handleFree(self, pyfile):
+        m = re.search(r'<script id="models-client" type="application/json">(.+?)</script>', self.html)
+        if m is None:
+            self.fail(_("could not find required json data"))
+
+        res = json_loads(m.group(1))
+
+        yadisk_ver  = None
+        yadisk_sk   = None
+        yadisk_id   = None
+        yadisk_size = None
+        yadisk_name = None
+        yadisk_hash = None
+        try:  #@TODO: Copy to apiInfo method
+            for sect in res:
+                if 'model' in sect:
+                    if sect['model'] == 'config':
+                        yadisk_ver = sect['data']['version']
+                        yadisk_sk  = sect['data']['sk']
+
+                    elif sect['model'] == 'resource':
+                        yadisk_id   = sect['data']['id']
+                        yadisk_size = sect['data']['meta']['size']
+                        yadisk_name = sect['data']['name']
+
+        except Exception:
+            self.fail(_("Unexpected server response"))
+
+        if None is in (yadisk_id, yadisk_sk, yadisk_id, yadisk_size, yadisk_name):
+           self.fail(_("json data is missing important information, cannot continue"))
+
+        self.pyfile.size = yadisk_size
+        self.pyfile.name = yadisk_name
+
+        yadisk_idclient = ""
+        for _i in range(1, 32):
+            yadisk_idclient += random.choice('0123456abcdef')
+
+        result_json = self.load("https://yadi.sk/models/?_m=do-get-resource-url",
+                                post={'idClient': yadisk_idclient,
+                                      'version' : yadisk_ver,
+                                      '_model.0': 'do-get-resource-url',
+                                      'sk'      : yadisk_sk,
+                                      'id.0'    : yadisk_id})
+
+        res = json_loads(result_json)
+        try:
+            self.link = res['models'][0]['data']['file']
+
+        except Exception:
+            self.fail(_("faild to retrieve the download url"))
+
+
+getInfo = create_getInfo(YadiSk)