核心代码

  1. import os
  3. import win32api
  4. import win32con
  5. import win32gui
  6. import win32gui_struct
  9. class SysTrayIcon(object):
  10.     """SysTrayIcon类用于显示任务栏图标"""
  11.     QUIT = 'QUIT'
  12.     SPECIAL_ACTIONS = [QUIT]
  13.     FIRST_ID = 5320
  15.     def __init__(s, icon, hover_text, menu_options, on_quit, tk_window=None, default_menu_index=None,
  16.                  window_class_name=None):
  17.         """
  18.         icon         需要显示的图标文件路径
  19.         hover_text   鼠标停留在图标上方时显示的文字
  20.         menu_options 右键菜单，格式: (('a', None, callback), ('b', None, (('b1', None, callback),)))
  21.         on_quit      传递退出函数，在执行退出时一并运行
  22.         tk_window    传递Tk窗口，s.root，用于单击图标显示窗口
  23.         default_menu_index 不显示的右键菜单序号
  24.         window_class_name  窗口类名
  25.         """
  26.         s.icon = icon
  27.         s.hover_text = hover_text
  28.         s.on_quit = on_quit
  29.         s.root = tk_window
  31.         menu_options = menu_options + (('退出', None, s.QUIT),)
  32.         s._next_action_id = s.FIRST_ID
  33.         s.menu_actions_by_id = set()
  34.         s.menu_options = s._add_ids_to_menu_options(list(menu_options))
  35.         s.menu_actions_by_id = dict(s.menu_actions_by_id)
  36.         del s._next_action_id
  38.         s.default_menu_index = (default_menu_index or 0)
  39.         s.window_class_name = window_class_name or "SysTrayIconPy"
  41.         message_map = {win32gui.RegisterWindowMessage("TaskbarCreated"): s.restart,
  42.                        win32con.WM_DESTROY: s.destroy,
  43.                        win32con.WM_COMMAND: s.command,
  44.                        win32con.WM_USER + 20: s.notify, }
  45.         # 注册窗口类。
  46.         wc = win32gui.WNDCLASS()
  47.         wc.hInstance = win32gui.GetModuleHandle(None)
  48.         wc.lpszClassName = s.window_class_name
  49.         wc.style = win32con.CS_VREDRAW | win32con.CS_HREDRAW
  50.         wc.hCursor = win32gui.LoadCursor(0, win32con.IDC_ARROW)
  51.         wc.hbrBackground = win32con.COLOR_WINDOW
  52.         wc.lpfnWndProc = message_map  # 也可以指定wndproc.
  53.         s.classAtom = win32gui.RegisterClass(wc)
  55.     def restart(s, hwnd, msg, wparam, lparam):
  56.         s.refresh()
  58.     def activation(s):
  59.         """激活任务栏图标，不用每次都重新创建新的托盘图标"""
  60.         hinst = win32gui.GetModuleHandle(None)  # 创建窗口。
  61.         style = win32con.WS_OVERLAPPED | win32con.WS_SYSMENU
  62.         s.hwnd = win32gui.CreateWindow(s.classAtom,
  63.                                        s.window_class_name,
  64.                                        style,
  65.                                        0, 0,
  66.                                        win32con.CW_USEDEFAULT,
  67.                                        win32con.CW_USEDEFAULT,
  68.                                        0, 0, hinst, None)
  69.         win32gui.UpdateWindow(s.hwnd)
  70.         s.notify_id = None
  71.         s.refresh_icon()
  73.         win32gui.PumpMessages()
  75.     def refresh(s, title='', msg='', time=500):
  76.         """刷新托盘图标
  77.            title 标题
  78.            msg   内容，为空的话就不显示提示
  79.            time  提示显示时间"""
  80.         hinst = win32gui.GetModuleHandle(None)
  81.         if os.path.isfile(s.icon):
  82.             icon_flags = win32con.LR_LOADFROMFILE | win32con.LR_DEFAULTSIZE
  83.             hicon = win32gui.LoadImage(hinst, s.icon, win32con.IMAGE_ICON,
  84.                                        0, 0, icon_flags)
  85.         else:  # 找不到图标文件 - 使用默认值
  86.             hicon = win32gui.LoadIcon(0, win32con.IDI_APPLICATION)
  88.         if s.notify_id:
  89.             message = win32gui.NIM_MODIFY
  90.         else:
  91.             message = win32gui.NIM_ADD
  93.         s.notify_id = (s.hwnd, 0,  # 句柄、托盘图标ID
  94.                        win32gui.NIF_ICON | win32gui.NIF_MESSAGE | win32gui.NIF_TIP | win32gui.NIF_INFO,
  95.                        # 托盘图标可以使用的功能的标识
  96.                        win32con.WM_USER + 20, hicon,
  97.                        s.hover_text,  # 回调消息ID、托盘图标句柄、图标字符串
  98.                        msg,
  99.                        time,
  100.                        title,  # 提示内容、提示显示时间、提示标题
  101.                        win32gui.NIIF_INFO  # 提示用到的图标
  102.                        )
  103.         win32gui.Shell_NotifyIcon(message, s.notify_id)
  105.     def refresh_icon(s, **data):
  106.         hinst = win32gui.GetModuleHandle(None)
  107.         if os.path.isfile(s.icon):  # 尝试找到自定义图标
  108.             icon_flags = win32con.LR_LOADFROMFILE | win32con.LR_DEFAULTSIZE
  109.             hicon = win32gui.LoadImage(hinst,
  110.                                        s.icon,
  111.                                        win32con.IMAGE_ICON,
  112.                                        0,
  113.                                        0,
  114.                                        icon_flags)
  115.         else:  # 找不到图标文件 - 使用默认值
  116.             hicon = win32gui.LoadIcon(0, win32con.IDI_APPLICATION)
  118.         if s.notify_id:
  119.             message = win32gui.NIM_MODIFY
  120.         else:
  121.             message = win32gui.NIM_ADD
  122.         s.notify_id = (s.hwnd,
  123.                        0,
  124.                        win32gui.NIF_ICON | win32gui.NIF_MESSAGE | win32gui.NIF_TIP,
  125.                        win32con.WM_USER + 20,
  126.                        hicon,
  127.                        s.hover_text)
  128.         win32gui.Shell_NotifyIcon(message, s.notify_id)
  130.     def show_menu(s):
  131.         """显示右键菜单"""
  132.         menu = win32gui.CreatePopupMenu()
  133.         s.create_menu(menu, s.menu_options)
  135.         pos = win32gui.GetCursorPos()
  136.         win32gui.SetForegroundWindow(s.hwnd)
  137.         win32gui.TrackPopupMenu(menu,
  138.                                 win32con.TPM_LEFTALIGN,
  139.                                 pos[0],
  140.                                 pos[1],
  141.                                 0,
  142.                                 s.hwnd,
  143.                                 None)
  144.         win32gui.PostMessage(s.hwnd, win32con.WM_NULL, 0, 0)
  146.     def show_window(self):
  147.         self.root.deiconify()  # 显示tk窗口
  149.     def _add_ids_to_menu_options(s, menu_options):
  150.         result = []
  151.         for menu_option in menu_options:
  152.             option_text, option_icon, option_action = menu_option
  153.             if callable(option_action) or option_action in s.SPECIAL_ACTIONS:
  154.                 s.menu_actions_by_id.add((s._next_action_id, option_action))
  155.                 result.append(menu_option + (s._next_action_id,))
  156.             else:
  157.                 result.append((option_text,
  158.                                option_icon,
  159.                                s._add_ids_to_menu_options(option_action),
  160.                                s._next_action_id))
  161.             s._next_action_id += 1
  162.         return result
  164.     def destroy(s, hwnd=None, msg=None, wparam=None, lparam=None, exit=1):
  165.         nid = (s.hwnd, 0)
  166.         win32gui.Shell_NotifyIcon(win32gui.NIM_DELETE, nid)
  167.         win32gui.PostQuitMessage(0)  # 退出托盘图标
  168.         if s.on_quit:
  169.             s.on_quit()  # 运行传递的on_quit
  171.     def notify(s, hwnd, msg, wparam, lparam):
  172.         '''鼠标事件'''
  173.         """
  174.         可能的鼠标事件：
  175.           WM_MOUSEMOVE      #光标经过图标
  176.           WM_LBUTTONDOWN    #左键按下
  177.           WM_LBUTTONUP      #左键弹起
  178.           WM_LBUTTONDBLCLK  #双击左键
  179.           WM_RBUTTONDOWN    #右键按下
  180.           WM_RBUTTONUP      #右键弹起
  181.           WM_RBUTTONDBLCLK  #双击右键
  182.           WM_MBUTTONDOWN    #滚轮按下
  183.           WM_MBUTTONUP      #滚轮弹起
  184.           WM_MBUTTONDBLCLK  #双击滚轮
  185.         """
  186.         if lparam == win32con.WM_LBUTTONDBLCLK:  # 双击左键
  187.             pass
  188.         elif lparam == win32con.WM_RBUTTONUP:  # 右键弹起
  189.             s.show_menu()
  190.         elif lparam == win32con.WM_LBUTTONUP:  # 左键弹起
  191.             s.show_window()
  192.         return True
  194.     def create_menu(s, menu, menu_options):
  195.         for option_text, option_icon, option_action, option_id in menu_options[::-1]:
  196.             if option_icon:
  197.                 option_icon = s.prep_menu_icon(option_icon)
  199.             if option_id in s.menu_actions_by_id:
  200.                 item, extras = win32gui_struct.PackMENUITEMINFO(text=option_text,
  201.                                                                 hbmpItem=option_icon,
  202.                                                                 wID=option_id)
  203.                 win32gui.InsertMenuItem(menu, 0, 1, item)
  204.             else:
  205.                 submenu = win32gui.CreatePopupMenu()
  206.                 s.create_menu(submenu, option_action)
  207.                 item, extras = win32gui_struct.PackMENUITEMINFO(text=option_text,
  208.                                                                 hbmpItem=option_icon,
  209.                                                                 hSubMenu=submenu)
  210.                 win32gui.InsertMenuItem(menu, 0, 1, item)
  212.     def prep_menu_icon(s, icon):
  213.         # 加载图标。
  214.         ico_x = win32api.GetSystemMetrics(win32con.SM_CXSMICON)
  215.         ico_y = win32api.GetSystemMetrics(win32con.SM_CYSMICON)
  216.         hicon = win32gui.LoadImage(0, icon, win32con.IMAGE_ICON, ico_x, ico_y, win32con.LR_LOADFROMFILE)
  218.         hdcBitmap = win32gui.CreateCompatibleDC(0)
  219.         hdcScreen = win32gui.GetDC(0)
  220.         hbm = win32gui.CreateCompatibleBitmap(hdcScreen, ico_x, ico_y)
  221.         hbmOld = win32gui.SelectObject(hdcBitmap, hbm)
  222.         brush = win32gui.GetSysColorBrush(win32con.COLOR_MENU)
  223.         win32gui.FillRect(hdcBitmap, (0, 0, 16, 16), brush)
  224.         win32gui.DrawIconEx(hdcBitmap, 0, 0, hicon, ico_x, ico_y, 0, 0, win32con.DI_NORMAL)
  225.         win32gui.SelectObject(hdcBitmap, hbmOld)
  226.         win32gui.DeleteDC(hdcBitmap)
  228.         return hbm
  230.     def command(s, hwnd, msg, wparam, lparam):
  231.         id = win32gui.LOWORD(wparam)
  232.         s.execute_menu_option(id)
  234.     def execute_menu_option(s, id):
  235.         menu_action = s.menu_actions_by_id[id]
  236.         if menu_action == s.QUIT:
  237.             print("ss")
  238.             win32gui.DestroyWindow(s.hwnd)
  239.         else:
  240.             menu_action(s)

使用方法

  1. # 重现关闭按钮事件
  2. self.root.protocol("WM_DELETE_WINDOW", self.close_)
  4. # 开启常驻后台线程
  5. self.backend_thread = threading.Thread(target=self.backend)
  6. self.backend_thread.setDaemon(True)
  7. self.backend_thread.start()
  10. def backend(self, hover_text=None):
  11.     """
  12.         后台图标线程
  13.     """
  14.     if hover_text is None:
  15.         hover_text = self.root.title()
  16.         menu_options = (('主页', None, self.show),)
  17.         self.SysTrayIcon = SysTrayIcon(
  18.             icon=icon_ico,  # 图标
  19.             hover_text=hover_text,  # 光标停留显示文字
  20.             menu_options=menu_options,  # 右键菜单
  21.             on_quit=self.exit,  # 退出调用
  22.             tk_window=self.root,  # Tk窗口
  23.         )
  24.         self.SysTrayIcon.activation()
  26. def show(self):
  27.     self.root.deiconify()
  28. def exit(self):
  29.     self.root.destroy()
  30. def close_(self):
  31.     self.root.withdraw()