__init__.py
Python script, ASCII text executable
1""" 2Traditional window list applet for the Panorama panel, compatible 3with wlroots-based compositors. 4Copyright 2025, roundabout-host.com <vlad@roundabout-host.com> 5 6This program is free software: you can redistribute it and/or modify 7it under the terms of the GNU General Public Licence as published by 8the Free Software Foundation, either version 3 of the Licence, or 9(at your option) any later version. 10 11This program is distributed in the hope that it will be useful, 12but WITHOUT ANY WARRANTY; without even the implied warranty of 13MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 14GNU General Public Licence for more details. 15 16You should have received a copy of the GNU General Public Licence 17along with this program. If not, see <https://www.gnu.org/licenses/>. 18""" 19 20import dataclasses 21import os 22import sys 23import locale 24import typing 25from pathlib import Path 26from pywayland.client import Display, EventQueue 27from pywayland.protocol.wayland import WlRegistry, WlSeat, WlSurface, WlCompositor, WlOutput 28from pywayland.protocol.wayland.wl_output import WlOutputProxy 29from pywayland.protocol.wlr_foreign_toplevel_management_unstable_v1 import ( 30ZwlrForeignToplevelManagerV1, 31ZwlrForeignToplevelHandleV1 32) 33import panorama_panel 34 35import gi 36 37gi.require_version("Gtk", "4.0") 38gi.require_version("GdkWayland", "4.0") 39 40from gi.repository import Gtk, GLib, Gtk4LayerShell, Gio, Gdk, Pango, GObject 41 42 43module_directory = Path(__file__).resolve().parent 44 45locale.bindtextdomain("panorama-window-list", module_directory / "locale") 46_ = lambda x: locale.dgettext("panorama-window-list", x) 47 48 49import ctypes 50from cffi import FFI 51ffi = FFI() 52ffi.cdef(""" 53void * gdk_wayland_display_get_wl_display (void * display); 54void * gdk_wayland_surface_get_wl_surface (void * surface); 55void * gdk_wayland_monitor_get_wl_output (void * monitor); 56""") 57gtk = ffi.dlopen("libgtk-4.so.1") 58 59 60@Gtk.Template(filename=str(module_directory / "panorama-window-list-options.ui")) 61class WindowListOptions(Gtk.Window): 62__gtype_name__ = "WindowListOptions" 63button_width_adjustment: Gtk.Adjustment = Gtk.Template.Child() 64workspace_filter_checkbutton: Gtk.CheckButton = Gtk.Template.Child() 65 66def __init__(self, **kwargs): 67super().__init__(**kwargs) 68 69self.connect("close-request", lambda *args: self.destroy()) 70 71 72def split_bytes_into_ints(array: bytes, size: int = 4) -> list[int]: 73if len(array) % size: 74raise ValueError(f"The byte string's length must be a multiple of {size}") 75 76values: list[int] = [] 77for i in range(0, len(array), size): 78values.append(int.from_bytes(array[i : i+size], byteorder=sys.byteorder)) 79 80return values 81 82 83def get_widget_rect(widget: Gtk.Widget) -> tuple[int, int, int, int]: 84width = int(widget.get_width()) 85height = int(widget.get_height()) 86 87toplevel = widget.get_root() 88if not toplevel: 89return None, None, width, height 90 91x, y = widget.translate_coordinates(toplevel, 0, 0) 92x = int(x) 93y = int(y) 94 95return x, y, width, height 96 97 98@dataclasses.dataclass 99class WindowState: 100minimised: bool 101maximised: bool 102fullscreen: bool 103focused: bool 104 105@classmethod 106def from_state_array(cls, array: bytes): 107values = split_bytes_into_ints(array) 108instance = cls(False, False, False, False) 109for value in values: 110match value: 111case 0: 112instance.maximised = True 113case 1: 114instance.minimised = True 115case 2: 116instance.focused = True 117case 3: 118instance.fullscreen = True 119 120return instance 121 122 123from gi.repository import Gtk, Gdk 124 125class WindowButtonOptions: 126def __init__(self, max_width: int): 127self.max_width = max_width 128 129class WindowButtonLayoutManager(Gtk.LayoutManager): 130def __init__(self, options: WindowButtonOptions, **kwargs): 131super().__init__(**kwargs) 132self.options = options 133 134def do_measure(self, widget, orientation, for_size): 135child = widget.get_first_child() 136if child is None: 137return 0, 0, 0, 0 138 139if orientation == Gtk.Orientation.HORIZONTAL: 140min_width, nat_width, min_height, nat_height = child.measure(Gtk.Orientation.HORIZONTAL, for_size) 141width = self.options.max_width 142return min_width, width, min_height, nat_height 143else: 144min_width, nat_width, min_height, nat_height = child.measure(Gtk.Orientation.VERTICAL, for_size) 145return min_height, nat_height, 0, 0 146 147def do_allocate(self, widget, width, height, baseline): 148child = widget.get_first_child() 149if child is None: 150return 151alloc_width = min(width, self.options.max_width) 152alloc = Gdk.Rectangle() 153alloc.x = 0 154alloc.y = 0 155alloc.width = alloc_width 156alloc.height = height 157child.allocate(alloc.width, alloc.height, baseline) 158 159 160class WindowButton(Gtk.ToggleButton): 161def __init__(self, window_id, window_title, **kwargs): 162super().__init__(**kwargs) 163 164self.window_id: ZwlrForeignToplevelHandleV1 = window_id 165self.wf_ipc_id: typing.Optional[int] = None 166self.set_has_frame(False) 167self.label = Gtk.Label() 168self.icon = Gtk.Image.new_from_icon_name("application-x-executable") 169box = Gtk.Box() 170box.append(self.icon) 171box.append(self.label) 172self.set_child(box) 173 174self.window_title = window_title 175self.window_state = WindowState(False, False, False, False) 176 177self.label.set_ellipsize(Pango.EllipsizeMode.END) 178self.set_hexpand(True) 179self.set_vexpand(True) 180 181self.drag_source = Gtk.DragSource(actions=Gdk.DragAction.MOVE) 182self.drag_source.connect("prepare", self.provide_drag_data) 183self.drag_source.connect("drag-begin", self.drag_begin) 184self.drag_source.connect("drag-cancel", self.drag_cancel) 185 186self.add_controller(self.drag_source) 187 188self.menu = Gio.Menu() 189# TODO: toggle the labels when needed 190self.minimise_item = Gio.MenuItem.new(_("_Minimise"), "button.minimise") 191self.maximise_item = Gio.MenuItem.new(_("Ma_ximise"), "button.maximise") 192self.menu.append_item(self.minimise_item) 193self.menu.append_item(self.maximise_item) 194self.menu.append(_("_Close"), "button.close") 195self.menu.append(_("Window list _options"), "applet.options") 196self.popover_menu = Gtk.PopoverMenu.new_from_model(self.menu) 197self.popover_menu.set_parent(self) 198self.popover_menu.set_flags(Gtk.PopoverMenuFlags.NESTED) 199self.popover_menu.set_has_arrow(False) 200self.popover_menu.set_halign(Gtk.Align.END) 201 202self.right_click_controller = Gtk.GestureClick(button=3) 203self.right_click_controller.connect("pressed", self.show_menu) 204self.add_controller(self.right_click_controller) 205 206self.action_group = Gio.SimpleActionGroup() 207close_action = Gio.SimpleAction.new("close") 208close_action.connect("activate", self.close_associated) 209self.action_group.insert(close_action) 210minimise_action = Gio.SimpleAction.new("minimise") 211minimise_action.connect("activate", self.minimise_associated) 212self.action_group.insert(minimise_action) 213maximise_action = Gio.SimpleAction.new("maximise") 214maximise_action.connect("activate", self.maximise_associated) 215self.action_group.insert(maximise_action) 216 217self.insert_action_group("button", self.action_group) 218 219self.middle_click_controller = Gtk.GestureClick(button=2) 220self.middle_click_controller.connect("released", self.close_associated) 221self.add_controller(self.middle_click_controller) 222 223def show_menu(self, gesture, n_presses, x, y): 224rect = Gdk.Rectangle() 225rect.x = int(x) 226rect.y = int(y) 227rect.width = 1 228rect.height = 1 229self.popover_menu.popup() 230 231def close_associated(self, *args): 232self.window_id.close() 233 234def minimise_associated(self, action, *args): 235if self.window_state.minimised: 236self.window_id.unset_minimized() 237else: 238self.window_id.set_minimized() 239 240def maximise_associated(self, action, *args): 241if self.window_state.maximised: 242self.window_id.unset_maximized() 243else: 244self.window_id.set_maximized() 245 246def provide_drag_data(self, source: Gtk.DragSource, x: float, y: float): 247app = self.get_root().get_application() 248app.drags[id(self)] = self 249value = GObject.Value() 250value.init(GObject.TYPE_UINT64) 251value.set_uint64(id(self)) 252return Gdk.ContentProvider.new_for_value(value) 253 254def drag_begin(self, source: Gtk.DragSource, drag: Gdk.Drag): 255paintable = Gtk.WidgetPaintable.new(self).get_current_image() 256source.set_icon(paintable, 0, 0) 257self.hide() 258 259def drag_cancel(self, source: Gtk.DragSource, drag: Gdk.Drag, reason: Gdk.DragCancelReason): 260self.show() 261return False 262 263@property 264def window_title(self): 265return self.label.get_text() 266 267@window_title.setter 268def window_title(self, value): 269self.label.set_text(value) 270 271def set_icon_from_app_id(self, app_id): 272app_ids = app_id.split() 273 274# If on Wayfire, find the IPC ID 275for app_id in app_ids: 276if app_id.startswith("wf-ipc-"): 277self.wf_ipc_id = int(app_id.removeprefix("wf-ipc-")) 278break 279 280# Try getting an icon from the correct theme 281icon_theme = Gtk.IconTheme.get_for_display(self.get_display()) 282 283for app_id in app_ids: 284if icon_theme.has_icon(app_id): 285self.icon.set_from_icon_name(app_id) 286return 287 288# If that doesn't work, try getting one from .desktop files 289for app_id in app_ids: 290try: 291desktop_file = Gio.DesktopAppInfo.new(app_id + ".desktop") 292if desktop_file: 293self.icon.set_from_gicon(desktop_file.get_icon()) 294return 295except TypeError: 296# Due to a bug, the constructor may sometimes return C NULL 297pass 298 299 300class WFWindowList(panorama_panel.Applet): 301name = _("Wayfire window list") 302description = _("Traditional window list (for Wayfire and other wlroots compositors)") 303 304def __init__(self, orientation=Gtk.Orientation.HORIZONTAL, config=None): 305super().__init__(orientation=orientation, config=config) 306if config is None: 307config = {} 308 309self.set_homogeneous(True) 310self.window_button_options = WindowButtonOptions(config.get("max_button_width", 256)) 311self.show_only_this_wf_workspace = config.get("show_only_this_wf_workspace", True) 312self.show_only_this_output = config.get("show_only_this_output", True) 313 314self.toplevel_buttons: dict[ZwlrForeignToplevelHandleV1, WindowButton] = {} 315self.toplevel_buttons_by_wf_id: dict[int, WindowButton] = {} 316# This button doesn't belong to any window but is used for the button group and to be 317# selected when no window is focused 318self.initial_button = Gtk.ToggleButton() 319 320self.display = None 321self.my_output = None 322self.wl_surface_ptr = None 323self.registry = None 324self.compositor = None 325self.seat = None 326 327self.context_menu = self.make_context_menu() 328panorama_panel.track_popover(self.context_menu) 329 330right_click_controller = Gtk.GestureClick() 331right_click_controller.set_button(3) 332right_click_controller.connect("pressed", self.show_context_menu) 333 334self.add_controller(right_click_controller) 335 336action_group = Gio.SimpleActionGroup() 337options_action = Gio.SimpleAction.new("options", None) 338options_action.connect("activate", self.show_options) 339action_group.add_action(options_action) 340self.insert_action_group("applet", action_group) 341# Wait for the widget to be in a layer-shell window before doing this 342self.connect("realize", self.get_wl_resources) 343 344self.options_window = None 345 346# Support button reordering 347self.drop_target = Gtk.DropTarget.new(GObject.TYPE_UINT64, Gdk.DragAction.MOVE) 348self.drop_target.set_gtypes([GObject.TYPE_UINT64]) 349self.drop_target.connect("drop", self.drop_button) 350 351self.add_controller(self.drop_target) 352 353# Make a Wayfire socket for workspace handling 354try: 355import wayfire 356self.wf_socket = wayfire.WayfireSocket() 357self.wf_socket.watch() 358fd = self.wf_socket.client.fileno() 359GLib.io_add_watch(GLib.IOChannel.unix_new(fd), GLib.IO_IN, self.on_wf_event, priority=GLib.PRIORITY_HIGH) 360except: 361# Wayfire raises Exception itself, so it cannot be narrowed down 362self.wf_socket = None 363 364def get_wf_output_by_name(self, name): 365if not self.wf_socket: 366return None 367for output in self.wf_socket.list_outputs(): 368if output["name"] == name: 369return output 370return None 371 372def on_wf_event(self, source, condition): 373if condition & GLib.IO_IN: 374try: 375message = self.wf_socket.read_next_event() 376event = message.get("event") 377match event: 378case "view-workspace-changed": 379view = message.get("view", {}) 380if view["output-name"] == self.get_root().monitor_name: 381output = self.get_wf_output_by_name(view["output-name"]) 382current_workspace = output["workspace"]["x"], output["workspace"]["y"] 383if not self.show_only_this_wf_workspace or (message["to"]["x"], message["to"]["y"]) == current_workspace: 384if self.toplevel_buttons_by_wf_id[view["id"]].get_parent() is None: 385self.append(self.toplevel_buttons_by_wf_id[view["id"]]) 386else: 387if self.toplevel_buttons_by_wf_id[view["id"]].get_parent() is self: 388# Remove out-of-workspace window 389self.remove(self.toplevel_buttons_by_wf_id[view["id"]]) 390case "wset-workspace-changed": 391output_name = self.get_root().monitor_name 392if message["wset-data"]["output-name"] == output_name: 393# It has changed on this monitor; refresh the window list 394self.filter_to_wf_workspace() 395 396except Exception as e: 397print("Error reading Wayfire event:", e) 398return True 399 400def drop_button(self, drop_target: Gtk.DropTarget, value: int, x: float, y: float): 401button: WindowButton = self.get_root().get_application().drags.pop(value) 402if button.get_parent() is not self: 403# Prevent dropping a button from another window list 404return False 405 406self.remove(button) 407# Find the position where to insert the applet 408# Probably we could use the assumption that buttons are homogeneous here for efficiency 409child = self.get_first_child() 410while child: 411allocation = child.get_allocation() 412child_x, child_y = self.translate_coordinates(self, 0, 0) 413if self.get_orientation() == Gtk.Orientation.HORIZONTAL: 414midpoint = child_x + allocation.width / 2 415if x < midpoint: 416button.insert_before(self, child) 417break 418elif self.get_orientation() == Gtk.Orientation.VERTICAL: 419midpoint = child_y + allocation.height / 2 420if y < midpoint: 421button.insert_before(self, child) 422break 423child = child.get_next_sibling() 424else: 425self.append(button) 426button.show() 427 428self.set_all_rectangles() 429return True 430 431def filter_to_wf_workspace(self): 432if not self.show_only_this_wf_workspace: 433return 434 435output = self.get_wf_output_by_name(self.get_root().monitor_name) 436for wf_id, button in self.toplevel_buttons_by_wf_id.items(): 437view = self.wf_socket.get_view(wf_id) 438mid_x = view["geometry"]["x"] + view["geometry"]["width"] / 2 439mid_y = view["geometry"]["y"] + view["geometry"]["height"] / 2 440output_width = output["geometry"]["width"] 441output_height = output["geometry"]["height"] 442if 0 <= mid_x < output_width and 0 <= mid_y < output_height: 443# It is in this workspace; keep it 444if button.get_parent() is None: 445self.append(button) 446else: 447# Remove it from this window list 448if button.get_parent() is self: 449self.remove(button) 450 451def get_wl_resources(self, widget): 452ctypes.pythonapi.PyCapsule_GetPointer.restype = ctypes.c_void_p 453ctypes.pythonapi.PyCapsule_GetPointer.argtypes = (ctypes.py_object,) 454 455self.display = Display() 456wl_display_ptr = gtk.gdk_wayland_display_get_wl_display( 457ffi.cast("void *", ctypes.pythonapi.PyCapsule_GetPointer(self.get_root().get_native().get_display().__gpointer__, None))) 458self.display._ptr = wl_display_ptr 459self.event_queue = EventQueue(self.display) 460 461# Intentionally commented: the display is already connected by GTK 462# self.display.connect() 463 464my_monitor = Gtk4LayerShell.get_monitor(self.get_root()) 465 466# Iterate through monitors and get their Wayland output (wl_output) 467# This is a hack to ensure output_enter/leave is called for toplevels 468for monitor in self.get_root().get_application().monitors: 469wl_output = gtk.gdk_wayland_monitor_get_wl_output(ffi.cast("void *", ctypes.pythonapi.PyCapsule_GetPointer(monitor.__gpointer__, None))) 470if wl_output and not monitor.output_proxy: 471print("Create proxy") 472output_proxy = WlOutputProxy(wl_output, self.display) 473output_proxy.interface.registry[output_proxy._ptr] = output_proxy 474monitor.output_proxy = output_proxy 475 476if monitor == my_monitor: 477self.my_output = monitor.output_proxy 478# End hack 479 480self.manager = None 481self.registry = self.display.get_registry() 482self.registry.dispatcher["global"] = self.on_global 483self.display.roundtrip() 484if self.manager is None: 485print("Could not load wf-window-list. Is foreign-toplevel protocol advertised by the compositor?") 486print("(Wayfire requires enabling foreign-toplevel plugin)") 487return 488self.manager.dispatcher["toplevel"] = self.on_new_toplevel 489self.manager.dispatcher["finished"] = lambda *a: print("Toplevel manager finished") 490self.display.roundtrip() 491fd = self.display.get_fd() 492GLib.io_add_watch(fd, GLib.IO_IN, self.on_display_event) 493 494if self.wf_socket is not None: 495self.filter_to_wf_workspace() 496 497def on_display_event(self, source, condition): 498if condition & GLib.IO_IN: 499self.display.dispatch(queue=self.event_queue) 500return True 501 502def on_global(self, registry, name, interface, version): 503if interface == "zwlr_foreign_toplevel_manager_v1": 504self.print_log("Interface registered") 505self.manager = registry.bind(name, ZwlrForeignToplevelManagerV1, version) 506elif interface == "wl_seat": 507self.print_log("Seat found") 508self.seat = registry.bind(name, WlSeat, version) 509elif interface == "wl_compositor": 510self.compositor = registry.bind(name, WlCompositor, version) 511self.wl_surface_ptr = gtk.gdk_wayland_surface_get_wl_surface( 512ffi.cast("void *", ctypes.pythonapi.PyCapsule_GetPointer( 513self.get_root().get_native().get_surface().__gpointer__, None))) 514 515def on_new_toplevel(self, manager: ZwlrForeignToplevelManagerV1, 516handle: ZwlrForeignToplevelHandleV1): 517handle.dispatcher["title"] = self.on_title_changed 518handle.dispatcher["app_id"] = self.on_app_id_changed 519handle.dispatcher["output_enter"] = self.on_output_entered 520handle.dispatcher["output_leave"] = self.on_output_left 521handle.dispatcher["state"] = self.on_state_changed 522handle.dispatcher["closed"] = self.on_closed 523 524def on_output_entered(self, handle, output): 525button = WindowButton(handle, handle.title) 526 527if self.show_only_this_output and output != self.my_output: 528return 529 530self.set_title(button, handle.title) 531button.set_group(self.initial_button) 532button.set_layout_manager(WindowButtonLayoutManager(self.window_button_options)) 533button.connect("clicked", self.on_button_click) 534self.set_app_id(button, handle.app_id) 535self.toplevel_buttons[handle] = button 536self.append(button) 537self.set_all_rectangles() 538 539def on_output_left(self, handle, output): 540if handle in self.toplevel_buttons: 541button = self.toplevel_buttons[handle] 542self.remove(button) 543self.set_all_rectangles() 544 545def set_title(self, button, title): 546button.window_title = title 547button.set_tooltip_text(title) 548 549def on_title_changed(self, handle, title): 550handle.title = title 551if handle in self.toplevel_buttons: 552button = self.toplevel_buttons[handle] 553self.set_title(button, title) 554 555def set_all_rectangles(self): 556child = self.get_first_child() 557while child is not None: 558if isinstance(child, WindowButton): 559surface = WlSurface() 560surface._ptr = self.wl_surface_ptr 561child.window_id.set_rectangle(surface, *get_widget_rect(child)) 562 563child = child.get_next_sibling() 564 565def on_button_click(self, button: WindowButton): 566# Set a rectangle for animation 567surface = WlSurface() 568surface._ptr = self.wl_surface_ptr 569button.window_id.set_rectangle(surface, *get_widget_rect(button)) 570if button.window_state.focused: 571# Already pressed in, so minimise the focused window 572button.window_id.set_minimized() 573else: 574button.window_id.unset_minimized() 575button.window_id.activate(self.seat) 576 577def on_state_changed(self, handle, states): 578if handle in self.toplevel_buttons: 579state_info = WindowState.from_state_array(states) 580button = self.toplevel_buttons[handle] 581button.window_state = state_info 582if state_info.focused: 583button.set_active(True) 584else: 585self.initial_button.set_active(True) 586 587self.set_all_rectangles() 588 589def set_app_id(self, button, app_id): 590button.set_icon_from_app_id(app_id) 591app_ids = app_id.split() 592for app_id in app_ids: 593if app_id.startswith("wf-ipc-"): 594self.toplevel_buttons_by_wf_id[int(app_id.removeprefix("wf-ipc-"))] = button 595 596def on_app_id_changed(self, handle, app_id): 597handle.app_id = app_id 598if handle in self.toplevel_buttons: 599button = self.toplevel_buttons[handle] 600self.set_app_id(button, app_id) 601 602def on_closed(self, handle): 603button: WindowButton = self.toplevel_buttons[handle] 604wf_id = button.wf_ipc_id 605if handle in self.toplevel_buttons: 606self.remove(self.toplevel_buttons[handle]) 607self.toplevel_buttons.pop(handle) 608if wf_id in self.toplevel_buttons_by_wf_id: 609self.toplevel_buttons_by_wf_id.pop(wf_id) 610 611self.set_all_rectangles() 612 613def make_context_menu(self): 614menu = Gio.Menu() 615menu.append(_("Window list _options"), "applet.options") 616context_menu = Gtk.PopoverMenu.new_from_model(menu) 617context_menu.set_has_arrow(False) 618context_menu.set_parent(self) 619context_menu.set_halign(Gtk.Align.START) 620context_menu.set_flags(Gtk.PopoverMenuFlags.NESTED) 621return context_menu 622 623def show_context_menu(self, gesture, n_presses, x, y): 624rect = Gdk.Rectangle() 625rect.x = int(x) 626rect.y = int(y) 627rect.width = 1 628rect.height = 1 629 630self.context_menu.set_pointing_to(rect) 631self.context_menu.popup() 632 633def show_options(self, _0=None, _1=None): 634if self.options_window is None: 635self.options_window = WindowListOptions() 636self.options_window.button_width_adjustment.set_value(self.window_button_options.max_width) 637self.options_window.button_width_adjustment.connect("value-changed", self.update_button_options) 638self.options_window.workspace_filter_checkbutton.set_active(self.show_only_this_wf_workspace) 639self.options_window.workspace_filter_checkbutton.connect("toggled", self.update_workspace_filter) 640 641def reset_window(*args): 642self.options_window = None 643 644self.options_window.connect("close-request", reset_window) 645self.options_window.present() 646 647def remove_workspace_filtering(self): 648for button in self.toplevel_buttons.values(): 649if button.get_parent() is None: 650self.append(button) 651 652def update_workspace_filter(self, checkbutton): 653self.show_only_this_wf_workspace = checkbutton.get_active() 654if checkbutton.get_active(): 655self.filter_to_wf_workspace() 656else: 657self.remove_workspace_filtering() 658 659def update_button_options(self, adjustment): 660self.window_button_options.max_width = adjustment.get_value() 661child: Gtk.Widget = self.get_first_child() 662while child: 663child.queue_allocate() 664child.queue_resize() 665child.queue_draw() 666child = child.get_next_sibling() 667 668self.emit("config-changed") 669 670def get_config(self): 671return { 672"max_button_width": self.window_button_options.max_width, 673"show_only_this_wf_workspace": self.show_only_this_wf_workspace, 674"show_only_this_output": self.show_only_this_output, 675} 676 677def output_changed(self): 678self.get_wl_resources() 679 680def make_draggable(self): 681for button in self.toplevel_buttons.values(): 682button.remove_controller(button.drag_source) 683button.set_sensitive(False) 684 685def restore_drag(self): 686for button in self.toplevel_buttons.values(): 687button.add_controller(button.drag_source) 688button.set_sensitive(True) 689