CouchPotatoServer/couchpotato/core/providers/nzb/nzbx/main.py

from couchpotato.core.event import fireEvent
from couchpotato.core.helpers.encoding import toUnicode, tryUrlencode, \
    simplifyString
from couchpotato.core.helpers.rss import RSS
from couchpotato.core.helpers.variable import tryInt, getTitle
from couchpotato.core.logger import CPLog
from couchpotato.core.providers.nzb.base import NZBProvider
from couchpotato.environment import Env
from dateutil.parser import parse
import re
import time
import traceback
import json

log = CPLog(__name__)


class Nzbx(NZBProvider, RSS):
    endpoint = 'https://nzbx.co/api/'
    
    urls = {
        'search': 'https://nzbx.co/api/search',
        'details': 'https://nzbx.co/api/details?guid=%s',
        'comments': 'https://nzbx.co/api/get-comments?guid=%s',
        'ratings': 'https://nzbx.co/api/get-votes?guid=%s',
        'downloads': 'https://nzbx.co/api/get-downloads-count?guid=%s',
        'categories': 'https://nzbx.co/api/categories',
        'groups': 'https://nzbx.co/api/groups',
    }
    
    http_time_between_calls = 1 # Seconds

    def search(self, movie, quality):

        results = []
        if self.isDisabled():
            return results

        q = '"%s %s" %s' % (simplifyString(getTitle(movie['library'])), movie['library']['year'], quality.get('identifier'))
        arguments = tryUrlencode({
            'q': q,
            'l': 250, # Limit on number of files returned
            #'i': '', # index of file
            #'sf': '' # size filter
        })
        url = "%s?%s" % (self.urls['search'], arguments)

        cache_key = 'nzbx.%s.%s' % (movie['library']['identifier'], quality.get('identifier'))

        data = self.getCache(cache_key, url)
        if data:
            try:
                try:
                    nzbs = json.loads(data)
                except Exception, e:
                    log.debug('%s, %s', (self.getName(), e))
                    return results

                for nzb in nzbs:

                    nzbx_guid = nzb['guid']
                    
                    def extra_score(item):
                        score = 0
                        if item['votes']['upvotes'] > item['votes']['downvotes']:
                            score += 5
                            
                        return score
                    
                    new = {
                        'guid': nzbx_guid,
                        'type': 'nzb',
                        'provider': self.getName(),
                        'download': self.download,
                        'url': nzb['nzb'],
                        'name': nzb['name'],
                        'age': self.calculateAge(int(nzb['postdate'])),
                        'size': tryInt(nzb['size']) / 1024 / 1024,
                        'description': '',
                        'extra_score': extra_score,
                        'votes': nzb['votes'],
                        'check_nzb': True,
                    }

                    is_correct_movie = fireEvent('searcher.correct_movie',
                                                 nzb = new, movie = movie, quality = quality,
                                                 imdb_results = False, single = True)

                    if is_correct_movie:
                        new['score'] = fireEvent('score.calculate', new, movie, single = True)
                        results.append(new)
                        self.found(new)

                return results
            except:
                log.error('Failed to parsing %s: %s', (self.getName(), traceback.format_exc()))

        return results

    def isEnabled(self):
        return NZBProvider.isEnabled(self) and self.conf('enabled')