cola/widgets/filetree.py

   1 from __future__ import absolute_import, division, unicode_literals
   2
   3 from qtpy import QtCore
   4 from qtpy import QtWidgets
   5
   6 from .. import icons
   7 from . import standard
   8
   9
  10 class FileTree(standard.TreeWidget):
  11
  12     def __init__(self, parent=None):
  13         standard.TreeWidget.__init__(self, parent=parent)
  14         self.setSelectionMode(self.ExtendedSelection)
  15         self.setHeaderHidden(True)
  16
  17     def set_filenames(self, filenames, select=False):
  18         self.clear()
  19         if not filenames:
  20             return
  21         items = []
  22         from_filename = icons.from_filename
  23         for filename in filenames:
  24             icon = from_filename(filename)
  25             item = QtWidgets.QTreeWidgetItem()
  26             item.setIcon(0, icon)
  27             item.setText(0, filename)
  28             item.setData(0, QtCore.Qt.UserRole, filename)
  29             items.append(item)
  30         self.addTopLevelItems(items)
  31         if select and items:
  32             items[0].setSelected(True)
  33
  34     def has_selection(self):
  35         return bool(self.selectedItems())
  36
  37     def selected_filenames(self):
  38         items = self.selectedItems()
  39         if not items:
  40             return []
  41         return [filename_from_item(i) for i in items]
  42
  43
  44 def filename_from_item(item):
  45     return item.data(0, QtCore.Qt.UserRole)