# Demonstrates some advanced menu concepts using win32gui.
# This creates a taskbar icon which has some fancy menus (but note that
# selecting the menu items does nothing useful - see win32gui_taskbar.py
# for examples of this.
# NOTE: This is a work in progress. Todo:
# * The "Checked" menu items don't work correctly - I'm not sure why.
# * No support for GetMenuItemInfo.
# Based on Andy McKay's demo code.
from win32api import *
# Try and use XP features, so we get alpha-blending etc.
try:
from winxpgui import *
except ImportError:
from win32gui import *
from win32gui_struct import *
import win32con
import sys, os
import struct
import array
this_dir = os.path.split(sys.argv[0])[0]
class MainWindow:
def __init__(self):
message_map = {
win32con.WM_DESTROY: self.OnDestroy,
win32con.WM_COMMAND: self.OnCommand,
win32con.WM_USER+20 : self.OnTaskbarNotify,
# owner-draw related handlers.
win32con.WM_MEASUREITEM: self.OnMeasureItem,
win32con.WM_DRAWITEM: self.OnDrawItem,
}
# Register the Window class.
wc = WNDCLASS()
hinst = wc.hInstance = GetModuleHandle(None)
wc.lpszClassName = "PythonTaskbarDemo"
wc.lpfnWndProc = message_map # could also specify a wndproc.
classAtom = RegisterClass(wc)
# Create the Window.
style = win32con.WS_OVERLAPPED | win32con.WS_SYSMENU
self.hwnd = CreateWindow( classAtom, "Taskbar Demo", style, \
0, 0, win32con.CW_USEDEFAULT, win32con.CW_USEDEFAULT, \
0, 0, hinst, None)
UpdateWindow(self.hwnd)
iconPathName = os.path.abspath(os.path.join( sys.prefix, "pyc.ico" ))
# py2.5 includes the .ico files in the DLLs dir for some reason.
if not os.path.isfile(iconPathName):
iconPathName = os.path.abspath(os.path.join( os.path.split(sys.executable)[0], "DLLs", "pyc.ico" ))
if not os.path.isfile(iconPathName):
# Look in the source tree.
iconPathName = os.path.abspath(os.path.join( os.path.split(sys.executable)[0], "..\\PC\\pyc.ico" ))
if os.path.isfile(iconPathName):
icon_flags = win32con.LR_LOADFROMFILE | win32con.LR_DEFAULTSIZE
hicon = LoadImage(hinst, iconPathName, win32con.IMAGE_ICON, 0, 0, icon_flags)
else:
iconPathName = None
print "Can't find a Python icon file - using default"
hicon = LoadIcon(0, win32con.IDI_APPLICATION)
self.iconPathName = iconPathName
# Load up some information about menus needed by our owner-draw code.
# The font to use on the menu.
ncm = SystemParametersInfo(win32con.SPI_GETNONCLIENTMETRICS)
self.font_menu = CreateFontIndirect(ncm['lfMenuFont'])
# spacing for our ownerdraw menus - not sure exactly what constants
# should be used (and if you owner-draw all items on the menu, it
# doesn't matter!)
self.menu_icon_height = GetSystemMetrics(win32con.SM_CYMENU) - 4
self.menu_icon_width = self.menu_icon_height
self.icon_x_pad = 8 # space from end of icon to start of text.
# A map we use to stash away data we need for ownerdraw. Keyed
# by integer ID - that ID will be set in dwTypeData of the menu item.
self.menu_item_map = {}
# Finally, create the menu
self.createMenu()
flags = NIF_ICON | NIF_MESSAGE | NIF_TIP
nid = (self.hwnd, 0, flags, win32con.WM_USER+20, hicon, "Python Demo")
Shell_NotifyIcon(NIM_ADD, nid)
print "Please right-click on the Python icon in the taskbar"
def createMenu(self):
self.hmenu = menu = CreatePopupMenu()
# Create our 'Exit' item with the standard, ugly 'close' icon.
item, extras = PackMENUITEMINFO(text = "Exit",
hbmpItem=win32con.HBMMENU_MBAR_CLOSE,
wID=1000)
InsertMenuItem(menu, 0, 1, item)
# Create a 'text only' menu via InsertMenuItem rather then
# AppendMenu, just to prove we can!
item, extras = PackMENUITEMINFO(text = "Text only item",
wID=1001)
InsertMenuItem(menu, 0, 1, item)
load_bmp_flags=win32con.LR_LOADFROMFILE | \
win32con.LR_LOADTRANSPARENT
# These images are "over sized", so we load them scaled.
hbmp = LoadImage(0, os.path.join(this_dir, "images/smiley.bmp"),
win32con.IMAGE_BITMAP, 20, 20, load_bmp_flags)
# Create a top-level menu with a bitmap
item, extras = PackMENUITEMINFO(text="Menu with bitmap",
hbmpItem=hbmp,
wID=1002)
InsertMenuItem(menu, 0, 1, item)
# Owner-draw menus mainly from:
# http://windowssdk.msdn.microsoft.com/en-us/library/ms647558.aspx
# and:
# http://www.codeguru.com/cpp/controls/menu/bitmappedmenus/article.php/c165
# Create one with an icon - this is *lots* more work - we do it
# owner-draw! The primary reason is to handle transparency better -
# converting to a bitmap causes the background to be incorrect when
# the menu item is selected. I can't see a simpler way.
# First, load the icon we want to use.
ico_x = GetSystemMetrics(win32con.SM_CXSMICON)
ico_y = GetSystemMetrics(win32con.SM_CYSMICON)
if self.iconPathName:
hicon = LoadImage(0, self.iconPathName, win32con.IMAGE_ICON, ico_x, ico_y, win32con.LR_LOADFROMFILE)
else:
shell_dll = os.path.join(GetSystemDirectory(), "shell32.dll")
large, small = win32gui.ExtractIconEx(shell_dll, 4, 1)
hicon = small[0]
DestroyIcon(large[0])
# Stash away the text and hicon in our map, and add the owner-draw
# item to the menu.
index = 0
self.menu_item_map[index] = (hicon, "Menu with owner-draw icon")
item, extras = PackMENUITEMINFO(fType=win32con.MFT_OWNERDRAW,
dwItemData=index,
wID=1009)
InsertMenuItem(menu, 0, 1, item)
# Add another icon-based icon - but this time using HBMMENU_CALLBACK
# in the hbmpItem elt, so we only need to draw the icon (ie, not the
# text or checkmark)
index = 1
self.menu_item_map[index] = (hicon, None)
item, extras = PackMENUITEMINFO(text="Menu with o-d icon 2",
dwItemData=index,
hbmpItem=win32con.HBMMENU_CALLBACK,
wID=1010)
InsertMenuItem(menu, 0, 1, item)
# Add another icon-based icon - this time by converting
# via bitmap. Note the icon background when selected is ugly :(
hdcBitmap = CreateCompatibleDC(0)
hdcScreen = GetDC(0)
hbm = CreateCompatibleBitmap(hdcScreen, ico_x, ico_y)
hbmOld = SelectObject(hdcBitmap, hbm)
SetBkMode(hdcBitmap, win32con.TRANSPARENT)
# Fill the background.
brush = GetSysColorBrush(win32con.COLOR_MENU)
FillRect(hdcBitmap, (0, 0, 16, 16), brush)
# unclear if brush needs to be freed. Best clue I can find is:
# "GetSysColorBrush returns a cached brush instead of allocating a new
# one." - implies no DeleteObject.
# draw the icon
DrawIconEx(hdcBitmap, 0, 0, hicon, ico_x, ico_y, 0, 0, win32con.DI_NORMAL)
SelectObject(hdcBitmap, hbmOld)
DeleteDC(hdcBitmap)
item, extras = PackMENUITEMINFO(text="Menu with icon",
hbmpItem=hbm.Detach(),
wID=1011)
InsertMenuItem(menu, 0, 1, item)
# Create a sub-menu, and put a few funky ones there.
self.sub_menu = sub_menu = CreatePopupMenu()
# A 'checkbox' menu.
item, extras = PackMENUITEMINFO(fState=win32con.MFS_CHECKED,
text="Checkbox menu",
hbmpItem=hbmp,
wID=1003)
InsertMenuItem(sub_menu, 0, 1, item)
# A 'radio' menu.
InsertMenu(sub_menu, 0, win32con.MF_BYPOSITION, win32con.MF_SEPARATOR, None)
item, extras = PackMENUITEMINFO(fType=win32con.MFT_RADIOCHECK,
fState=win32con.MFS_CHECKED,
text="Checkbox menu - bullet 1",
hbmpItem=hbmp,
wID=1004)
InsertMenuItem(sub_menu, 0, 1, item)
item, extras = PackMENUITEMINFO(fType=win32con.MFT_RADIOCHECK,
fState=win32con.MFS_UNCHECKED,
text="Checkbox menu - bullet 2",
hbmpItem=hbmp,
wID=1005)
InsertMenuItem(sub_menu, 0, 1, item)
# And add the sub-menu to the top-level menu.
item, extras = PackMENUITEMINFO(text="Sub-Menu",
hSubMenu=sub_menu)
InsertMenuItem(menu, 0, 1, item)
# Set 'Exit' as the default option.
SetMenuDefaultItem(menu, 1000, 0)
def OnDestroy(self, hwnd, msg, wparam, lparam):
nid = (self.hwnd, 0)
Shell_NotifyIcon(NIM_DELETE, nid)
PostQuitMessage(0) # Terminate the app.
def OnTaskbarNotify(self, hwnd, msg, wparam, lparam):
if lparam==win32con.WM_RBUTTONUP:
print "You right clicked me."
# display the menu at the cursor pos.
pos = GetCursorPos()
SetForegroundWindow(self.hwnd)
TrackPopupMenu(self.hmenu, win32con.TPM_LEFTALIGN, pos[0], pos[1], 0, self.hwnd, None)
PostMessage(self.hwnd, win32con.WM_NULL, 0, 0)
elif lparam==win32con.WM_LBUTTONDBLCLK:
print "You double-clicked me"
# find the default menu item and fire it.
cmd = GetMenuDefaultItem(self.hmenu, False, 0)
if cmd == -1:
print "Can't find a default!"
# and just pretend it came from the menu
self.OnCommand(hwnd, win32con.WM_COMMAND, cmd, 0)
return 1
def OnCommand(self, hwnd, msg, wparam, lparam):
id = LOWORD(wparam)
if id == 1000:
print "Goodbye"
DestroyWindow(self.hwnd)
elif id in (1003, 1004, 1005):
# Our 'checkbox' and 'radio' items
state = GetMenuState(self.sub_menu, id, win32con.MF_BYCOMMAND)
if state==-1:
raise RuntimeError("No item found")
if state & win32con.MF_CHECKED:
check_flags = win32con.MF_UNCHECKED
print "Menu was checked - unchecking"
else:
check_flags = win32con.MF_CHECKED
print "Menu was unchecked - checking"
if id == 1003:
# simple checkbox
rc = CheckMenuItem(self.sub_menu, id,
win32con.MF_BYCOMMAND | check_flags)
else:
# radio button - must pass the first and last IDs in the
# "group", and the ID in the group that is to be selected.
rc = CheckMenuRadioItem(self.sub_menu, 1004, 1005, id,
win32con.MF_BYCOMMAND)
# Get and check the new state - first the simple way...
new_state = GetMenuState(self.sub_menu, id, win32con.MF_BYCOMMAND)
if new_state & win32con.MF_CHECKED != check_flags:
raise RuntimeError("The new item didn't get the new checked state!")
# Now the long-winded way via GetMenuItemInfo...
buf, extras = EmptyMENUITEMINFO()
win32gui.GetMenuItemInfo(self.sub_menu, id, False, buf)
fType, fState, wID, hSubMenu, hbmpChecked, hbmpUnchecked, \
dwItemData, text, hbmpItem = UnpackMENUITEMINFO(buf)
if fState & win32con.MF_CHECKED != check_flags:
raise RuntimeError("The new item didn't get the new checked state!")
else:
print "OnCommand for ID", id
# Owner-draw related functions. We only have 1 owner-draw item, but
# we pretend we have more than that :)
def OnMeasureItem(self, hwnd, msg, wparam, lparam):
## Last item of MEASUREITEMSTRUCT is a ULONG_PTR
fmt = "5iP"
buf = PyMakeBuffer(struct.calcsize(fmt), lparam)
data = struct.unpack(fmt, buf)
ctlType, ctlID, itemID, itemWidth, itemHeight, itemData = data
hicon, text = self.menu_item_map[itemData]
if text is None:
# Only drawing icon due to HBMMENU_CALLBACK
cx = self.menu_icon_width
cy = self.menu_icon_height
else:
# drawing the lot!
dc = GetDC(hwnd)
oldFont = SelectObject(dc, self.font_menu)
cx, cy = GetTextExtentPoint32(dc, text)
SelectObject(dc, oldFont)
ReleaseDC(hwnd, dc)
cx += GetSystemMetrics(win32con.SM_CXMENUCHECK)
cx += self.menu_icon_width + self.icon_x_pad
cy = GetSystemMetrics(win32con.SM_CYMENU)
new_data = struct.pack(fmt, ctlType, ctlID, itemID, cx, cy, itemData)
PySetMemory(lparam, new_data)
return True
def OnDrawItem(self, hwnd, msg, wparam, lparam):
## lparam is a DRAWITEMSTRUCT
fmt = "5i2P4iP"
data = struct.unpack(fmt, PyGetMemory(lparam, struct.calcsize(fmt)))
ctlType, ctlID, itemID, itemAction, itemState, hwndItem, \
hDC, left, top, right, bot, itemData = data
rect = left, top, right, bot
hicon, text = self.menu_item_map[itemData]
if text is None:
# This means the menu-item had HBMMENU_CALLBACK - so all we
# draw is the icon. rect is the entire area we should use.
DrawIconEx(hDC, left, top, hicon, right-left, bot-top,
0, 0, win32con.DI_NORMAL)
else:
# If the user has selected the item, use the selected
# text and background colors to display the item.
selected = itemState & win32con.ODS_SELECTED
if selected:
crText = SetTextColor(hDC, GetSysColor(win32con.COLOR_HIGHLIGHTTEXT))
crBkgnd = SetBkColor(hDC, GetSysColor(win32con.COLOR_HIGHLIGHT))
each_pad = self.icon_x_pad // 2
x_icon = left + GetSystemMetrics(win32con.SM_CXMENUCHECK) + each_pad
x_text = x_icon + self.menu_icon_width + each_pad
# Draw text first, specifying a complete rect to fill - this sets
# up the background (but overwrites anything else already there!)
# Select the font, draw it, and restore the previous font.
hfontOld = SelectObject(hDC, self.font_menu)
ExtTextOut(hDC, x_text, top+2, win32con.ETO_OPAQUE, rect, text)
SelectObject(hDC, hfontOld)
# Icon image next. Icons are transparent - no need to handle
# selection specially.
DrawIconEx(hDC, x_icon, top+2, hicon,
self.menu_icon_width, self.menu_icon_height,
0, 0, win32con.DI_NORMAL)
# Return the text and background colors to their
# normal state (not selected).
if selected:
SetTextColor(hDC, crText)
SetBkColor(hDC, crBkgnd)
def main():
w=MainWindow()
PumpMessages()
if __name__=='__main__':
main()
|