PirateNzb Fix

NZBmegasearch/SearchModules/am_PirateNzb.py

@@ -0,0 +1,152 @@
+# # ## # ## # ## # ## # ## # ## # ## # ## # ## # ## # ## # ## # ## # ## #
+# ~ This file is part of NZBmegasearch by 0byte.
+# ~
+# ~ NZBmegasearch is free software: you can redistribute it and/or modify
+# ~ it under the terms of the GNU General Public License as published by
+# ~ the Free Software Foundation, either version 3 of the License, or
+# ~ (at your option) any later version.
+# ~
+# ~ NZBmegasearch is distributed in the hope that it will be useful,
+# ~ but WITHOUT ANY WARRANTY; without even the implied warranty of
+# ~ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# ~ GNU General Public License for more details.
+# ~
+# ~ You should have received a copy of the GNU General Public License
+# ~ along with NZBmegasearch.  If not, see <http://www.gnu.org/licenses/>.
+# # ## # ## # ## # ## # ## # ## # ## # ## # ## # ## # ## # ## # ## # ## #
+import ConfigParser
+from SearchModule import *
+from urllib2 import urlparse
+import time
+
+
+
+class am_PirateNzb(SearchModule):
+    # Set up class variables
+    def __init__(self):
+        super(am_PirateNzb, self).__init__()
+        self.name = 'PirateNZB'
+        self.typesrch = 'PIR'
+        self.queryURL = 'https://piratenzb.com/api'
+        self.baseURL = 'https://piratenzb.com'
+        self.active = 0
+        self.builtin = 1
+        self.login = 1
+        self.inapi = 1
+        self.api_catsearch = 1
+        self.caption_login_user = 'user'
+        self.caption_login_pwd = 'api'
+        self.agent_headers = {
+        'User-Agent': 'Mozilla/5.0 (Windows NT 6.1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/41.0.2228.0 Safari/537.36'}
+
+        self.categories = { #https://piratenzb.com/api?t=caps needs updating
+        'Console': {'code': [1000, 1010, 1020, 1030, 1040, 1050, 1060, 1070, 1080], 'pretty': 'Console'},
+        'Movie': {'code': [2000, 2010, 2020, 2030, 2040, 250, 2060], 'pretty': 'Movie (ALL)'},
+        'Movie_HD': {'code': [2040, 2050, 2060, 2080], 'pretty': 'Movie (HD)'},
+        'Movie_SD': {'code': [2030, 2070], 'pretty': 'Movie (SD)'},
+        'Audio': {'code': [3000, 3010, 3020, 3030, 3040], 'pretty': 'Audio'},
+        'Games': {'code': [4050, ], 'pretty': 'Games'},
+        'PC': {'code': [4000, 4010, 4020, 4030, 4040, 4050, 4060, 4070], 'pretty': 'PC'},
+        'TV': {'code': [5000, 5020], 'pretty': 'TV'},
+        'TV_SD': {'code': [5030], 'pretty': 'SD'},
+        'TV_HD': {'code': [5040], 'pretty': 'HD'},
+        'XXX': {'code': [6000, 6010, 6020, 6030, 6040, 6060, 6070], 'pretty': 'XXX'},
+        'Anime': {'code': [5070], 'pretty': 'Anime'},
+        'Books': {'code': [7000, 7020], 'pretty': 'Books'},
+        'Mags': {'code': [7010], 'pretty': 'Magazines'},
+        'Ebook': {'code': [7020], 'pretty': 'Ebook'},
+        'Comics': {'code': [7030], 'pretty': 'Comics'},
+        }
+        self.category_inv = {}
+        for key in self.categories.keys():
+            prettyval = self.categories[key]['pretty']
+            for i in xrange(len(self.categories[key]['code'])):
+                val = self.categories[key]['code'][i]
+                self.category_inv[str(val)] = prettyval
+
+    # ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~
+
+    def search(self, queryString, cfg): #OMG
+
+        urlParams = dict(
+            apikey=cfg['pwd'],
+            t='search',
+            q=queryString,
+            o='json',
+            extended=1
+        )
+
+        return self.search_internal(urlParams, self.queryURL, cfg)
+
+    # ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~
+    def search_internal(self, urlParams, urlq, cfg):
+        timestamp_s = time.time()
+
+        try:
+            http_result = requests.get(url=urlq, params=urlParams, verify=False, timeout=cfg['timeout'])
+        except Exception as e:
+            log.critical(str(e))
+            if (cfg is not None):
+                cfg['retcode'] = [600, 'Server timeout', tout, self.name]
+            return []
+
+        timestamp_e = time.time()
+        log.info('TS ' + self.baseURL + " " + str(timestamp_e - timestamp_s))
+
+        try:
+            data = http_result.json()
+        except Exception as e:
+            if (cfg is not None):
+                cfg['retcode'] = [700, 'Server responded in unexpected format', timestamp_e - timestamp_s, self.name]
+            return []
+
+        parsed_data = []
+
+        if ('notice' in data):
+            log.info('Wrong api/pass ' + self.baseURL + " " + str(timestamp_e - timestamp_s))
+            if (cfg is not None):
+                cfg['retcode'] = [100, 'Incorrect user credentials', timestamp_e - timestamp_s, self.name]
+
+            return []
+        for i in xrange(len(data)):
+            if (('guid' in data[i]) and ('title' in data[i]) and ('size' in data[i]) and (
+                'pubDate' in data[i]) and ('description' in data[i])):
+
+                category_found = {}
+                if ('category' in data[i]):
+                    val = str(data[i]['category'])
+                    if (val in self.category_inv):
+                        category_found[self.category_inv[val]] = 1
+                if (len(category_found) == 0):
+                    category_found['N/A'] = 1
+
+                d1 = {
+                    'title': data[i]['title'],
+                    'poster': 'poster',
+                    'size': int(data[i]['size']),
+                    'url': data[i]['link'],
+                    'filelist_preview': '',
+                    'group': 'alt.binaries',
+                    'posting_date_timestamp': int(data[i]['pubDate']),
+                    'release_comments': data[i]['description'],
+                    'categ': data[i]['category'],
+                    'ignore': 0,
+                    'provider': self.baseURL,
+                    'providertitle': self.name
+                }
+
+                parsed_data.append(d1)
+
+        if (cfg is not None):
+            returncode = self.default_retcode
+            if (len(parsed_data) == 0 and len(data) < 300):
+               returncode = self.checkreturn(data)
+            returncode[2] = timestamp_e - timestamp_s
+            returncode[3] = self.name
+            cfg['retcode'] = copy.deepcopy(returncode)
+
+        return parsed_data
+
+    # ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~ ~
+
+