From 7fbc12ee4e6aea8a8ad0098deb03054976056371 Mon Sep 17 00:00:00 2001
From: Kirill Isakov <ukwt@ya.ru>
Date: Sat, 26 Mar 2016 05:28:58 +0600
Subject: Add Torrentz.eu search engine

---
 searx/engines/nyaa.py | 26 ++++++++++++++------------
 1 file changed, 14 insertions(+), 12 deletions(-)

(limited to 'searx/engines/nyaa.py')

diff --git a/searx/engines/nyaa.py b/searx/engines/nyaa.py
index 81caed76f..b0f884b77 100644
--- a/searx/engines/nyaa.py
+++ b/searx/engines/nyaa.py
@@ -43,6 +43,19 @@ def int_or_zero(num):
         return int(num)
     return 0
 
+# get multiplier to convert torrent size to bytes
+def get_filesize_mul(suffix):
+    return {
+        'KB': 1024,
+        'MB': 1024 ** 2,
+        'GB': 1024 ** 3,
+        'TB': 1024 ** 4,
+
+        'KIB': 1024,
+        'MIB': 1024 ** 2,
+        'GIB': 1024 ** 3,
+        'TIB': 1024 ** 4
+    }[str(suffix).upper()]
 
 # do search-request
 def request(query, params):
@@ -74,18 +87,7 @@ def response(resp):
         # torrent size
         try:
             file_size, suffix = result.xpath(xpath_filesize)[0].split(' ')
-
-            # convert torrent size to bytes.
-            # if there is no correct index in this dictionary,
-            # the try block fails as it should
-            multiplier = {
-                'KIB': 1024,
-                'MIB': 1024 ** 2,
-                'GIB': 1024 ** 3,
-                'TIB': 1024 ** 4
-            }[suffix.upper()]
-
-            file_size = int(float(file_size) * multiplier)
+            file_size = int(float(file_size) * get_filesize_mul(suffix))
         except Exception as e:
             file_size = None
 
-- 
cgit v1.2.3-54-g00ecf