youtube-dl-gui/youtube_dl_gui/UpdateThread.py

#! /usr/bin/env python

from wx import CallAfter
from wx.lib.pubsub import setuparg1
from wx.lib.pubsub import pub as Publisher

from threading import Thread
from urllib2 import urlopen, URLError, HTTPError

from .Utils import (
  fix_path,
  file_exist,
  makedir
)

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

class UpdateThread(Thread):
  
  def __init__(self, updatePath, youtubeDLFile):
    super(UpdateThread, self).__init__()
    self.youtubeDLFile = youtubeDLFile
    self.updatePath = fix_path(updatePath)
    self.url = LATEST_YOUTUBE_DL + youtubeDLFile
    self.check_path()
    self.start()
    
  def run(self):
    CallAfter(Publisher.sendMessage, PUBLISHER_TOPIC, "Downloading latest youtube-dl. Please wait...")
    try:
      f = urlopen(self.url, timeout=DOWNLOAD_TIMEOUT)
      with open(self.updatePath + self.youtubeDLFile, 'wb') as lf:
	lf.write(f.read())
      msg = 'Youtube-dl downloaded correctly'
    except (HTTPError, URLError, IOError) as e:
      msg = 'Youtube-dl download failed ' + str(e)
    CallAfter(Publisher.sendMessage, PUBLISHER_TOPIC, msg)
    CallAfter(Publisher.sendMessage, PUBLISHER_TOPIC, 'finish')

  def check_path(self):
    if not file_exist(self.updatePath):
      makedir(self.updatePath)