def test_nonatomic(self):
m = ObjectTreeStore()
self.assertRaises(AssertionError, m.append, None)
self.assertRaises(AssertionError, m.insert, None, 0)
self.assertRaises(AssertionError, m.prepend, None)
self.assertRaises(AssertionError, m.insert_before, None, None)
self.assertRaises(AssertionError, m.insert_after, None, None)
def test_iterrows(self):
m = ObjectTreeStore()
obj = object()
obj2 = object()
it = m.append(None, row=[obj])
m.append(it, row=[obj2])
self.assertEqual(list(m.iterrows(None))[0][1], obj)
self.assertEqual(list(m.iterrows(it))[0][1], obj2)
def test_iterrows(self):
m = ObjectTreeStore()
obj = object()
obj2 = object()
it = m.append(None, row=[obj])
m.append(it, row=[obj2])
self.assertEqual(list(m.iterrows(None))[0][1], obj)
self.assertEqual(list(m.iterrows(it))[0][1], obj2)
def __init__(self, initial=None, folders=None):
"""
initial -- the path to select/scroll to
folders -- a list of paths to show in the tree view, None
will result in a separator.
"""
model = ObjectTreeStore()
super(DirectoryTree, self).__init__(model=model)
if initial is not None:
assert is_fsnative(initial)
column = TreeViewColumn(title=_("Folders"))
column.set_sizing(Gtk.TreeViewColumnSizing.AUTOSIZE)
render = Gtk.CellRendererPixbuf()
render.set_property('icon-name', Icons.FOLDER)
render.props.xpad = 3
column.pack_start(render, False)
render = Gtk.CellRendererText()
if self.supports_hints():
render.set_property('ellipsize', Pango.EllipsizeMode.END)
column.pack_start(render, True)
def cell_data(column, cell, model, iter_, userdata):
value = model.get_value(iter_)
if value is not None:
text = fsdecode(os.path.basename(value) or value)
cell.set_property('text', text)
column.set_cell_data_func(render, cell_data)
self.append_column(column)
self.set_search_equal_func(search_func, True)
self.set_search_column(0)
if folders is None:
folders = []
for path in folders:
niter = model.append(None, [path])
if path is not None:
assert is_fsnative(path)
model.append(niter, ["dummy"])
self.get_selection().set_mode(Gtk.SelectionMode.MULTIPLE)
self.connect(
'test-expand-row', DirectoryTree.__expanded, model)
self.set_row_separator_func(
lambda model, iter_, data: model.get_value(iter_) is None, None)
if initial:
self.go_to(initial)
menu = Gtk.Menu()
m = qltk.MenuItem(_(u"_New Folder…"), Icons.DOCUMENT_NEW)
m.connect('activate', self.__mkdir)
menu.append(m)
m = qltk.MenuItem(_("_Delete"), Icons.EDIT_DELETE)
m.connect('activate', self.__rmdir)
menu.append(m)
m = qltk.MenuItem(_("_Refresh"), Icons.VIEW_REFRESH)
m.connect('activate', self.__refresh)
menu.append(m)
m = qltk.MenuItem(_("_Select All Subfolders"), Icons.FOLDER)
m.connect('activate', self.__expand)
menu.append(m)
menu.show_all()
connect_obj(self, 'popup-menu', self.__popup_menu, menu)
# Allow to drag and drop files from outside
targets = [
("text/uri-list", 0, 42)
]
targets = [Gtk.TargetEntry.new(*t) for t in targets]
self.drag_dest_set(Gtk.DestDefaults.ALL, targets, Gdk.DragAction.COPY)
self.connect('drag-data-received', self.__drag_data_received)
def __init__(self, initial=None, folders=None):
"""
initial -- the path to select/scroll to
folders -- a list of paths to show in the tree view, None
will result in a separator.
"""
model = ObjectTreeStore()
super(DirectoryTree, self).__init__(model=model)
if initial is not None:
initial = util.fsnative(initial)
column = TreeViewColumn(_("Folders"))
column.set_sizing(Gtk.TreeViewColumnSizing.AUTOSIZE)
render = Gtk.CellRendererPixbuf()
render.set_property('stock_id', Gtk.STOCK_DIRECTORY)
column.pack_start(render, False)
render = Gtk.CellRendererText()
column.pack_start(render, True)
def cell_data(column, cell, model, iter_, userdata):
value = model.get_value(iter_)
if value is not None:
text = fsdecode(os.path.basename(value) or value)
cell.set_property('text', text)
column.set_cell_data_func(render, cell_data)
self.append_column(column)
self.set_search_equal_func(search_func, True)
if folders is None:
folders = []
for path in folders:
niter = model.append(None, [path])
if path is not None:
model.append(niter, ["dummy"])
self.get_selection().set_mode(Gtk.SelectionMode.MULTIPLE)
self.connect('test-expand-row', DirectoryTree.__expanded, model)
self.set_row_separator_func(
lambda model, iter_, data: model.get_value(iter_) is None, None)
if initial:
self.go_to(initial)
menu = Gtk.Menu()
m = qltk.MenuItem(_("_New Folder..."), Gtk.STOCK_NEW)
m.connect('activate', self.__mkdir)
menu.append(m)
m = Gtk.ImageMenuItem.new_from_stock(Gtk.STOCK_DELETE, None)
m.connect('activate', self.__rmdir)
menu.append(m)
m = Gtk.ImageMenuItem.new_from_stock(Gtk.STOCK_REFRESH, None)
m.connect('activate', self.__refresh)
menu.append(m)
m = qltk.MenuItem(_("_Select All Subfolders"), Gtk.STOCK_DIRECTORY)
m.connect('activate', self.__expand)
menu.append(m)
menu.show_all()
self.connect_object('popup-menu', self.__popup_menu, menu)
def __init__(self, initial=None, folders=None):
"""
initial -- the path to select/scroll to
folders -- a list of paths to show in the tree view, None
will result in a separator.
"""
model = ObjectTreeStore()
super(DirectoryTree, self).__init__(model=model)
if initial is not None:
assert isinstance(initial, fsnative)
column = TreeViewColumn(title=_("Folders"))
column.set_sizing(Gtk.TreeViewColumnSizing.AUTOSIZE)
render = Gtk.CellRendererPixbuf()
render.set_property('icon-name', Icons.FOLDER)
render.props.xpad = 3
column.pack_start(render, False)
render = Gtk.CellRendererText()
if self.supports_hints():
render.set_property('ellipsize', Pango.EllipsizeMode.END)
column.pack_start(render, True)
def cell_data(column, cell, model, iter_, userdata):
value = model.get_value(iter_)
if value is not None:
text = fsn2text(os.path.basename(value) or value)
cell.set_property('text', text)
column.set_cell_data_func(render, cell_data)
self.append_column(column)
self.set_search_equal_func(search_func, True)
self.set_search_column(0)
if folders is None:
folders = []
for path in folders:
niter = model.append(None, [path])
if path is not None:
assert isinstance(path, fsnative)
model.append(niter, [fsnative(u"dummy")])
self.get_selection().set_mode(Gtk.SelectionMode.MULTIPLE)
self.connect('test-expand-row', DirectoryTree.__expanded, model)
self.set_row_separator_func(
lambda model, iter_, data: model.get_value(iter_) is None, None)
if initial:
self.go_to(initial)
menu = self._create_menu()
connect_obj(self, 'popup-menu', self._popup_menu, menu)
# Allow to drag and drop files from outside
targets = [("text/uri-list", 0, 42)]
targets = [Gtk.TargetEntry.new(*t) for t in targets]
self.drag_dest_set(Gtk.DestDefaults.ALL, targets, Gdk.DragAction.COPY)
self.connect('drag-data-received', self.__drag_data_received)
def test_signal_count(self):
m = ObjectTreeStore()
def handler(model, path, iter_, result):
result[0] += 1
inserted = [0]
m.connect("row-inserted", handler, inserted)
changed = [0]
m.connect("row-changed", handler, changed)
m.append(None, [1])
m.insert(None, 0, [1])
m.prepend(None, [1])
m.insert_before(None, None, [1])
m.insert_after(None, None, [1])
self.assertEqual(changed[0], 0)
self.assertEqual(inserted[0], len(m))
def test_iter_path_changed(self):
m = ObjectTreeStore()
def handler(model, path, iter_, result):
result[0] += 1
result = [0]
m.connect("row-changed", handler, result)
m.append(None, [object()])
iter_ = m.get_iter_first()
m.iter_changed(iter_)
self.assertEqual(result[0], 1)
m.path_changed(m.get_path(iter_))
self.assertEqual(result[0], 2)
def test_signal_count(self):
m = ObjectTreeStore()
def handler(model, path, iter_, result):
result[0] += 1
inserted = [0]
m.connect("row-inserted", handler, inserted)
changed = [0]
m.connect("row-changed", handler, changed)
m.append(None, [1])
m.insert(None, 0, [1])
m.prepend(None, [1])
m.insert_before(None, None, [1])
m.insert_after(None, None, [1])
self.assertEqual(changed[0], 0)
self.assertEqual(inserted[0], len(m))
def test_iter_path_changed(self):
m = ObjectTreeStore()
def handler(model, path, iter_, result):
result[0] += 1
result = [0]
m.connect("row-changed", handler, result)
m.append(None, [object()])
iter_ = m.get_iter_first()
m.iter_changed(iter_)
self.assertEqual(result[0], 1)
m.path_changed(m.get_path(iter_))
self.assertEqual(result[0], 2)
def test_validate(self):
self.failUnlessRaises(ValueError, ObjectTreeStore, int)
ObjectTreeStore()
ObjectTreeStore(object)
self.failUnlessRaises(ValueError, ObjectTreeStore, object, object)
def test_tree_store_insert_after_none(self):
store = ObjectTreeStore()
root = store.append(None, [42])
sub = store.append(root, [24])
iter_ = store.insert_after(None, None, [1])
assert store.get_path(iter_).get_indices() == [0]
iter_ = store.insert_after(root, None, [1])
assert store.get_path(iter_).get_indices() == [1, 0]
iter_ = store.insert_after(sub, None, [1])
assert store.get_path(iter_).get_indices() == [1, 1, 0]
iter_ = store.insert_after(None, root, [1])
assert store.get_path(iter_).get_indices() == [2]
iter_ = store.insert_after(None, sub, [1])
assert store.get_path(iter_).get_indices() == [1, 2]
def __init__(self, initial=None, folders=None):
"""
initial -- the path to select/scroll to
folders -- a list of paths to show in the tree view, None
will result in a separator.
"""
model = ObjectTreeStore()
super(DirectoryTree, self).__init__(model=model)
if initial is not None:
assert is_fsnative(initial)
column = TreeViewColumn(_("Folders"))
column.set_sizing(Gtk.TreeViewColumnSizing.AUTOSIZE)
render = Gtk.CellRendererPixbuf()
render.set_property('stock_id', Gtk.STOCK_DIRECTORY)
column.pack_start(render, False)
render = Gtk.CellRendererText()
if self.supports_hints():
render.set_property('ellipsize', Pango.EllipsizeMode.END)
column.pack_start(render, True)
def cell_data(column, cell, model, iter_, userdata):
value = model.get_value(iter_)
if value is not None:
text = fsdecode(os.path.basename(value) or value)
cell.set_property('text', text)
column.set_cell_data_func(render, cell_data)
self.append_column(column)
self.set_search_equal_func(search_func, True)
self.set_search_column(0)
if folders is None:
folders = []
for path in folders:
niter = model.append(None, [path])
if path is not None:
assert is_fsnative(path)
model.append(niter, ["dummy"])
self.get_selection().set_mode(Gtk.SelectionMode.MULTIPLE)
self.connect('test-expand-row', DirectoryTree.__expanded, model)
self.set_row_separator_func(
lambda model, iter_, data: model.get_value(iter_) is None, None)
if initial:
self.go_to(initial)
menu = Gtk.Menu()
m = qltk.MenuItem(_(u"_New Folder…"), Gtk.STOCK_NEW)
m.connect('activate', self.__mkdir)
menu.append(m)
m = Gtk.ImageMenuItem.new_from_stock(Gtk.STOCK_DELETE, None)
m.connect('activate', self.__rmdir)
menu.append(m)
m = Gtk.ImageMenuItem.new_from_stock(Gtk.STOCK_REFRESH, None)
m.connect('activate', self.__refresh)
menu.append(m)
m = qltk.MenuItem(_("_Select All Subfolders"), Gtk.STOCK_DIRECTORY)
m.connect('activate', self.__expand)
menu.append(m)
menu.show_all()
connect_obj(self, 'popup-menu', self.__popup_menu, menu)
# Allow to drag and drop files from outside
targets = [("text/uri-list", 0, 42)]
targets = [Gtk.TargetEntry.new(*t) for t in targets]
self.drag_dest_set(Gtk.DestDefaults.ALL, targets, Gdk.DragAction.COPY)
self.connect('drag-data-received', self.__drag_data_received)
