Support for download-only option in webtorrent plugin

This commit is contained in:
Fabio Manganiello 2019-01-20 15:10:14 +01:00
parent d297b5cc42
commit 3c0ed443a2

View file

@ -119,7 +119,8 @@ class MediaWebtorrentPlugin(MediaPlugin):
return re.sub('\x1b\[((\d+m)|(.{1,2}))', '', line).strip() return re.sub('\x1b\[((\d+m)|(.{1,2}))', '', line).strip()
def _process_monitor(self, resource, download_dir, player_type, player_args): def _process_monitor(self, resource, download_dir, download_only,
player_type, player_args):
def _thread(): def _thread():
if not self._webtorrent_process: if not self._webtorrent_process:
return return
@ -191,7 +192,7 @@ class MediaWebtorrentPlugin(MediaPlugin):
if not output_dir: if not output_dir:
raise RuntimeError('Could not download torrent') raise RuntimeError('Could not download torrent')
if not media_file or not webtorrent_url: if not download_only and (not media_file or not webtorrent_url):
if not media_file: if not media_file:
self.logger.warning( self.logger.warning(
'The torrent does not contain any video files') 'The torrent does not contain any video files')
@ -203,7 +204,10 @@ class MediaWebtorrentPlugin(MediaPlugin):
except: pass except: pass
return return
# Then wait until we have enough chunks to start the player player = None
if not download_only:
# Wait until we have enough chunks to start the player
while True: while True:
result = poll.poll(0) result = poll.poll(0)
if not result: if not result:
@ -231,6 +235,7 @@ class MediaWebtorrentPlugin(MediaPlugin):
player.play(media, **player_args) player.play(media, **player_args)
self.logger.info('Waiting for player to terminate') self.logger.info('Waiting for player to terminate')
self._wait_for_player(player) self._wait_for_player(player)
self.logger.info('Torrent player terminated') self.logger.info('Torrent player terminated')
bus.post(TorrentDownloadCompletedEvent(resource=resource)) bus.post(TorrentDownloadCompletedEvent(resource=resource))
@ -242,9 +247,11 @@ class MediaWebtorrentPlugin(MediaPlugin):
return _thread return _thread
def _wait_for_player(self, player): def _wait_for_player(self, player):
media_cls = player.__class__.__name__
stop_evt = None stop_evt = None
if player:
media_cls = player.__class__.__name__
if media_cls == 'MediaMplayerPlugin': if media_cls == 'MediaMplayerPlugin':
stop_evt = player._mplayer_stopped_event stop_evt = player._mplayer_stopped_event
elif media_cls == 'MediaOmxplayerPlugin': elif media_cls == 'MediaOmxplayerPlugin':
@ -270,7 +277,7 @@ class MediaWebtorrentPlugin(MediaPlugin):
@action @action
def play(self, resource, player=None, **player_args): def play(self, resource, player=None, download_only=False, **player_args):
""" """
Download and stream a torrent Download and stream a torrent
@ -307,10 +314,16 @@ class MediaWebtorrentPlugin(MediaPlugin):
threading.Thread(target=self._process_monitor( threading.Thread(target=self._process_monitor(
resource=resource, download_dir=download_dir, resource=resource, download_dir=download_dir,
player_type=player, player_args=player_args)).start() player_type=player, player_args=player_args,
download_only=download_only)).start()
return { 'resource': resource } return { 'resource': resource }
@action
def download(resource, *args, **kwargs):
return self.play(resource, download_only=True, *args, **kwargs)
@action @action
def stop(self): def stop(self):
""" Stop the playback """ """ Stop the playback """