from Xlib import X, display, Xatom

def get_window_under_cursor():
    """Get the window under the cursor and its WM_CLASS"""
    d = display.Display()
    root = d.screen().root

    # Query pointer to get window under cursor
    pointer = root.query_pointer()
    window = pointer.child

    # The pointer.child might be a parent window, we need to find the actual client window
    # by traversing up to find the window that has WM_CLASS set
    while window:
        try:
            print(f"Window: {window}")
            wm_class = window.get_wm_class()
            if wm_class:
                # wm_class returns a tuple: (instance, class)
                print(f"WM_CLASS: {wm_class}")
                print(f"  Instance: {wm_class[0]}")
                print(f"  Class: {wm_class[1]}")
                return wm_class
        except:
            pass

        # Try parent window
        parent = window.query_tree().parent
        if parent == root or parent == 0:
            break
        window = parent

    print("No WM_CLASS found")
    return None

if __name__ == "__main__":
    get_window_under_cursor()