AppKit Python 创建系统托盘

引言

在许多操作系统中，系统托盘（System Tray）是一个位于任务栏上的小图标区域，通常用来显示常驻后台的应用程序或者提供快速访问功能。在 macOS 系统中，我们可以使用 AppKit 框架和 Python 语言编写代码，来创建自己的系统托盘应用。

本文将详细介绍如何使用 AppKit 和 Python 来创建一个简单的系统托盘应用。我们将一步步地讲解如何创建托盘图标、添加菜单选项以及响应菜单点击事件。

步骤1：导入必要的模块

首先，我们需要导入需要用到的模块。使用 pip 命令安装 pyobjc 库，然后在 Python 代码中导入 AppKit 模块。

import os
import AppKit

步骤2：创建托盘图标

接下来，我们将创建一个函数 create_tray_icon() 来添加一个托盘图标。在函数中，我们首先获取应用程序图标的路径，然后使用 NSImage 类来创建一个图像对象。最后，通过 NSStatusBar 的 statusItemWithLength 方法，我们创建了一个具有指定长度的状态栏项目，并将图像对象设置为托盘图标。

def create_tray_icon():
    icon_path = os.path.join(os.path.dirname(__file__), 'icon.png')  # 替换为你的图标路径
    image = AppKit.NSImage.alloc().initByReferencingFile_(icon_path)
    image.setSize_((20, 20))  # 设置图标尺寸

    status_item = AppKit.NSStatusBar.systemStatusBar().statusItemWithLength_(AppKit.NSVariableStatusItemLength)
    status_item.setHighlightMode_(True)
    status_item.setImage_(image)

步骤3：添加菜单选项

现在，我们将扩展 create_tray_icon() 函数，以添加菜单选项。我们使用 NSMenu 和 NSMenuItem 类来创建菜单对象和菜单项对象。通过使用 NSMenu 的 addItem_() 方法，我们可以将菜单项添加到菜单中。最后，我们将 NSMenu 对象设置为托盘图标的菜单属性。

def create_tray_icon():
    # ... 上一步的代码 ...

    menu = AppKit.NSMenu.alloc().init()
    menu_item_1 = AppKit.NSMenuItem.alloc().initWithTitle_action_keyEquivalent_('菜单项1', None, '')
    menu_item_2 = AppKit.NSMenuItem.alloc().initWithTitle_action_keyEquivalent_('菜单项2', None, '')

    menu.addItem_(menu_item_1)
    menu.addItem_(menu_item_2)

    status_item.setMenu_(menu)

步骤4：响应菜单点击事件

接下来，我们将为菜单项添加点击事件的处理函数。我们创建了一个名为 handle_menu_item_1() 的函数，并将其绑定到菜单项 menu_item_1 的 setTarget() 和 setAction() 方法上。这样，当用户点击菜单项1时，函数 handle_menu_item_1() 将被调用。

def handle_menu_item_1(sender):
    AppKit.NSApplication.sharedApplication().terminate_(sender)

def create_tray_icon():
    # ... 上一步的代码 ...

    menu_item_1.setTarget_(AppKit.NSApplication.sharedApplication())
    menu_item_1.setAction_('handle_menu_item_1:')

    # ... 下面的代码 ...

步骤5：运行应用程序主循环

最后，我们需要将应用程序的主循环运行起来，以便监听事件并响应用户操作。在 macOS 中，我们可以使用 NSRunLoop 的 currentRunLoop() 方法来获取当前的运行循环，并使用 run() 方法开始运行。

def main():
    create_tray_icon()
    AppKit.NSRunLoop.currentRunLoop().run()

if __name__ == '__main__':
    main()

完整示例代码

下面是完整的示例代码：

import os
import AppKit

def handle_menu_item_1(sender):
    AppKit.NSApplication.sharedApplication().terminate_(sender)

def create_tray_icon():
    icon_path = os.path.join(os.path.dirname(__file__), 'icon.png')
    image = AppKit.NSImage.alloc().initByReferencingFile_(icon_path)
    image.setSize_((20, 20))

    status_item = AppKit.NSStatusBar.systemStatusBar().statusItemWithLength_(AppKit.NSVariableStatusItemLength)
    status_item.setHighlightMode_(True)
    status_item.setImage_(image)

    menu = AppKit.NSMenu.alloc().init()
    menu_item_1 = AppKit.NSMenuItem.alloc().initWithTitle_action_keyEquivalent_('菜单项1', None, '')
    menu_item_2 = AppKit.NSMenuItem.alloc().initWithTitle_action_keyEquivalent_('菜单项2', None, '')

    menu.addItem_(menu_item_1)
    menu.addItem_(menu_item_2)

    menu_item_1.setTarget_(AppKit.NSApplication.sharedApplication())
    menu_item_1.setAction_('handle_menu_item_1:')

    status_item.setMenu_(menu)

def main():
    create_tray_icon()
    AppKit.NSRunLoop.currentRunLoop().run()

if __name__ == '__main__':
    main()