SickGear/sickbeard/clients/deluge.py

# Author: Mr_Orange <mr_orange@hotmail.it>
# URL: http://code.google.com/p/sickbeard/
#
# This file is part of SickGear.
#
# SickGear 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.
#
# SickGear 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 SickGear.  If not, see <http://www.gnu.org/licenses/>.

import json
from base64 import b64encode

import sickbeard
from sickbeard import logger
from sickbeard.clients.generic import GenericClient
from lib.requests.exceptions import RequestException


class DelugeAPI(GenericClient):
    def __init__(self, host=None, username=None, password=None):

        super(DelugeAPI, self).__init__('Deluge', host, username, password)

        self.url = self.host + 'json'

    def _get_auth(self):

        post_data = json.dumps({'method': 'auth.login',
                                'params': [self.password],
                                'id': 1})

        try:
            self.auth = self.session.post(
                self.url,
                data=post_data.encode('utf-8'),
                verify=sickbeard.TORRENT_VERIFY_CERT
            ).json()['result']

            post_data = json.dumps({'method': 'web.connected',
                                    'params': [],
                                    'id': 10})

            connected = self.session.post(
                self.url,
                data=post_data.encode('utf-8'),
                verify=sickbeard.TORRENT_VERIFY_CERT
            ).json()['result']

            if not connected:
                post_data = json.dumps({'method': 'web.get_hosts',
                                        'params': [],
                                        'id': 11})

                hosts = self.session.post(
                    self.url,
                    data=post_data.encode('utf-8'),
                    verify=sickbeard.TORRENT_VERIFY_CERT
                ).json()['result']

                if len(hosts) == 0:
                    logger.log(self.name + u': WebUI does not contain daemons',
                               logger.ERROR)
                    return None

                post_data = json.dumps({'method': 'web.connect',
                                        'params': [hosts[0][0]],
                                        'id': 11})
                self.session.post(self.url, data=post_data.encode('utf-8'),
                                  verify=sickbeard.TORRENT_VERIFY_CERT)

                post_data = json.dumps({'method': 'web.connected',
                                        'params': [],
                                        'id': 10})
                connected = self.session.post(
                    self.url,
                    data=post_data.encode('utf-8'),
                    verify=sickbeard.TORRENT_VERIFY_CERT
                ).json()['result']

            if not connected:
                logger.log(self.name + u': WebUI could not connect to daemon',
                           logger.ERROR)
                return None
        except RequestException:
            return None

        return self.auth

    def _add_torrent_uri(self, result):

        post_data = json.dumps({
            'method': 'core.add_torrent_magnet',
            'params': [result.url, {
                'move_completed': 'true',
                'move_completed_path': sickbeard.TV_DOWNLOAD_DIR
            }],
            'id': 2
        })
        result.hash = self._request(method='post',
                                    data=post_data).json()['result']

        return result.hash

    def _add_torrent_file(self, result):

        post_data = json.dumps({'method':
                                'core.add_torrent_file',
                                'params': [result.name + '.torrent',
                                           b64encode(result.content),
                                           {'move_completed': 'true',
                                            'move_completed_path':
                                            sickbeard.TV_DOWNLOAD_DIR}],
                                'id': 2})
        result.hash = self._request(method='post',
                                    data=post_data).json()['result']

        return result.hash

    def _set_torrent_label(self, result):

        label = sickbeard.TORRENT_LABEL
        if ' ' in label:
            logger.log(self.name +
                       u': Invalid label. Label must not contain a space',
                       logger.ERROR)
            return False

        if label:
            # check if label already exists and create it if not
            post_data = json.dumps({
                'method': 'label.get_labels',
                'params': [],
                'id': 3
            })
            labels = self._request(method='post',
                                   data=post_data).json()['result']

            if labels is not None:
                if label not in labels:
                    logger.log(self.name + ': ' + label +
                               u' label does not exist in ' +
                               u'Deluge we must add it',
                               logger.DEBUG)
                    post_data = json.dumps({
                        'method': 'label.add',
                        'params': [label],
                        'id': 4
                    })
                    self._request(method='post', data=post_data)
                    logger.log(self.name + ': ' + label +
                               u' label added to Deluge', logger.DEBUG)

                # add label to torrent
                post_data = json.dumps({
                    'method': 'label.set_torrent',
                    'params': [result.hash, label],
                    'id': 5
                })
                self._request(method='post', data=post_data)
                logger.log(self.name + ': ' + label +
                           u' label added to torrent',
                           logger.DEBUG)
            else:
                logger.log(self.name + ': ' +
                           u'label plugin not detected',
                           logger.DEBUG)
                return False

        return True

    def _set_torrent_ratio(self, result):

        ratio = None
        if result.ratio:
            ratio = result.ratio

        if ratio:
            post_data = json.dumps({'method': 'core.set_torrent_stop_at_ratio',
                                    'params': [result.hash, True],
                                    'id': 5})
            self._request(method='post', data=post_data)

            post_data = json.dumps({'method': 'core.set_torrent_stop_ratio',
                                    'params': [result.hash, float(ratio)],
                                    'id': 6})
            self._request(method='post', data=post_data)
        return True

    def _set_torrent_path(self, result):

        if sickbeard.TORRENT_PATH:
            post_data = json.dumps({
                'method': 'core.set_torrent_move_completed',
                'params': [result.hash, True],
                'id': 7
            })
            self._request(method='post', data=post_data)

            post_data = json.dumps({
                'method': 'core.set_torrent_move_completed_path',
                'params': [result.hash, sickbeard.TORRENT_PATH],
                'id': 8
            })
            self._request(method='post', data=post_data)
        return True

    def _set_torrent_pause(self, result):

        if sickbeard.TORRENT_PAUSED:
            post_data = json.dumps({'method': 'core.pause_torrent',
                                    'params': [[result.hash]],
                                    'id': 9})
            self._request(method='post', data=post_data)
        return True


api = DelugeAPI()