In the attached program, I get the default icon when I use
enable_model_drag_source to enable the drag (set MANUAL to False).  When I
use drag_begin (set MANUAL to True), the icon is wrong.  I need to use
drag_begin so that I can disallow the drag under certain conditions. 
Thoughts?
-- 
Jeffrey Barish
# When I start the drag manually (using drag_begin), I do not get the default
# icon.

import gtk

MANUAL = True

numbers = ('zero', 'one', 'two', 'three', 'four', 'five', 'six')
data = [(str(i), d) for i, d in enumerate(numbers)]

class TreeViewDnDTest:
    def __init__(self):
        self.drag_context = None

        self.window = gtk.Window(gtk.WINDOW_TOPLEVEL)
        self.window.set_title("DND Test")
        self.window.set_size_request(200, 300)
        self.window.connect('delete-event', self.on_delete_event)

        self.liststore = gtk.ListStore(str, str)
        for item in data:
            self.liststore.append(item)

        self.treeview = gtk.TreeView(self.liststore)
        self.treeview.connect('drag-data-received', self.on_drag_data_received)
        if MANUAL:
            self.treeview.connect_after('drag-begin', self.on_drag_begin)
            self.treeview.connect('button-press-event', self.on_treeview_button_press_event)
            self.treeview.connect('motion-notify-event', self.on_treeview_motion_notify_event)

        renderer = gtk.CellRendererText()
        column = gtk.TreeViewColumn('Int', renderer, text=0)
        self.treeview.append_column(column)
        column = gtk.TreeViewColumn('Str', renderer, text=1)
        self.treeview.append_column(column)

        if not MANUAL:
            self.treeview.enable_model_drag_source(gtk.gdk.BUTTON1_MASK,
                    [('tree_model_row', gtk.TARGET_SAME_WIDGET, 1)],
                    gtk.gdk.ACTION_MOVE)
        self.treeview.enable_model_drag_dest(
                [('tree_model_row', 0, 1), ('tree_model_row_from_button', 0, 0)],
                gtk.gdk.ACTION_DEFAULT)

        self.vbox = gtk.VBox()
        self.vbox.pack_start(self.treeview, True)
        self.window.add(self.vbox)
        self.window.show_all()

    def on_delete_event(self, widget, event, data=None):
        gtk.main_quit()
        return False

    def on_drag_begin(self, widget, drag_context):
        # This call does not affect the icon:
        drag_context.set_icon_default()

    def on_drag_data_received(self, treeview, context, x, y, selection,
            info, etime):
        source_widget = context.get_source_widget()
        if source_widget == treeview:
            model, source_iter = treeview.get_selection().get_selected()
            source_row = model[source_iter]

        drop_info = treeview.get_dest_row_at_pos(x, y)
        if drop_info:
            path, position = drop_info
            iter = model.get_iter(path)
            if (position == gtk.TREE_VIEW_DROP_BEFORE
                or position == gtk.TREE_VIEW_DROP_INTO_OR_BEFORE):
                model.insert_before(iter, source_row)
            else:
                model.insert_after(iter, source_row)
        else:
            model.append(source_row)
        if source_iter:
            model.remove(source_iter)
        self.drag_context = None

    if MANUAL:
        def on_treeview_button_press_event(self, treeview, event):
            if event.button == 1:
                self.press_start_x = int(event.x)
                self.press_start_y = int(event.y)
                self.press_start_event = event

        def on_treeview_motion_notify_event(self, treeview, event):
            if event.state == gtk.gdk.BUTTON1_MASK:
                row = self.treeview.get_dest_row_at_pos(int(event.x), int(event.y))
                if row:
                    treepath, drop_pos = row
                    threshold = treeview.drag_check_threshold(
                            self.press_start_x, self.press_start_y,
                            int(event.x), int(event.y))
                    if self.drag_context is None and threshold:
                        targets = [('tree_model_row', gtk.TARGET_SAME_WIDGET, 1)]
                        actions = gtk.gdk.ACTION_MOVE
                        button = 1
                        self.drag_context = treeview.drag_begin(targets, actions,
                                button, self.press_start_event)
                        # This call does not affect the icon:
                        self.drag_context.set_icon_default()


if __name__ == '__main__':
    treeviewdnd = TreeViewDnDTest()
    gtk.main()

_______________________________________________
pygtk mailing list   pygtk@daa.com.au
http://www.daa.com.au/mailman/listinfo/pygtk
Read the PyGTK FAQ: http://www.async.com.br/faq/pygtk/

Reply via email to