Python QTreeView.expand方法代码示例

# 需要导入模块: from PyQt5.Qt import QTreeView [as 别名]
# 或者: from PyQt5.Qt.QTreeView import expand [as 别名]
class ShortcutConfig(QWidget):  # {{{

    changed_signal = pyqtSignal()

    def __init__(self, parent=None):
        QWidget.__init__(self, parent)
        self._layout = l = QGridLayout()
        self.setLayout(self._layout)
        self.header = QLabel(_('Double click on any entry to change the'
            ' keyboard shortcuts associated with it'))
        l.addWidget(self.header, 0, 0, 1, 3)
        self.view = QTreeView(self)
        self.view.setAlternatingRowColors(True)
        self.view.setHeaderHidden(True)
        self.view.setAnimated(True)
        l.addWidget(self.view, 1, 0, 1, 3)
        self.delegate = Delegate()
        self.view.setItemDelegate(self.delegate)
        self.delegate.sizeHintChanged.connect(self.editor_opened,
                type=Qt.QueuedConnection)
        self.delegate.changed_signal.connect(self.changed_signal)
        self.search = SearchBox2(self)
        self.search.initialize('shortcuts_search_history',
                help_text=_('Search for a shortcut by name'))
        self.search.search.connect(self.find)
        l.addWidget(self.search, 2, 0, 1, 1)
        self.nb = QPushButton(QIcon(I('arrow-down.png')), _('&Next'), self)
        self.pb = QPushButton(QIcon(I('arrow-up.png')), _('&Previous'), self)
        self.nb.clicked.connect(self.find_next)
        self.pb.clicked.connect(self.find_previous)
        l.addWidget(self.nb, 2, 1, 1, 1)
        l.addWidget(self.pb, 2, 2, 1, 1)
        l.setColumnStretch(0, 100)

    def restore_defaults(self):
        self._model.restore_defaults()
        self.changed_signal.emit()

    def commit(self):
        if self.view.state() == self.view.EditingState:
            self.delegate.accept_changes()
        self._model.commit()

    def initialize(self, keyboard):
        self._model = ConfigModel(keyboard, parent=self)
        self.view.setModel(self._model)

    def editor_opened(self, index):
        self.view.scrollTo(index, self.view.EnsureVisible)

    @property
    def is_editing(self):
        return self.view.state() == self.view.EditingState

    def find(self, query):
        if not query:
            return
        try:
            idx = self._model.find(query)
        except ParseException:
            self.search.search_done(False)
            return
        self.search.search_done(True)
        if not idx.isValid():
            info_dialog(self, _('No matches'),
                    _('Could not find any shortcuts matching %s')%query,
                    show=True, show_copy_button=False)
            return
        self.highlight_index(idx)

    def highlight_index(self, idx):
        self.view.scrollTo(idx)
        self.view.selectionModel().select(idx,
                self.view.selectionModel().ClearAndSelect)
        self.view.setCurrentIndex(idx)
        self.view.setFocus(Qt.OtherFocusReason)

    def find_next(self, *args):
        idx = self.view.currentIndex()
        if not idx.isValid():
            idx = self._model.index(0, 0)
        idx = self._model.find_next(idx,
                unicode(self.search.currentText()))
        self.highlight_index(idx)

    def find_previous(self, *args):
        idx = self.view.currentIndex()
        if not idx.isValid():
            idx = self._model.index(0, 0)
        idx = self._model.find_next(idx,
            unicode(self.search.currentText()), backwards=True)
        self.highlight_index(idx)

    def highlight_group(self, group_name):
        idx = self.view.model().index_for_group(group_name)
        if idx is not None:
            self.view.expand(idx)
            self.view.scrollTo(idx, self.view.PositionAtTop)
            self.view.selectionModel().select(idx,
                    self.view.selectionModel().ClearAndSelect)
#.........这里部分代码省略.........

# 需要导入模块: from PyQt5.Qt import QTreeView [as 别名]
# 或者: from PyQt5.Qt.QTreeView import expand [as 别名]
class Navigation(QWidget):
    """a navigation widget, 
    displaying all known projects and samples
    """
    changed_projects = pyqtSignal(str, str)
    changed_allele = pyqtSignal(str, int, str)
    change_view = pyqtSignal(int)
    refresh = pyqtSignal(str)
    
    def __init__(self, log, settings):
        self.log = log
        self.settings = settings
        super().__init__()
        self.init_UI()
        self.create_model()
        self.tree.expand(self.open_node)
        self.tree.clicked.connect(self.onClicked)
        
        self.tree.setContextMenuPolicy(Qt.CustomContextMenu)
        self.tree.customContextMenuRequested.connect(self.open_menu)
        
    def init_UI(self):
        """creates the UI
        """
        self.log.debug("Setting up Navigation area UI...")
        self.tree = QTreeView()
        self.tree.doubleClicked.connect(self.onDoubleClick)
        self.grid = QGridLayout()
        self.setLayout(self.grid)
        self.header_lbl = QLabel("Projects and Samples:", self)
        self.header_lbl.setStyleSheet(general.label_style_2nd)
        self.grid.addWidget(self.header_lbl, 0,0)
        
        self.grid.addWidget(self.tree, 1, 0)
        header = self.tree.header()
        header.hide()
        self.log.debug("\t=> Done!")
        
    def create_model(self):
        """creates or recreates the model
        """
        self.log.debug("Updating Navigation area...")
        root_node = Node("All")
        openNode = Node("Open", root_node)
        closedNode = Node("Closed", root_node)
        project_nodes = {}
        
        query = """select distinct sample_id_int, projects.project_name, project_status, allele_status, allele_nr 
        from projects left join alleles
            on projects.project_name = alleles.project_name 
        order by project_status desc, projects.project_name desc, sample_id_int
        """
        q = QtSql.QSqlQuery()
        q.exec_(query)
        while q.next():
            sample = q.value(0)
            project = q.value(1)
            pstatus = q.value(2)
            astatus = q.value(3)
            nr = q.value(4)
            try:
                pnode = project_nodes[project]
            except KeyError:
                if pstatus == "Closed":
                    topnode = closedNode
                else:
                    topnode = openNode
                pnode = Node(project, topnode, "Project")
                project_nodes[project] = pnode
            if sample:
                if nr != 1:
                    sample = "{} ({})".format(sample, nr)
                Node(sample, pnode, "Sample", astatus)
        
        self.model = NavigationModel(root_node)
        self.tree.setModel(self.model)
        
        self.open_node = self.model.index(0, 0, QModelIndex())
        self.closed_node = self.model.index(1, 0, QModelIndex())
        self.root_node = self.open_node.parent()
        self.log.debug("\t=> Done!")
        
    @pyqtSlot(str, str)
    def expand_project(self, project, status = None):
        """expands the node of the given project, 
        returns its index and parent-node's index 
        """
        self.log.debug("Expanding project {} ({})...".format(project, status))
        if not status in (None, "Open", "Closed"):
            return None, None
        myindex = self.open_node
        myparent_node = self.open_node
        if status == "Open":
            self.log.debug("Open")
            parent_node = self.open_node
            self.tree.expand(parent_node)
            self.tree.collapse(self.closed_node)
        elif status == "Closed":
            self.log.debug("Closed")
            parent_node = self.closed_node
#.........这里部分代码省略.........