CouchPotatoServer/couchpotato/core/providers/torrent/base.py

from couchpotato.core.helpers.variable import getImdb
from couchpotato.core.logger import CPLog
from couchpotato.core.providers.base import YarrProvider
import cookielib
import traceback
import urllib2

log = CPLog(__name__)


class TorrentProvider(YarrProvider):

    type = 'torrent'
    login_opener = None

    def imdbMatch(self, url, imdbId):
        if getImdb(url) == imdbId:
            return True

        if url[:4] == 'http':
            try:
                data = self.urlopen(url)
            except IOError:
                log.error('Failed to open %s.', url)
                return False

            return getImdb(data) == imdbId

        return False

    def login(self):

        try:
            cookiejar = cookielib.CookieJar()
            opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cookiejar))
            urllib2.install_opener(opener)
            f = opener.open(self.urls['login'], self.getLoginParam())
            f.read()
            f.close()
            self.login_opener = opener
            return True
        except:
            log.error('Failed to login %s: %s', (self.getName(), traceback.format_exc()))

        return False

    def download(self, url = '', nzb_id = ''):

        try:
            if not self.login_opener and not self.login():
                log.error('Failed downloading from %s', self.getName())

            return self.urlopen(url, opener = self.login_opener)
        except:
            log.error('Failed downloading from %s: %s', (self.getName(), traceback.format_exc()))