SickGear/sickgear/notifiers/pytivo.py

#
# 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 os

from .generic import BaseNotifier
import sickgear
from exceptions_helper import ex

from _23 import urlencode
# noinspection PyUnresolvedReferences
from six.moves import urllib


class PyTivoNotifier(BaseNotifier):

    def update_library(self, ep_obj=None, **kwargs):

        host = sickgear.PYTIVO_HOST
        share_name = sickgear.PYTIVO_SHARE_NAME
        tsn = sickgear.PYTIVO_TIVO_NAME

        # There are two more values required, the container and file.
        #
        # container: The share name, show name and season
        #
        # file: The file name
        #
        # Some slicing and dicing of variables is required to get at these values.
        #
        # There might be better ways to arrive at the values, but this is the best I have been able to
        # come up with.
        #

        # Calculated values

        show_path = ep_obj.show_obj.location
        show_name = ep_obj.show_obj.name
        root_show_and_season = os.path.dirname(ep_obj.location)
        abs_path = ep_obj.location

        # Some show names have colons in them which are illegal in a path location, so strip them out.
        # (Are there other characters?)
        show_name = show_name.replace(':', '')

        root = show_path.replace(show_name, '')
        show_and_season = root_show_and_season.replace(root, '')

        container = share_name + '/' + show_and_season
        file_path = '/' + abs_path.replace(root, '')

        # Finally create the url and make request
        request_url = 'http://%s/TiVoConnect?%s' % (host, urlencode(
            dict(Command='Push', Container=container, File=file_path, tsn=tsn)))

        self._log_debug(f'Requesting {request_url}')

        request = urllib.request.Request(request_url)

        try:
            http_response_obj = urllib.request.urlopen(request)  # PY2 http_response_obj has no `with` context manager
            http_response_obj.close()

        except urllib.error.HTTPError as e:
            if hasattr(e, 'reason'):
                self._log_error('Error, failed to reach a server - ' + e.reason)
                return False
            elif hasattr(e, 'code'):
                self._log_error('Error, the server couldn\'t fulfill the request - ' + e.code)
            return False

        except (BaseException, Exception) as e:
            self._log_error(f'Unknown exception: {ex(e)}')
            return False

        self._log('Successfully requested transfer of file')
        return True


notifier = PyTivoNotifier