CouchPotatoServer/couchpotato/core/downloaders/blackhole/main.py

from couchpotato.core.downloaders.base import Downloader
from couchpotato.core.logger import CPLog
import os
import traceback

log = CPLog(__name__)

class Blackhole(Downloader):

    type = ['nzb', 'torrent']

    def download(self, data = {}, movie = {}):
        if self.isDisabled() or (not self.isCorrectType(data.get('type')) or (not self.conf('use_for') in ['both', data.get('type')])):
            return

        directory = self.conf('directory')
        if not directory or not os.path.isdir(directory):
            log.error('No directory set for blackhole %s download.' % data.get('type'))
        else:
            try:
                filedata = data.get('download')(url = data.get('url'), nzb_id = data.get('id'))

                if len(filedata) < 50:
                    log.error('No nzb available!')
                    return False

                fullPath = os.path.join(directory, self.createFileName(data, filedata, movie))

                try:
                    if not os.path.isfile(fullPath):
                        log.info('Downloading %s to %s.' % (data.get('type'), fullPath))
                        with open(fullPath, 'wb') as f:
                            f.write(filedata)
                        return True
                    else:
                        log.info('File %s already exists.' % fullPath)
                        return True

                except:
                    log.error('Failed to download to blackhole %s' % traceback.format_exc())
                    pass

            except:
                log.debug('Failed to download file: %s' % data.get('name'))
                return False
        return False