91 lines
2.7 KiB
Python
91 lines
2.7 KiB
Python
|
import os
|
||
|
|
import logging
|
||
|
|
import win32con
|
||
|
|
import win32gui
|
||
|
|
|
||
|
|
from time import sleep
|
||
|
|
|
||
|
|
|
||
|
|
class ToastNotifier:
|
||
|
|
def show_toast(self, title, msg, icon_path=None, duration=5):
|
||
|
|
message_map = {win32con.WM_DESTROY: self.on_destroy}
|
||
|
|
# Register the Window class.
|
||
|
|
wc = win32gui.WNDCLASS()
|
||
|
|
wc.lpszClassName = "PythonTaskbar"
|
||
|
|
wc.lpfnWndProc = message_map # could also specify a wndproc.
|
||
|
|
|
||
|
|
classAtom = win32gui.RegisterClass(wc)
|
||
|
|
self.hinst = wc.hInstance = win32gui.GetModuleHandle(None)
|
||
|
|
|
||
|
|
# Create the Window.
|
||
|
|
style = win32con.WS_OVERLAPPED | win32con.WS_SYSMENU
|
||
|
|
self.hwnd = win32gui.CreateWindow(
|
||
|
|
classAtom,
|
||
|
|
"Taskbar",
|
||
|
|
style,
|
||
|
|
0,
|
||
|
|
0,
|
||
|
|
win32con.CW_USEDEFAULT,
|
||
|
|
win32con.CW_USEDEFAULT,
|
||
|
|
0,
|
||
|
|
0,
|
||
|
|
self.hinst,
|
||
|
|
None,
|
||
|
|
)
|
||
|
|
win32gui.UpdateWindow(self.hwnd)
|
||
|
|
|
||
|
|
self.hicon = self._set_icon(icon_path)
|
||
|
|
|
||
|
|
flags = win32gui.NIF_ICON | win32gui.NIF_MESSAGE | win32gui.NIF_TIP
|
||
|
|
nid = (self.hwnd, 0, flags, win32con.WM_USER + 20, self.hicon, "Tooltip")
|
||
|
|
# Add a new notification icon
|
||
|
|
win32gui.Shell_NotifyIcon(win32gui.NIM_ADD, nid)
|
||
|
|
win32gui.Shell_NotifyIcon(
|
||
|
|
win32gui.NIM_MODIFY,
|
||
|
|
(
|
||
|
|
self.hwnd,
|
||
|
|
0,
|
||
|
|
win32gui.NIF_INFO,
|
||
|
|
win32con.WM_USER + 20,
|
||
|
|
self.hicon,
|
||
|
|
"Balloon Tooltip",
|
||
|
|
msg,
|
||
|
|
200,
|
||
|
|
title,
|
||
|
|
),
|
||
|
|
)
|
||
|
|
|
||
|
|
sleep(duration)
|
||
|
|
win32gui.DestroyWindow(self.hwnd)
|
||
|
|
win32gui.UnregisterClass(wc.lpszClassName, None)
|
||
|
|
|
||
|
|
def _set_icon(self, icon_path):
|
||
|
|
# icon
|
||
|
|
if icon_path is not None:
|
||
|
|
icon_path = os.path.realpath(icon_path)
|
||
|
|
else:
|
||
|
|
return None
|
||
|
|
icon_flags = win32con.LR_LOADFROMFILE | win32con.LR_DEFAULTSIZE
|
||
|
|
try:
|
||
|
|
hicon = win32gui.LoadImage(
|
||
|
|
self.hinst, icon_path, win32con.IMAGE_ICON, 0, 0, icon_flags
|
||
|
|
)
|
||
|
|
except Exception as e:
|
||
|
|
logging.error("Some trouble with the icon ({}): {}".format(icon_path, e))
|
||
|
|
hicon = win32gui.LoadIcon(0, win32con.IDI_APPLICATION)
|
||
|
|
return hicon
|
||
|
|
|
||
|
|
def on_destroy(self, hwnd, msg, wparam, lparam):
|
||
|
|
nid = (hwnd, 0)
|
||
|
|
win32gui.Shell_NotifyIcon(win32gui.NIM_DELETE, nid)
|
||
|
|
win32gui.PostQuitMessage(0)
|
||
|
|
|
||
|
|
|
||
|
|
_notifier = ToastNotifier()
|
||
|
|
def show_toast(title, msg, icon_path=None, duration=5):
|
||
|
|
_notifier.show_toast(title, msg, icon_path=None, duration=5)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
toaster = ToastNotifier()
|
||
|
|
toaster.show_toast("Hello world", "Test")
|