| 8 |
# * Right-Click on a .py file - locate and click on 'Hello from Python' on |
# * Right-Click on a .py file - locate and click on 'Hello from Python' on |
| 9 |
# the context menu. |
# the context menu. |
| 10 |
|
|
| 11 |
|
# see also: http://www.codeproject.com/shell/ShellExtGuide2.asp |
| 12 |
|
|
| 13 |
import pythoncom |
import pythoncom |
| 14 |
from win32com.shell import shell, shellcon |
from win32com.shell import shell, shellcon |
| 15 |
import win32gui |
import win32gui |
| 16 |
import win32con |
import win32con |
| 17 |
|
import win32api, winerror |
| 18 |
|
|
| 19 |
IContextMenu_Methods = ["QueryContextMenu", "InvokeCommand", "GetCommandString"] |
IContextMenu_Methods = ["QueryContextMenu", "InvokeCommand", "GetCommandString"] |
| 20 |
IShellExtInit_Methods = ["Initialize"] |
IShellExtInit_Methods = ["Initialize"] |
| 21 |
|
|
| 22 |
class ShellExtension: |
class ShellExtension: |
| 23 |
_reg_progid_ = "Python.ShellExtension.ContextMenu" |
_reg_progid_ = "Python.ShellExtension.ContextMenu" |
| 24 |
_reg_desc_ = "Python Sample Shell Extension (context menu)" |
_reg_name_ = "MessClient" |
| 25 |
|
_reg_desc_ = "Mess Shell Extension" |
| 26 |
_reg_clsid_ = "{CED0336C-C9EE-4a7f-8D7F-C660393C381F}" |
_reg_clsid_ = "{CED0336C-C9EE-4a7f-8D7F-C660393C381F}" |
| 27 |
_com_interfaces_ = [shell.IID_IShellExtInit, shell.IID_IContextMenu] |
_com_interfaces_ = [shell.IID_IShellExtInit, shell.IID_IContextMenu] |
| 28 |
_public_methods_ = IContextMenu_Methods + IShellExtInit_Methods |
_public_methods_ = IContextMenu_Methods + IShellExtInit_Methods |
| 32 |
self.dataobj = dataobj |
self.dataobj = dataobj |
| 33 |
|
|
| 34 |
def QueryContextMenu(self, hMenu, indexMenu, idCmdFirst, idCmdLast, uFlags): |
def QueryContextMenu(self, hMenu, indexMenu, idCmdFirst, idCmdLast, uFlags): |
| 35 |
|
|
| 36 |
|
# debug |
| 37 |
print "QCM", hMenu, indexMenu, idCmdFirst, idCmdLast, uFlags |
print "QCM", hMenu, indexMenu, idCmdFirst, idCmdLast, uFlags |
| 38 |
# Query the items clicked on |
|
| 39 |
|
# query the items clicked on |
| 40 |
format_etc = win32con.CF_HDROP, None, 1, -1, pythoncom.TYMED_HGLOBAL |
format_etc = win32con.CF_HDROP, None, 1, -1, pythoncom.TYMED_HGLOBAL |
| 41 |
sm = self.dataobj.GetData(format_etc) |
sm = self.dataobj.GetData(format_etc) |
| 42 |
num_files = shell.DragQueryFile(sm.data_handle, -1) |
num_files = shell.DragQueryFile(sm.data_handle, -1) |
| 43 |
if num_files>1: |
|
| 44 |
msg = "&Hello from Python (with %d files selected)" % num_files |
#if num_files>1: |
| 45 |
else: |
# msg = "&Hello from Python (with %d files selected)" % num_files |
| 46 |
fname = shell.DragQueryFile(sm.data_handle, 0) |
#else: |
| 47 |
msg = "&Hello from Python (with '%s' selected)" % fname |
# fname = shell.DragQueryFile(sm.data_handle, 0) |
| 48 |
|
# msg = "&Hello from Python (with '%s' selected)" % fname |
| 49 |
|
self.files_selected = [] |
| 50 |
|
for i in range(0, num_files): |
| 51 |
|
fname = shell.DragQueryFile(sm.data_handle, i) |
| 52 |
|
if fname == 0: continue |
| 53 |
|
self.files_selected.append(fname) |
| 54 |
|
|
| 55 |
|
# build context menu(s) |
| 56 |
|
msg = "A&rchiveMe" |
| 57 |
idCmd = idCmdFirst |
idCmd = idCmdFirst |
| 58 |
items = [] |
items = [] |
| 59 |
if (uFlags & 0x000F) == shellcon.CMF_NORMAL: # Check == here, since CMF_NORMAL=0 |
if (uFlags & 0x000F) == shellcon.CMF_NORMAL: # Check == here, since CMF_NORMAL=0 |
| 60 |
print "CMF_NORMAL..." |
print "CMF_NORMAL..." |
| 61 |
items.append(msg) |
items.append(msg + " - normal") |
| 62 |
elif uFlags & shellcon.CMF_VERBSONLY: |
elif uFlags & shellcon.CMF_VERBSONLY: |
| 63 |
print "CMF_VERBSONLY..." |
print "CMF_VERBSONLY..." |
| 64 |
items.append(msg + " - shortcut") |
items.append(msg + " - shortcut") |
| 65 |
elif uFlags & shellcon.CMF_EXPLORE: |
elif uFlags & shellcon.CMF_EXPLORE: |
| 66 |
print "CMF_EXPLORE..." |
print "CMF_EXPLORE..." |
| 67 |
items.append(msg + " - normal file, right-click in Explorer") |
#items.append(msg + " - normal file, right-click in Explorer") |
| 68 |
|
items.append(msg) |
| 69 |
|
items.append("Named Shortcut &1") |
| 70 |
|
items.append("Named Shortcut &2") |
| 71 |
elif uFlags & CMF_DEFAULTONLY: |
elif uFlags & CMF_DEFAULTONLY: |
| 72 |
print "CMF_DEFAULTONLY...\r\n" |
print "CMF_DEFAULTONLY...\r\n" |
| 73 |
|
items.append("hello world - 1!") |
| 74 |
else: |
else: |
| 75 |
print "** unknown flags", uFlags |
print "** unknown flags", uFlags |
| 76 |
|
items.append("hello world - 2!") |
| 77 |
|
|
| 78 |
|
# create menu entries |
| 79 |
win32gui.InsertMenu(hMenu, indexMenu, |
win32gui.InsertMenu(hMenu, indexMenu, |
| 80 |
win32con.MF_SEPARATOR|win32con.MF_BYPOSITION, |
win32con.MF_SEPARATOR|win32con.MF_BYPOSITION, |
| 81 |
0, None) |
idCmd, None) |
| 82 |
indexMenu += 1 |
indexMenu += 1 |
| 83 |
|
idCmd += 1 |
| 84 |
|
|
| 85 |
for item in items: |
for item in items: |
| 86 |
win32gui.InsertMenu(hMenu, indexMenu, |
win32gui.InsertMenu(hMenu, indexMenu, |
| 87 |
win32con.MF_STRING|win32con.MF_BYPOSITION, |
win32con.MF_STRING|win32con.MF_BYPOSITION, |
| 91 |
|
|
| 92 |
win32gui.InsertMenu(hMenu, indexMenu, |
win32gui.InsertMenu(hMenu, indexMenu, |
| 93 |
win32con.MF_SEPARATOR|win32con.MF_BYPOSITION, |
win32con.MF_SEPARATOR|win32con.MF_BYPOSITION, |
| 94 |
0, None) |
idCmd, None) |
| 95 |
indexMenu += 1 |
indexMenu += 1 |
| 96 |
|
idCmd += 1 |
| 97 |
|
|
| 98 |
return idCmd-idCmdFirst # Must return number of menu items we added. |
return idCmd-idCmdFirst # Must return number of menu items we added. |
| 99 |
|
|
| 100 |
|
def GetCommandString(self, cmdId, typ): |
| 101 |
|
#if cmdId == 0: |
| 102 |
|
# return "Hello from Python!!" |
| 103 |
|
#else: |
| 104 |
|
# return "abc" |
| 105 |
|
return self._reg_name_ + "_" + cmdId; |
| 106 |
|
|
| 107 |
def InvokeCommand(self, ci): |
def InvokeCommand(self, ci): |
| 108 |
mask, hwnd, verb, params, dir, nShow, hotkey, hicon = ci |
mask, hwnd, verb, params, dir, nShow, hotkey, hicon = ci |
|
win32gui.MessageBox(hwnd, "Hello", "Wow", win32con.MB_OK) |
|
| 109 |
|
|
| 110 |
def GetCommandString(self, cmd, typ): |
# If lpVerb really points to a string, ignore this function call and bail out. |
| 111 |
return "Hello from Python!!" |
if win32api.HIWORD( verb ) != 0: |
| 112 |
|
return winerror.E_INVALIDARG |
| 113 |
|
|
| 114 |
|
function_id = win32api.LOWORD(verb) |
| 115 |
|
# TODO: establish mapping with building the context menus in "QueryContextMenu" |
| 116 |
|
if function_id == 1: |
| 117 |
|
method = "ArchiveMe" |
| 118 |
|
elif function_id == 2: |
| 119 |
|
method = "Shortcut 1" |
| 120 |
|
elif function_id == 3: |
| 121 |
|
method = "Shortcut 2" |
| 122 |
|
else: |
| 123 |
|
method = "Unknown" |
| 124 |
|
|
| 125 |
|
# do action(s) |
| 126 |
|
win32gui.MessageBox(hwnd, "\n".join(self.files_selected), method, win32con.MB_OK) |
| 127 |
|
|
| 128 |
|
|
| 129 |
def DllRegisterServer(): |
def DllRegisterServer(): |
| 130 |
import _winreg |
import _winreg |
| 131 |
key = _winreg.CreateKey(_winreg.HKEY_CLASSES_ROOT, |
|
| 132 |
"Python.File\\shellex") |
# extended (via http://mail.python.org/pipermail/python-list/2003-December/198390.html) |
| 133 |
subkey = _winreg.CreateKey(key, "ContextMenuHandlers") |
|
| 134 |
subkey2 = _winreg.CreateKey(subkey, "PythonSample") |
# for folders |
| 135 |
_winreg.SetValueEx(subkey2, None, 0, _winreg.REG_SZ, ShellExtension._reg_clsid_) |
folder_key = _winreg.CreateKey(_winreg.HKEY_CLASSES_ROOT, "Folder\\shellex") |
| 136 |
|
folder_subkey = _winreg.CreateKey(folder_key, "ContextMenuHandlers") |
| 137 |
|
folder_subkey2 = _winreg.CreateKey(folder_subkey, ShellExtension._reg_name_) |
| 138 |
|
_winreg.SetValueEx(folder_subkey2, None, 0, _winreg.REG_SZ, ShellExtension._reg_clsid_) |
| 139 |
|
|
| 140 |
|
# for files |
| 141 |
|
file_key = _winreg.CreateKey(_winreg.HKEY_CLASSES_ROOT, "*\\shellex") |
| 142 |
|
file_subkey = _winreg.CreateKey(file_key, "ContextMenuHandlers") |
| 143 |
|
file_subkey2 = _winreg.CreateKey(file_subkey, ShellExtension._reg_name_) |
| 144 |
|
_winreg.SetValueEx(file_subkey2, None, 0, _winreg.REG_SZ, ShellExtension._reg_clsid_) |
| 145 |
|
|
| 146 |
print ShellExtension._reg_desc_, "registration complete." |
print ShellExtension._reg_desc_, "registration complete." |
| 147 |
|
|
| 148 |
|
|
| 149 |
def DllUnregisterServer(): |
def DllUnregisterServer(): |
| 150 |
import _winreg |
import _winreg |
| 151 |
|
|
| 152 |
|
# extended (via http://mail.python.org/pipermail/python-list/2003-December/198390.html) |
| 153 |
|
|
| 154 |
try: |
try: |
| 155 |
key = _winreg.DeleteKey(_winreg.HKEY_CLASSES_ROOT, |
# for folders |
| 156 |
"Python.File\\shellex\\ContextMenuHandlers\\PythonSample") |
folder_key = _winreg.DeleteKey(_winreg.HKEY_CLASSES_ROOT, |
| 157 |
|
"Folder\\shellex\\ContextMenuHandlers\\" + ShellExtension._reg_name_) |
| 158 |
|
# for files |
| 159 |
|
file_key = _winreg.DeleteKey(_winreg.HKEY_CLASSES_ROOT, |
| 160 |
|
"*\\shellex\\ContextMenuHandlers\\" + ShellExtension._reg_name_) |
| 161 |
except WindowsError, details: |
except WindowsError, details: |
| 162 |
import errno |
import errno |
| 163 |
if details.errno != errno.ENOENT: |
if details.errno != errno.ENOENT: |