threadTrack.py

import sublime
from .salesforce import message

from . import util
from . import context
from .salesforce.lib.panel import Printer


class ThreadProgress():
    """
    Stolen from Package Control Source Code, ni dong de

    Animates an indicator, [=   ], in the status area while a thread runs

    :param thread:
        The thread to track for activity

    :param message:
        The message to display next to the activity indicator

    :param success_message:
        The message to display once the thread is complete
    """

    def __init__(self, api, thread, message, success_message, show_error=True):
        self.api = api
        self.thread = thread
        self.message = message
        self.success_message = success_message
        self.addend = 1
        self.size = 15
        self.show_error = show_error
        sublime.set_timeout(lambda: self.run(0), 100)

    def run(self, i):
        if not self.thread.is_alive():
            if hasattr(self.thread, 'result') and not self.thread.result:
                sublime.status_message('')
                return

            # After thread is end, display feedback to end user
            # according to response
            result = self.api.result
            if isinstance(result, dict) and "Error Message" in result:
                Printer.get('error').write(result["Error Message"])
            elif isinstance(result, dict) and "success" in result and not result["success"]:
                if not self.show_error: return
                Printer.get('error').write(message.SEPRATE.format(util.format_error_message(result)))
            else:
                sublime.status_message(self.success_message)
            return

        before = i % self.size
        after = (self.size - 1) - before

        sublime.status_message('%s [%s=%s]' % \
            (self.message, ' ' * before, ' ' * after))

        if not after:
            self.addend = -1
        if not before:
            self.addend = 1
        i += self.addend

        sublime.set_timeout(lambda: self.run(i), 100)

class ThreadsProgress():
    """
    Stolen from Package Control Source Code, ni dong de

    Animates an indicator, [=       ], in the status area while a thread runs

    :param threads:
        The threads to track for activity

    :param message:
        The message to display next to the activity indicator

    :param success_message:
        The message to display once the thread is complete
    """

    def __init__(self, threads, message, success_message):
        self.threads = threads
        self.message = message
        self.success_message = success_message
        self.addend = 1
        self.size = 15
        sublime.set_timeout(lambda: self.run(0), 100)

    def run(self, i):
        if self.is_threads_end(self.threads):
            sublime.status_message(self.success_message)
            return

        before = i % self.size
        after = (self.size - 1) - before

        sublime.status_message('%s [%s=%s]' % \
                               (self.message, ' ' * before, ' ' * after))

        if not after:
            self.addend = -1
        if not before:
            self.addend = 1
        i += self.addend

        sublime.set_timeout(lambda: self.run(i), 100)

    def is_threads_end(self, threads):
        """
        Indicate whether all threads are end

        @threads: Threads
        @return: Bool
        """

        is_alive = True
        for thread in threads:
            if thread.is_alive(): is_alive = False

        return is_alive