youtube-dl-gui/youtube_dl_gui/updthread.py

#!/usr/bin/env python2

''' Youtube-dlG module to download youtube-dl. '''

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
)


class UpdateThread(Thread):

    '''
    Download latest youtube-dl.

    Params
        download_path: Absolute path where UpdateThread
                       should download the latest youtube-dl.

        quiet: If True UpdateThread won't send any messages back to caller.
    '''

    LATEST_YOUTUBE_DL = 'https://yt-dl.org/latest/'
    PUBLISHER_TOPIC = 'update'
    DOWNLOAD_TIMEOUT = 20

    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._callafter("Downloading latest youtube-dl. Please wait...")

        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())

            msg = 'Youtube-dl downloaded correctly'
        except (HTTPError, URLError, IOError) as e:
            msg = 'Youtube-dl download failed ' + str(e)

        self._callafter(msg)
        if not self.quiet:
            self._callafter('finish')

    def _callafter(self, data):
        ''' CallAfter wrapper. '''
        CallAfter(Publisher.sendMessage, self.PUBLISHER_TOPIC, data)