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