searx

commit: 1e2b60f3805a1cdc87cff0c5753bd355ad00cc98
parent e74aaa781e836eeac73af68aa1b4f5014b58b4d3
Author: misnyo <misnyo@misnyo.eu>
Date:   Thu, 31 Aug 2017 21:31:40 +0200

[mod] tokyotoshokan dependecy to other engine removed

Diffstat:
M
searx/engines/tokyotoshokan.py
16
+++++++++++++---

1 file changed, 13 insertions(+), 3 deletions(-)
diff --git a/searx/engines/tokyotoshokan.py b/searx/engines/tokyotoshokan.py
@@ -14,8 +14,8 @@ import re
 from lxml import html
 from searx.engines.xpath import extract_text
 from datetime import datetime
-from searx.engines.nyaa import int_or_zero, get_filesize_mul
 from searx.url_utils import urlencode
+from searx.utils import get_torrent_size
 
 # engine dependent config
 categories = ['files', 'videos', 'music']
@@ -26,6 +26,17 @@ base_url = 'https://www.tokyotosho.info/'
 search_url = base_url + 'search.php?{query}'
 
 
+# convert a variable to integer or return 0 if it's not a number
+def int_or_zero(num):
+    if isinstance(num, list):
+        if len(num) < 1:
+            return 0
+        num = num[0]
+    if num.isdigit():
+        return int(num)
+    return 0
+
+
 # do search-request
 def request(query, params):
     query = urlencode({'page': params['pageno'], 'terms': query})
@@ -76,8 +87,7 @@ def response(resp):
                 try:
                     # ('1.228', 'GB')
                     groups = size_re.match(item).groups()
-                    multiplier = get_filesize_mul(groups[1])
-                    params['filesize'] = int(multiplier * float(groups[0]))
+                    params['filesize'] = get_torrent_size(groups[0], groups[1])
                 except:
                     pass
             elif item.startswith('Date:'):