403 lines
15 KiB
Python
403 lines
15 KiB
Python
import inspect
|
|
import threading
|
|
import time, os
|
|
import ostools
|
|
import logging
|
|
from PyQt5 import QtCore, QtGui, QtWidgets
|
|
|
|
try:
|
|
import pynotify
|
|
except:
|
|
pynotify = None
|
|
|
|
class DefaultToast(object):
|
|
def __init__(self, machine, title, msg, icon):
|
|
self.machine = machine
|
|
self.title = title
|
|
self.msg = msg
|
|
self.icon = icon
|
|
def show(self):
|
|
print(self.title, self.msg, self.icon)
|
|
self.done()
|
|
def done(self):
|
|
t = self.machine.toasts[0]
|
|
if t.title == self.title and t.msg == self.msg and t.icon == self.icon:
|
|
self.machine.toasts.pop(0)
|
|
self.machine.displaying = False
|
|
print("Done")
|
|
|
|
class ToastMachine(object):
|
|
class __Toast__(object):
|
|
def __init__(self, machine, title, msg, time=3000, icon="", importance=0):
|
|
self.machine = machine
|
|
self.title = title
|
|
self.msg = msg
|
|
self.time = time
|
|
if icon:
|
|
icon = os.path.abspath(icon)
|
|
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):
|
|
if self.machine.on:
|
|
# Use libnotify's queue if using libnotify
|
|
if self.machine.type == "libnotify" or self.machine.type == "twmn":
|
|
self.realShow()
|
|
elif self.machine.toasts:
|
|
self.machine.toasts.append(self)
|
|
else:
|
|
self.machine.toasts.append(self)
|
|
self.realShow()
|
|
|
|
def realShow(self):
|
|
self.machine.displaying = True
|
|
t = None
|
|
for (k,v) in self.machine.types.items():
|
|
if self.machine.type == k:
|
|
try:
|
|
args = inspect.getargspec(v.__init__).args
|
|
except:
|
|
args = []
|
|
|
|
extras = {}
|
|
if 'parent' in args:
|
|
extras['parent'] = self.machine.parent
|
|
if 'time' in args:
|
|
extras['time'] = self.time
|
|
if k == "libnotify" or k == "twmn":
|
|
t = v(self.title, self.msg, self.icon, **extras)
|
|
else:
|
|
t = v(self.machine, self.title, self.msg, self.icon, **extras)
|
|
# 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.machine.types:
|
|
if 'parent' in inspect.getargspec(self.machine.types['default'].__init__).args:
|
|
t = self.machine.types['default'](self.machine, self.title, self.msg, self.icon, self.machine.parent)
|
|
else:
|
|
t = self.machine.types['default'](self.machine, self.title, self.msg, self.icon)
|
|
else:
|
|
t = DefaultToast(self.title, self.msg, self.icon)
|
|
t.show()
|
|
|
|
def __init__(self, parent, name, on=True, type="default",
|
|
types=({'default' : DefaultToast,
|
|
'libnotify': pynotify.Notification}
|
|
if pynotify else
|
|
{'default' : DefaultToast}),
|
|
extras={}):
|
|
self.parent = parent
|
|
self.name = name
|
|
self.on = on
|
|
types.update(extras)
|
|
self.types = types
|
|
self.type = "default"
|
|
self.quit = False
|
|
self.displaying = False
|
|
|
|
self.setCurrentType(type)
|
|
|
|
self.toasts = []
|
|
|
|
def Toast(self, title, msg, icon="", time=3000):
|
|
return self.__Toast__(self, title, msg, time=time, icon=icon)
|
|
|
|
def setEnabled(self, on):
|
|
self.on = (on is True)
|
|
|
|
def currentType(self):
|
|
return self.type
|
|
|
|
def availableTypes(self):
|
|
return sorted(self.types.keys())
|
|
|
|
def setCurrentType(self, type):
|
|
if type in self.types:
|
|
if type == "libnotify":
|
|
if not pynotify or not pynotify.init("ToastMachine"):
|
|
print("Problem initilizing pynotify")
|
|
return
|
|
#self.type = type = "default"
|
|
elif type == "twmn":
|
|
import pytwmn
|
|
try:
|
|
pytwmn.init()
|
|
except pytwmn.ERROR as e:
|
|
print("Problem initilizing pytwmn: " + str(e))
|
|
return
|
|
#self.type = type = "default"
|
|
self.type = type
|
|
|
|
def appName(self):
|
|
if inspect.ismethod(self.name) or inspect.isfunction(self.name):
|
|
return self.name()
|
|
else:
|
|
return self.name
|
|
|
|
def showNext(self):
|
|
if not self.displaying and self.toasts:
|
|
self.toasts.sort(key=lambda x: x.importance)
|
|
self.toasts[0].realShow()
|
|
|
|
def showAll(self):
|
|
while self.toasts:
|
|
self.showNext()
|
|
|
|
def run(self):
|
|
while not self.quit:
|
|
if self.on and self.toasts:
|
|
self.showNext()
|
|
|
|
|
|
class PesterToast(QtWidgets.QWidget, DefaultToast):
|
|
def __init__(self, machine, title, msg, icon, time=3000, parent=None):
|
|
logging.info(isinstance(parent, QtWidgets.QWidget))
|
|
kwds = dict(machine=machine, title=title, msg=msg, icon=icon)
|
|
super().__init__(parent, **kwds)
|
|
|
|
self.machine = machine
|
|
self.time = time
|
|
|
|
if ostools.isWin32():
|
|
self.setWindowFlags(QtCore.Qt.ToolTip)
|
|
else:
|
|
self.setWindowFlags(QtCore.Qt.WindowStaysOnTopHint | QtCore.Qt.X11BypassWindowManagerHint | QtCore.Qt.ToolTip)
|
|
|
|
self.m_animation = QtCore.QParallelAnimationGroup()
|
|
anim = QtCore.QPropertyAnimation(self)
|
|
anim.setTargetObject(self)
|
|
self.m_animation.addAnimation(anim)
|
|
anim.setEasingCurve(QtCore.QEasingCurve.OutBounce)
|
|
anim.setDuration(1000)
|
|
anim.finished.connect(self.reverseTrigger)
|
|
|
|
self.m_animation.setDirection(QtCore.QAnimationGroup.Forward)
|
|
|
|
self.title = QtWidgets.QLabel(title, self)
|
|
self.msg = QtWidgets.QLabel(msg, self)
|
|
self.content = msg
|
|
if icon:
|
|
self.icon = QtWidgets.QLabel("")
|
|
self.icon.setPixmap(QtGui.QPixmap(icon).scaledToWidth(30))
|
|
else:
|
|
self.icon = QtWidgets.QLabel("")
|
|
self.icon.setPixmap(QtGui.QPixmap(30, 30))
|
|
self.icon.pixmap().fill(QtGui.QColor(0,0,0,0))
|
|
|
|
layout_0 = QtWidgets.QVBoxLayout()
|
|
layout_0.setContentsMargins(0, 0, 0, 0)
|
|
layout_0.setContentsMargins(0, 0, 0, 0)
|
|
|
|
if self.icon:
|
|
layout_1 = QtWidgets.QGridLayout()
|
|
layout_1.addWidget(self.icon, 0,0, 1,1)
|
|
layout_1.addWidget(self.title, 0,1, 1,7)
|
|
layout_1.setAlignment(self.msg, QtCore.Qt.AlignTop)
|
|
layout_0.addLayout(layout_1)
|
|
else:
|
|
layout_0.addWidget(self.title)
|
|
layout_0.addWidget(self.msg)
|
|
|
|
self.setMaximumWidth(self.parent().theme["toasts/width"])
|
|
self.msg.setMaximumWidth(self.parent().theme["toasts/width"])
|
|
self.title.setMinimumHeight(self.parent().theme["toasts/title/minimumheight"])
|
|
|
|
self.setLayout(layout_0)
|
|
|
|
self.setGeometry(0,0, self.parent().theme["toasts/width"], self.parent().theme["toasts/height"])
|
|
self.setStyleSheet(self.parent().theme["toasts/style"])
|
|
self.title.setStyleSheet(self.parent().theme["toasts/title/style"])
|
|
if self.icon:
|
|
self.icon.setStyleSheet(self.parent().theme["toasts/icon/style"])
|
|
self.msg.setStyleSheet(self.parent().theme["toasts/content/style"])
|
|
self.layout().setSpacing(0)
|
|
|
|
self.msg.setText(PesterToast.wrapText(self.msg.font(), str(self.msg.text()), self.parent().theme["toasts/width"], self.parent().theme["toasts/content/style"]))
|
|
|
|
p = QtWidgets.QApplication.desktop().availableGeometry(self).bottomRight()
|
|
o = QtWidgets.QApplication.desktop().screenGeometry(self).bottomRight()
|
|
anim.setStartValue(p.y() - o.y())
|
|
anim.setEndValue(100)
|
|
anim.valueChanged[QtCore.QVariant].connect(self.updateBottomLeftAnimation)
|
|
|
|
self.byebye = False
|
|
|
|
@QtCore.pyqtSlot()
|
|
def show(self):
|
|
self.m_animation.start()
|
|
|
|
@QtCore.pyqtSlot()
|
|
def done(self):
|
|
QtWidgets.QWidget.hide(self)
|
|
t = self.machine.toasts[0]
|
|
if t.title == str(self.title.text()) and \
|
|
t.msg == str(self.content):
|
|
self.machine.toasts.pop(0)
|
|
self.machine.displaying = False
|
|
if self.machine.on:
|
|
self.machine.showNext()
|
|
del self
|
|
|
|
@QtCore.pyqtSlot()
|
|
def reverseTrigger(self):
|
|
if self.time >= 0:
|
|
QtCore.QTimer.singleShot(self.time, self.reverseStart)
|
|
|
|
@QtCore.pyqtSlot()
|
|
def reverseStart(self):
|
|
if not self.byebye:
|
|
self.byebye = True
|
|
anim = self.m_animation.animationAt(0)
|
|
self.m_animation.setDirection(QtCore.QAnimationGroup.Backward)
|
|
anim.setEasingCurve(QtCore.QEasingCurve.InCubic)
|
|
anim.finished.disconnect(self.reverseTrigger)
|
|
anim.finished.connect(self.done)
|
|
self.m_animation.start()
|
|
|
|
@QtCore.pyqtSlot(QtCore.QVariant)
|
|
def updateBottomLeftAnimation(self, value):
|
|
p = QtWidgets.QApplication.desktop().availableGeometry(self).bottomRight()
|
|
val = float(self.height())/100
|
|
self.move(p.x()-self.width(), p.y() - (value * val) +1)
|
|
self.layout().setSpacing(0)
|
|
QtWidgets.QWidget.show(self)
|
|
|
|
def mousePressEvent(self, event):
|
|
if event.button() == QtCore.Qt.RightButton:
|
|
self.reverseStart()
|
|
elif event.button() == QtCore.Qt.LeftButton:
|
|
pass
|
|
|
|
@staticmethod
|
|
def wrapText(font, text, maxwidth, css=""):
|
|
ret = []
|
|
metric = QtGui.QFontMetrics(font)
|
|
if "padding" in css:
|
|
if css[css.find("padding")+7] != "-":
|
|
colon = css.find(":", css.find("padding"))
|
|
semicolon = css.find(";", css.find("padding"))
|
|
if semicolon < 0:
|
|
stuff = css[colon+1:]
|
|
else:
|
|
stuff = css[colon+1:semicolon]
|
|
stuff = stuff.replace("px", "").lstrip().rstrip()
|
|
stuff = stuff.split(" ")
|
|
if len(stuff) == 1:
|
|
maxwidth -= int(stuff[0])*2
|
|
elif len(stuff) == 2:
|
|
maxwidth -= int(stuff[1])*2
|
|
elif len(stuff) == 3:
|
|
maxwidth -= int(stuff[1])*2
|
|
elif len(stuff) == 4:
|
|
maxwidth -= int(stuff[1]) + int(stuff[3])
|
|
else:
|
|
if "padding-left" in css:
|
|
colon = css.find(":", css.find("padding-left"))
|
|
semicolon = css.find(";", css.find("padding-left"))
|
|
if semicolon < 0:
|
|
stuff = css[colon+1:]
|
|
else:
|
|
stuff = css[colon+1:semicolon]
|
|
stuff = stuff.replace("px", "").lstrip().rstrip()
|
|
if stuff.isdigit():
|
|
maxwidth -= int(stuff)
|
|
if "padding-right" in css:
|
|
colon = css.find(":", css.find("padding-right"))
|
|
semicolon = css.find(";", css.find("padding-right"))
|
|
if semicolon < 0:
|
|
stuff = css[colon+1:]
|
|
else:
|
|
stuff = css[colon+1:semicolon]
|
|
stuff = stuff.replace("px", "").lstrip().rstrip()
|
|
if stuff.isdigit():
|
|
maxwidth -= int(stuff)
|
|
|
|
if metric.width(text) < maxwidth:
|
|
return text
|
|
while metric.width(text) > maxwidth:
|
|
lastspace = text.find(" ")
|
|
curspace = lastspace
|
|
while metric.width(text, curspace) < maxwidth:
|
|
lastspace = curspace
|
|
curspace = text.find(" ", lastspace+1)
|
|
if curspace == -1:
|
|
break
|
|
if (metric.width(text[:lastspace]) > maxwidth) or \
|
|
len(text[:lastspace]) < 1:
|
|
for i in range(len(text)):
|
|
if metric.width(text[:i]) > maxwidth:
|
|
lastspace = i-1
|
|
break
|
|
ret.append(text[:lastspace])
|
|
text = text[lastspace+1:]
|
|
ret.append(text)
|
|
return "\n".join(ret)
|
|
|
|
|
|
class PesterToastMachine(ToastMachine, QtCore.QObject):
|
|
def __init__(self, parent, name, on=True, type="default",
|
|
types=({'default' : DefaultToast,
|
|
'libnotify' : pynotify.Notification}
|
|
if pynotify else
|
|
{'default' : DefaultToast}),
|
|
extras={}):
|
|
ToastMachine.__init__(self, parent, name, on, type, types, extras)
|
|
QtCore.QObject.__init__(self, parent)
|
|
|
|
def setEnabled(self, on):
|
|
oldon = self.on
|
|
ToastMachine.setEnabled(self, on)
|
|
if oldon != self.on:
|
|
self.parent.config.set('notify', self.on)
|
|
if self.on:
|
|
self.timer.start()
|
|
else:
|
|
self.timer.stop()
|
|
|
|
def setCurrentType(self, type):
|
|
oldtype = self.type
|
|
ToastMachine.setCurrentType(self, type)
|
|
if oldtype != self.type:
|
|
self.parent.config.set('notifyType', self.type)
|
|
|
|
@QtCore.pyqtSlot()
|
|
def showNext(self):
|
|
ToastMachine.showNext(self)
|
|
|
|
def run(self):
|
|
pass
|
|
#~ self.timer = QtCore.QTimer(self)
|
|
#~ self.timer.setInterval(1000)
|
|
#~ self.connect(self.timer, QtCore.SIGNAL('timeout()'),
|
|
#~ self, QtCore.SLOT('showNext()'))
|
|
#~ if self.on:
|
|
#~ self.timer.start()
|