youtube-dl-gui/youtube_dl_gui/OutputHandler.py

#! /usr/bin/env python

from .Utils import (
    remove_empty_items,
    string_to_array,
    get_filename
)

class DownloadHandler():

    def __init__(self, ListCtrl):
        self.ListCtrl = ListCtrl
        self.finished = False
        self.closed = False
        self.error = False
        self.init_handlers()

    def init_handlers(self):
        ''' Initialise handlers '''
        self.handlers = [None for i in range(self.ListCtrl.ListIndex)]

    def _has_closed(self):
        return self.closed

    def _has_finished(self):
        return self.finished

    def _has_error(self):
        return self.error

    def _add_empty_handler(self):
        self.handlers.append(None)

    def handle(self, msg):
        ''' Handles msg base to Signals.txt '''
        pack = msg.data
        if pack.index == -1:
            self.global_handler(pack)
        else:
            self.index_handler(pack)

    def global_handler(self, pack):
        ''' Manage global index = -1 '''
        if pack.header == 'close':
            self.closed = True
        elif pack.header == 'finish':
            self.finished = True

    def index_handler(self, pack):
        ''' Manage handlers base on index '''
        if self.handlers[pack.index] == None:
            self.handlers[pack.index] = IndexDownloadHandler(self.ListCtrl, pack.index)
        self.handlers[pack.index].handle(pack)
        if self.handlers[pack.index].has_error():
            self.error = True

class IndexDownloadHandler():

    def __init__(self, ListCtrl, index):
        self.ListCtrl = ListCtrl
        self.index = index
        self.err = False
        self.info = ''

    def has_error(self):
        return self.err

    def handle(self, pack):
        ''' Handle its pack for current index '''
        if pack.header == 'finish':
            self.finish()
        elif pack.header == 'close':
            self.close()
        elif pack.header == 'error':
            self.error()
        elif pack.header == 'playlist':
            self.playlist(pack.data)
        elif pack.header == 'youtube':
            self.pre_proc()
        elif pack.header == 'download':
            self.download(pack.data)
        elif pack.header == 'ffmpeg':
            self.post_proc()
        elif pack.header == 'remove':
            self.remove()
        elif pack.header == 'filename':
            self.filename(pack.data)

    def finish(self):
        self.ListCtrl._write_data(self.index, 4, '')
        self.ListCtrl._write_data(self.index, 5, 'Finished')

    def close(self):
        self.ListCtrl._write_data(self.index, 3, '')
        self.ListCtrl._write_data(self.index, 4, '')
        self.ListCtrl._write_data(self.index, 5, 'Stopped')

    def error(self):
        self.err = True
        self.ListCtrl._write_data(self.index, 3, '')
        self.ListCtrl._write_data(self.index, 4, '')
        self.ListCtrl._write_data(self.index, 5, 'Error')

    def pre_proc(self):
        self.ListCtrl._write_data(self.index, 5, 'Pre-Processing %s' % self.info)

    def post_proc(self):
        self.ListCtrl._write_data(self.index, 4, '')
        self.ListCtrl._write_data(self.index, 5, 'Post-Processing %s' % self.info)

    def download(self, data):
        self.ListCtrl._write_data(self.index, 1, data[0])
        self.ListCtrl._write_data(self.index, 2, data[1])
        self.ListCtrl._write_data(self.index, 3, data[2])
        self.ListCtrl._write_data(self.index, 4, data[3])
        self.ListCtrl._write_data(self.index, 5, 'Downloading %s' % self.info)

    def playlist(self, data):
        self.ListCtrl._write_data(self.index, 1, '')
        self.ListCtrl._write_data(self.index, 2, '')
        self.ListCtrl._write_data(self.index, 3, '')
        self.ListCtrl._write_data(self.index, 4, '')
        self.info = '%s/%s' % (data[0], data[1])

    def remove(self):
        self.ListCtrl._write_data(self.index, 5, 'Removing DASH Files')

    def filename(self, fl):
        self.ListCtrl._write_data(self.index, 0, get_filename(fl))

class DataPack():

    def __init__(self, header, index=-1, data=None):
        self.header = header
        self.index = index
        self.data = data

class OutputFormatter():

    def __init__(self, stdout):
        dataPack = None

        self.stdout = remove_empty_items(string_to_array(stdout))
        # extract header from stdout
        header = self.extract_header()
        # extract special headers filename, playlist
        header = self.set_filename_header(header)
        header = self.set_playlist_header(header)
        # extract data base on header
        data = self.extract_data(header)
        # extract special ignore header base on header, data
        header = self.set_ignore_header(header, data)
        # create data pack
        self.dataPack = DataPack(header, data=data)

    def extract_header(self):
        header = self.stdout.pop(0).replace('[', '').replace(']', '')
        return header

    def extract_data(self, header):
        ''' Extract data base on header '''
        if header == 'download':
            if '%' in self.stdout[0]:
                if self.stdout[0] != '100%':
                    ''' size, percent, eta, speed '''
                    return [self.stdout[2], self.stdout[0], self.stdout[6], self.stdout[4]]
        if header == 'playlist':
            return [self.stdout[2], self.stdout[4]]
        if header == 'filename':
            return ' '.join(self.stdout[1:])
        return None

    def set_filename_header(self, header):
        if header != 'ffmpeg':
            if self.stdout[0] == 'Destination:':
                return 'filename'
        return header

    def set_playlist_header(self, header):
        if header == 'download':
            if self.stdout[0] == 'Downloading' and self.stdout[1] == 'video':
                return 'playlist'
        return header

    def set_ignore_header(self, header, data):
        if header == 'download' and data == None:
            return 'ignore'
        return header

    def get_data(self):
        return self.dataPack