Search code examples
pythonpywin32

Error when trying to reuse Windows Notification class in Python


I am trying to use the pywin32 module to create a somewhat more interactive experience when playing my game, but I have encountered some issues. I wish to use the class WindowsBalloonTip more than once, but encounter an issue when attempting to use it again.

class WindowsBalloonTip:
    def __init__(self, title, msg):
        message_map = {
                win32con.WM_DESTROY: self.OnDestroy,
        }
        # Register the Window class.
        wc = WNDCLASS()
        hinst = wc.hInstance = GetModuleHandle(None)
        wc.lpszClassName = "PythonTaskbar"
        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", style, \
                0, 0, win32con.CW_USEDEFAULT, win32con.CW_USEDEFAULT, \
                0, 0, hinst, None)
        UpdateWindow(self.hwnd)
        iconPathName = os.path.abspath(os.path.join( sys.path[0], "balloontip.ico" ))
        icon_flags = win32con.LR_LOADFROMFILE | win32con.LR_DEFAULTSIZE
        try:
           hicon = LoadImage(hinst, iconPathName, \
                    win32con.IMAGE_ICON, 0, 0, icon_flags)
        except:
          hicon = LoadIcon(0, win32con.IDI_APPLICATION)
        flags = NIF_ICON | NIF_MESSAGE | NIF_TIP
        nid = (self.hwnd, 0, flags, win32con.WM_USER+20, hicon, "tooltip")
        Shell_NotifyIcon(NIM_ADD, nid)
        Shell_NotifyIcon(NIM_MODIFY, \
                         (self.hwnd, 0, NIF_INFO, win32con.WM_USER+20,\
                          hicon, "Balloon  tooltip",msg,200,title))
        # self.show_balloon(title, msg)
        DestroyWindow(self.hwnd)
    def OnDestroy(self, hwnd, msg, wparam, lparam):
        nid = (self.hwnd, 0)
        Shell_NotifyIcon(NIM_DELETE, nid)
        PostQuitMessage(0) # Terminate the app.

def balloon_tip(title, msg):
    w=WindowsBalloonTip(title, msg)

To call on WindowsBalloonTip, I use:

balloon_tip("Fantasy Text", "Welcome to Fantasy Text!")

But later on when I attempt to use balloon_tip again, I encounter this error:

Traceback (most recent call last):
  File "C:\Users\austin\Documents\Programming\Fantasy Text\v1.0.1.0\Fantasy Text.py", line 456, in <module>
    Start()
  File "C:\Users\austin\Documents\Programming\Fantasy Text\v1.0.1.0\Fantasy Text.py", line 89, in Start
    Zombie_Mode()
  File "C:\Users\austin\Documents\Programming\Fantasy Text\v1.0.1.0\Fantasy Text.py", line 224, in Zombie_Mode
    Zombie_Attack()
  File "C:\Users\austin\Documents\Programming\Fantasy Text\v1.0.1.0\Fantasy Text.py", line 240, in Zombie_Attack
    balloon_tip("FT: Zombie", "Zombie did %s" % zombie_damage + ' damage!')
  File "C:\Users\austin\Documents\Programming\Fantasy Text\v1.0.1.0\Fantasy Text.py", line 60, in balloon_tip
    w=WindowsBalloonTip(title, msg)
  File "C:\Users\austin\Documents\Programming\Fantasy Text\v1.0.1.0\Fantasy Text.py", line 32, in __init__
    classAtom = RegisterClass(wc)
pywintypes.error: (1410, 'RegisterClass', 'Class already exists.')

I realize that the class already exists from the error given, but how can I resolve this?


Solution

  • Just restructure your code a bit. Create one instance of WindowsBalloonTip and use it to show all your tips:

    class WindowsBalloonTip:
        def __init__(self):
            message_map = {
                    win32con.WM_DESTROY: self.OnDestroy,
            }
            # Register the Window class.
            wc = WNDCLASS()
            self.hinst = wc.hInstance = GetModuleHandle(None)
            wc.lpszClassName = "PythonTaskbar"
            wc.lpfnWndProc = message_map # could also specify a wndproc.
            self.classAtom = RegisterClass(wc)
    
        def ShowWindow(self,title, msg):
            # Create the Window.
            style = win32con.WS_OVERLAPPED | win32con.WS_SYSMENU
            self.hwnd = CreateWindow( self.classAtom, "Taskbar", style, \
                    0, 0, win32con.CW_USEDEFAULT, win32con.CW_USEDEFAULT, \
                    0, 0, self.hinst, None)
            UpdateWindow(self.hwnd)
            iconPathName = os.path.abspath(os.path.join( sys.path[0], "balloontip.ico" ))
            icon_flags = win32con.LR_LOADFROMFILE | win32con.LR_DEFAULTSIZE
            try:
               hicon = LoadImage(self.hinst, iconPathName, \
                        win32con.IMAGE_ICON, 0, 0, icon_flags)
            except:
              hicon = LoadIcon(0, win32con.IDI_APPLICATION)
            flags = NIF_ICON | NIF_MESSAGE | NIF_TIP
            nid = (self.hwnd, 0, flags, win32con.WM_USER+20, hicon, "tooltip")
            Shell_NotifyIcon(NIM_ADD, nid)
            Shell_NotifyIcon(NIM_MODIFY, \
                             (self.hwnd, 0, NIF_INFO, win32con.WM_USER+20,\
                              hicon, "Balloon  tooltip",msg,200,title))
            # self.show_balloon(title, msg)
            DestroyWindow(self.hwnd)
    
        def OnDestroy(self, hwnd, msg, wparam, lparam):
            nid = (self.hwnd, 0)
            Shell_NotifyIcon(NIM_DELETE, nid)
            PostQuitMessage(0) # Terminate the app.
    
    w = WindowsBalloonTip()
    w.ShowWindow("Title1","Message One!")
    time.sleep(1) # some time later
    w.ShowWindow("Title2","Message Two!")