You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 

96 lines
2.8 KiB

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""Youtubedlg module to update youtube-dl binary.
Attributes:
UPDATE_PUB_TOPIC (string): wxPublisher subscription topic of the
UpdateThread thread.
"""
from __future__ import unicode_literals
import os.path
from threading import Thread
from urllib2 import urlopen, URLError, HTTPError
from wx import CallAfter
from wx.lib.pubsub import setuparg1
from wx.lib.pubsub import pub as Publisher
from .utils import (
YOUTUBEDL_BIN,
check_path
)
UPDATE_PUB_TOPIC = 'update'
class UpdateThread(Thread):
"""Python Thread that downloads youtube-dl binary.
Attributes:
LATEST_YOUTUBE_DL (string): URL with the latest youtube-dl binary.
DOWNLOAD_TIMEOUT (int): Download timeout in seconds.
Args:
download_path (string): Absolute path where UpdateThread will download
the latest youtube-dl.
quiet (boolean): If True UpdateThread won't send the finish signal
back to the caller. Finish signal can be used to make sure that
the UpdateThread has been completed in an asynchronous way.
"""
LATEST_YOUTUBE_DL = 'https://yt-dl.org/latest/'
DOWNLOAD_TIMEOUT = 10
def __init__(self, download_path, quiet=False):
super(UpdateThread, self).__init__()
self.download_path = download_path
self.quiet = quiet
self.start()
def run(self):
self._talk_to_gui('download')
source_file = self.LATEST_YOUTUBE_DL + YOUTUBEDL_BIN
destination_file = os.path.join(self.download_path, YOUTUBEDL_BIN)
check_path(self.download_path)
try:
stream = urlopen(source_file, timeout=self.DOWNLOAD_TIMEOUT)
with open(destination_file, 'wb') as dest_file:
dest_file.write(stream.read())
self._talk_to_gui('correct')
except (HTTPError, URLError, IOError) as error:
self._talk_to_gui('error', unicode(error))
if not self.quiet:
self._talk_to_gui('finish')
def _talk_to_gui(self, signal, data=None):
"""Communicate with the GUI using wxCallAfter and wxPublisher.
Args:
signal (string): Unique signal string that informs the GUI for the
update process.
data (string): Can be any string data to pass along with the
given signal. Default is None.
Note:
UpdateThread supports 4 signals.
1) download: The update process started
2) correct: The update process completed successfully
3) error: An error occured while downloading youtube-dl binary
4) finish: The update thread is ready to join
"""
CallAfter(Publisher.sendMessage, UPDATE_PUB_TOPIC, (signal, data))