pesterchum/toast.py

import inspect
import threading
import time
from PyQt4 import QtGui, QtCore

try:
    import pynotify
    pynote = True
except:
    pynote = False

class DefaultToast(object):
    def __init__(self, title, msg, icon):
        self.title = title
        self.msg   = msg
        self.icon  = icon
    def show(self):
        print self.title, self.msg, self.icon

class ToastMachine(object):
    class __Toast__(object):
        def __init__(self, parent, window, title, msg, time=3000, icon="", importance=0):
            self.parent     = parent
            self.mainwindow = window
            self.title      = title
            self.msg        = msg
            self.time       = time
            self.icon       = icon
            self.importance = importance
            if inspect.ismethod(self.title) or inspect.isfunction(self.title):
                self.title = self.title()

        def titleM(self, title=None):
            if title:
                self.title = title
                if inspect.ismethod(self.title) or inspect.isfunction(self.title):
                    self.title = self.title()
            else:     return self.title
        def msgM(self, msg=None):
            if msg:   self.msg = msg
            else:     return self.msg
        def timeM(self, time=None):
            if time:  self.time = time
            else:     return self.time
        def iconM(self, icon=None):
            if icon:  self.icon = icon
            else:     return self.icon
        def importanceM(self, importance=None):
            if importance != None: self.importance = importance
            else:                  return self.importance

        def show(self):
            # Use libnotify's queue if using libnotify
            if self.parent.toasts and self.parent.type != "libnotify":
                self.parent.toasts.append(self)
            else:
                self.realShow()

        def realShow(self):
            t = None
            for (k,v) in self.parent.types.iteritems():
                if self.parent.type == k:
                    t = v(self.title, self.msg, self.icon)
                    # Use libnotify's urgency setting
                    if k == "libnotify":
                        if self.importance < 0:
                            t.set_urgency(pynotify.URGENCY_CRITICAL)
                        elif self.importance == 0:
                            t.set_urgency(pynotify.URGENCY_NORMAL)
                        elif self.importance > 0:
                            t.set_urgency(pynotify.URGENCY_LOW)
                    break
            if not t:
                if 'default' in self.parent.types:
                    if 'parent' in inspect.getargspec(self.parent.types['default']).args:
                        t = self.parent.types['default'](self.title, self.msg, self.icon, self.mainwindow)
                    else:
                        t = self.parent.types['default'](self.title, self.msg, self.icon)
                else:
                    t = DefaultToast(self.title, self.msg, self.icon)
            t.show()
            print "SLEEPING"
            #time.sleep(self.time/1000)
            if self in self.parent.toasts:
                self.parent.toasts.remove(self)

    def __init__(self, parent, name, type="default", types={'default'  : DefaultToast,
                                                            'libnotify': pynotify.Notification},
                                                     extras={}):
        self.mainwindow = parent
        self.name       = name
        types.update(extras)
        self.types      = types
        self.type       = type
        self.quit       = False

        if type == "libnotify":
            try:
                if not pynote or not pynotify.init("ToastMachine"):
                    raise Exception
            except:
                print "Problem initilizing pynotify"
                self.type = type = "default"

        self.toasts = []

    def Toast(self, title, msg, icon=""):
        return self.__Toast__(self, self.mainwindow, title, msg, time=0, icon=icon)

    def appName(self):
        if inspect.ismethod(self.name) or inspect.isfunction(self.name):
            return self.name()
        else:
            return self.name

    def showNext(self):
        high   = filter(lambda x: x.importance < 0, self.toasts)
        normal = filter(lambda x: x.importance == 0, self.toasts)
        low    = filter(lambda x: x.importance > 0, self.toasts)

        if high:
            high.sort(key=lambda x: x.importance)
            high[0].realShow()
        elif normal:
            normal[0].realShow()
        elif low:
            low.sort(key=lambda x: x.importance)
            low[0].realShow()

    def showAll(self):
        while self.toasts:
            self.showNext()

    def run(self):
        while not self.quit:
            if self.toasts:
                self.showNext()


class PesterToast(QtGui.QFrame, DefaultToast):
    def __init__(self, title, msg, icon, parent=None):
        QtGui.QFrame.__init__(self, parent,
                              (QtCore.Qt.CustomizeWindowHint |
                               QtCore.Qt.FramelessWindowHint))
        #self.setAttribute(QtCore.Qt.WA_QuitOnClose, False)
        #self.setAttribute(QtCore.Qt.WA_TranslucentBackground, True)
        #self.setAttribute(QtCore.Qt.WA_NoSystemBackground, True)
        self.setObjectName("toast")
        self.setWindowTitle("toast")
        #self.setFocusPolicy(QtCore.Qt.ClickFocus)

        self.title = QtGui.QLabel(title, self)
        self.msg = QtGui.QLabel(msg, self)

        self.btn = QtGui.QPushButton("Push Me", self)
        self.connect(self.btn, QtCore.SIGNAL('clicked()'),
                     self, QtCore.SLOT('close()'))

        layout_0 = QtGui.QVBoxLayout()
        layout_0.addWidget(self.title)
        layout_0.addWidget(self.btn)
        layout_0.addWidget(self.msg)

        self.setLayout(layout_0)

        print self.isWindow()

    def show(self):
        QtGui.QFrame.setVisible(self, True)
        print "SHOWING"
        #~ themeWarning = QtGui.QMessageBox(self)
        #~ themeWarning.setText("ASDFASD")
        #~ themeWarning.exec_()

class PesterToastMachine(ToastMachine, QtCore.QObject):
    def __init__(self, parent, name, type="default",
                 types={'default'  : DefaultToast,
                        'libnotify': pynotify.Notification},
                 extras={}):
        ToastMachine.__init__(self, parent, name, type, types, extras)
        QtCore.QObject.__init__(self, parent)

    @QtCore.pyqtSlot()
    def showNext(self):
        ToastMachine.showNext(self)

    def run(self):
        self.timer = QtCore.QTimer(self)
        self.connect(self.timer, QtCore.SIGNAL('timeout()'),
                     self, QtCore.SLOT('showNext()'))
        #self.timer.start(1000)