Python 注册菜单

我们看到下面的示例图是 在右键文件夹上查看菜单，带有两级的菜单选项。

文件夹

空白处

需要使用管理员权限运行
因为右键菜单需要注册表信息，打开注册表需要管理员权限UAC，所以在执行命令过程中需要使用管理员权限来执行命令

import sys
import winreg as reg
import os


def add_context_menu_parent(parent_name, reg_root_key_path, reg_key_path):
    # 打开名称父键
    key = reg.OpenKey(reg_root_key_path, reg_key_path)
    # 为key创建一个名称为menu_name的sub_key，并设置sub_key的值为menu_name加上快捷键，数据类型为REG_SZ字符串类型
    reg.SetValue(key, parent_name, reg.REG_SZ, '')

    # 打开刚刚创建的名为menu_name的sub_key
    sub_key = reg.OpenKey(key, parent_name, 0, reg.KEY_ALL_ACCESS)

    # 为sub_key设置成可添加二级菜单
    # reg.SetValueEx(sub_key, 'MUIVerb', 0, reg.REG_SZ,parent_name + '(&{0})'.format(shortcut_key))
    # reg.SetValueEx(sub_key, 'icon', 0, reg.REG_SZ, icon_path)
    # reg.SetValueEx(sub_key, 'MultipleInvokePromptMinimum', 0, reg.REG_DWORD, 1)
    reg.SetValueEx(sub_key, 'SubCommands', 0, reg.REG_SZ, '')
    reg.SetValue(sub_key, 'shell', reg.REG_SZ, '')

    # 关闭sub_key和key
    reg.CloseKey(sub_key)
    reg.CloseKey(key)


def add_context_menu(menu_name, command, reg_root_key_path, reg_key_path, shortcut_key):
    # 打开名称父键
    key = reg.OpenKey(reg_root_key_path, reg_key_path)

    # 为key创建一个名称为menu_name的sub_key，并设置sub_key的值为menu_name加上快捷键，数据类型为REG_SZ字符串类型
    reg.SetValue(key, menu_name, reg.REG_SZ, f'{menu_name}(&{shortcut_key})')

    # 打开刚刚创建的名为menu_name的sub_key
    sub_key = reg.OpenKey(key, menu_name, 0, reg.KEY_ALL_ACCESS)
    # reg.SetValueEx(sub_key, 'icon', 0, reg.REG_SZ, icon_path)

    # 为sub_key添加名为'command'的子键，并设置其值为command + ' "%v"'，数据类型为REG_SZ字符串类型
    reg.SetValue(sub_key, 'command', reg.REG_SZ,  command)

    # 关闭sub_key和key
    reg.CloseKey(sub_key)
    reg.CloseKey(key)


if __name__ == "__main__":
 
    parent_name = '服务'
    add_context_menu_parent(parent_name, reg.HKEY_CLASSES_ROOT, r'Directory\shell')
    add_context_menu_parent(parent_name, reg.HKEY_CLASSES_ROOT, r'Directory\Background\shell')

    menu_name0 = '添加'
    # 执行一个python脚本的命令，用于打印命令行参数的第二个参数（即选中的文件路径）
    py_command0 = rf'addwm'
    # 添加文件右键菜单
    add_context_menu(menu_name0, py_command0, reg.HKEY_CLASSES_ROOT, rf'Directory\shell\{parent_name}\shell', 'A')
    add_context_menu(menu_name0, py_command0, reg.HKEY_CLASSES_ROOT, rf'Directory\Background\shell\{parent_name}\shell', 'A')

    menu_name0 = '删除'
    # 执行一个python脚本的命令，用于打印命令行参数的第二个参数（即选中的文件路径）
    py_command0 = rf'removewm'
    # 添加文件右键菜单
    add_context_menu(menu_name0, py_command0, reg.HKEY_CLASSES_ROOT, rf'Directory\shell\{parent_name}\shell', 'R')
    add_context_menu(menu_name0, py_command0, reg.HKEY_CLASSES_ROOT, rf'Directory\Background\shell\{parent_name}\shell', 'R')

    os.system('pause')