From 9c0c7d97b958c2ab281dbbeb448e17a20e6acc60 Mon Sep 17 00:00:00 2001
From: zoidberg10 <zoidberg@mujmail.cz>
Date: Wed, 5 Oct 2011 12:12:37 +0200
Subject: hoster plugin fixes; new plugins: mediafire, sendspace

---
 module/plugins/hoster/UloziskoSk.py | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

(limited to 'module/plugins/hoster/UloziskoSk.py')

diff --git a/module/plugins/hoster/UloziskoSk.py b/module/plugins/hoster/UloziskoSk.py
index b8ec0df55..e439a0f87 100644
--- a/module/plugins/hoster/UloziskoSk.py
+++ b/module/plugins/hoster/UloziskoSk.py
@@ -30,10 +30,19 @@ def getInfo(urls):
             result.append((url, 0, 1, url))
         else:
             # Get file info
+            name, size = url, 0
+
+            found = re.search(UloziskoSk.FILE_SIZE_PATTERN, html)
+            if found is not None:
+                size, units = found.groups()
+                size = float(size) * 1024 ** {'KB': 1, 'MB': 2, 'GB': 3}[units]
+
             found = re.search(UloziskoSk.FILE_NAME_PATTERN, html)
             if found is not None:
                 name = found.group(1)
-                result.append((name, 0, 2, url))
+
+            if found or size > 0:
+                result.append((name, size, 2, url))
     yield result
 
 
@@ -41,15 +50,16 @@ class UloziskoSk(Hoster):
     __name__ = "UloziskoSk"
     __type__ = "hoster"
     __pattern__ = r"http://(\w*\.)?ulozisko.sk/.*"
-    __version__ = "0.1"
+    __version__ = "0.2"
     __description__ = """Ulozisko.sk"""
     __author_name__ = ("zoidberg")
 
     URL_PATTERN = r'<form name = "formular" action = "([^"]+)" method = "post">'
     ID_PATTERN = r'<input type = "hidden" name = "id" value = "([^"]+)" />'
     FILE_NAME_PATTERN = r'<input type = "hidden" name = "name" value = "([^"]+)" />'
+    FILE_SIZE_PATTERN = ur'Veľkosť súboru: <strong>([0-9.]+) (KB|MB|GB)</strong><br />'
     CAPTCHA_PATTERN = r'<img src="(/obrazky/obrazky.php\?fid=[^"]+)" alt="" />'
-    FILE_OFFLINE_PATTERN = r'<span class = "red">Zadan� s�bor neexistuje z jedn�ho z nasleduj�cich d�vodov:</span>'
+    FILE_OFFLINE_PATTERN = ur'<span class = "red">Zadaný súbor neexistuje z jedného z nasledujúcich dôvodov:</span>'
 
     def setup(self):
         self.multiDL = False
@@ -69,6 +79,11 @@ class UloziskoSk(Hoster):
         if found is None:
             self.fail("Parse error (FILENAME)")
         pyfile.name = found.group(1)
+        
+        found = re.search(self.FILE_SIZE_PATTERN, self.html)
+        if found is not None:
+            size, units = found.groups()
+            pyfile.size = float(size) * 1024 ** {'KB': 1, 'MB': 2, 'GB': 3}[units]
 
         found = re.search(self.ID_PATTERN, self.html)
         if found is None:
@@ -92,4 +107,3 @@ class UloziskoSk(Hoster):
             "name": pyfile.name,
             "but": "++++STIAHNI+S%DABOR++++"
         })
-        
\ No newline at end of file
-- 
cgit v1.2.3