Exemple #1
0
 def __init__(self, window, parent=None):
     super().__init__(parent)
     self.window = window
     self.ui = Ui_ProjectDock()
     self.ui.setupUi(self)
     self.ui.projectTree.setContextMenuPolicy(Qt.CustomContextMenu)
     self.ui.projectTree.customContextMenuRequested.connect(
         self.onContextMenu)
     self.onDocsChanged()
 def __init__(self, window, parent=None):
     super().__init__(parent)
     self.window = window
     self.ui = Ui_ProjectDock()
     self.ui.setupUi(self)
     self.ui.projectTree.setContextMenuPolicy(Qt.CustomContextMenu)
     self.ui.projectTree.customContextMenuRequested.connect(self.onContextMenu)
     self.onDocsChanged()
class ProjectDock(QDockWidget):
    openPage = pyqtSignal("QObject")

    class PageElement(object):
        def __init__(self, page):
            self.name = page.name
            self.page = page
            self.subelements = []

        @property
        def doc(self):
            return self.page.parentDoc

    class SymElement(object):
        def __init__(self, sym):
            self.name = "Symbol: " + sym.name
            self.page = sym
            self.subelements = []

        @property
        def doc(self):
            return self.page.parentDoc

    class DocElement(QObject):
        def __init__(self, doc: MasterDocument):
            super().__init__()
            self._doc = doc
            doc.sigChanged.connect(self.onDocChanged)
            self.name = None
            self.subelements = []
            self.onDocChanged()

        @property
        def doc(self):
            return self._doc

        @property
        def page(self):
            return None

        @pyqtSlot()
        def onDocChanged(self):
            self.name = self._doc.name()
            self.subelements = []
            for s in self._doc.symbols:
                self.subelements.append(ProjectDock.SymElement(s))
            for p in self._doc.pages:
                self.subelements.append(ProjectDock.PageElement(p))

    class Node(AbstractTreeNode):
        def __init__(self, ref, parent, row):
            self.ref = ref
            super().__init__(parent, row)

        def _getChildren(self):
            return [ProjectDock.Node(elem, self, index) for index, elem in enumerate(self.ref.subelements)]

    class Model(AbstractTreeModel):
        def __init__(self, rootElements):
            self.rootElements = rootElements
            super().__init__()

        def _getRootNodes(self):
            return [ProjectDock.Node(elem, None, index) for index, elem in enumerate(self.rootElements)]

        def columnCount(self, parent):
            return 1

        def data(self, index, role):
            if not index.isValid():
                return None
            node = index.internalPointer()
            if role == Qt.DisplayRole:
                return node.ref.name
            if role == Qt.UserRole:
                return node.ref
            return None

    def __init__(self, window, parent=None):
        super().__init__(parent)
        self.window = window
        self.ui = Ui_ProjectDock()
        self.ui.setupUi(self)
        self.ui.projectTree.setContextMenuPolicy(Qt.CustomContextMenu)
        self.ui.projectTree.customContextMenuRequested.connect(self.onContextMenu)
        self.onDocsChanged()

    @pyqtSlot()
    def onDocsChanged(self):
        self.ui.projectTree.setModel(ProjectDock.Model([ProjectDock.DocElement(d) for d in self.window.docs]))
        self.ui.projectTree.expandAll()

    @pyqtSlot(QModelIndex)
    def on_projectTree_doubleClicked(self, idx: QModelIndex):
        p = idx.data(Qt.UserRole).page
        if p is not None:
            self.openPage.emit(p)

    @pyqtSlot(QPoint)
    def onContextMenu(self, pt: QPoint):
        index = self.ui.projectTree.indexAt(pt)
        newDocAction = QAction("New document", None)
        newDocAction.triggered.connect(self.window.on_actionNew_triggered)
        menu = QMenu()
        menu.addAction(newDocAction)
        if index.isValid():
            obj = index.data(Qt.UserRole)

            def newPage():
                obj.doc.appendNewPage()
                self.onDocsChanged()

            def newSym():
                obj.doc.appendNewSymbol()
                self.onDocsChanged()

            addPageAction = QAction("Add page", None)
            addPageAction.triggered.connect(newPage)
            addSymAction = QAction("Add symbol", None)
            addSymAction.triggered.connect(newSym)
            rmPageAction = QAction("Delete page", None)
            menu.addAction(addPageAction)
            menu.addAction(addSymAction)
            menu.addSeparator()
            if obj.page is not None:
                menu.addAction(rmPageAction)
        menu.exec(self.ui.projectTree.mapToGlobal(pt))
Exemple #4
0
class ProjectDock(QDockWidget):
    openPage = pyqtSignal('QObject')

    class PageElement(object):
        def __init__(self, page):
            self.name = page.name
            self.page = page
            self.subelements = []

        @property
        def doc(self):
            return self.page.parentDoc

    class SymElement(object):
        def __init__(self, sym):
            self.name = "Symbol: " + sym.name
            self.page = sym
            self.subelements = []

        @property
        def doc(self):
            return self.page.parentDoc

    class DocElement(QObject):
        def __init__(self, doc: MasterDocument):
            super().__init__()
            self._doc = doc
            doc.sigChanged.connect(self.onDocChanged)
            self.name = None
            self.subelements = []
            self.onDocChanged()

        @property
        def doc(self):
            return self._doc

        @property
        def page(self):
            return None

        @pyqtSlot()
        def onDocChanged(self):
            self.name = self._doc.name()
            self.subelements = []
            for s in self._doc.symbols:
                self.subelements.append(ProjectDock.SymElement(s))
            for p in self._doc.pages:
                self.subelements.append(ProjectDock.PageElement(p))

    class Node(AbstractTreeNode):
        def __init__(self, ref, parent, row):
            self.ref = ref
            super().__init__(parent, row)

        def _getChildren(self):
            return [
                ProjectDock.Node(elem, self, index)
                for index, elem in enumerate(self.ref.subelements)
            ]

    class Model(AbstractTreeModel):
        def __init__(self, rootElements):
            self.rootElements = rootElements
            super().__init__()

        def _getRootNodes(self):
            return [
                ProjectDock.Node(elem, None, index)
                for index, elem in enumerate(self.rootElements)
            ]

        def columnCount(self, parent):
            return 1

        def data(self, index, role):
            if not index.isValid():
                return None
            node = index.internalPointer()
            if role == Qt.DisplayRole:
                return node.ref.name
            if role == Qt.UserRole:
                return node.ref
            return None

    def __init__(self, window, parent=None):
        super().__init__(parent)
        self.window = window
        self.ui = Ui_ProjectDock()
        self.ui.setupUi(self)
        self.ui.projectTree.setContextMenuPolicy(Qt.CustomContextMenu)
        self.ui.projectTree.customContextMenuRequested.connect(
            self.onContextMenu)
        self.onDocsChanged()

    @pyqtSlot()
    def onDocsChanged(self):
        self.ui.projectTree.setModel(
            ProjectDock.Model(
                [ProjectDock.DocElement(d) for d in self.window.docs]))
        self.ui.projectTree.expandAll()

    @pyqtSlot(QModelIndex)
    def on_projectTree_doubleClicked(self, idx: QModelIndex):
        p = idx.data(Qt.UserRole).page
        if p is not None:
            self.openPage.emit(p)

    @pyqtSlot(QPoint)
    def onContextMenu(self, pt: QPoint):
        index = self.ui.projectTree.indexAt(pt)
        newDocAction = QAction("New document", None)
        newDocAction.triggered.connect(self.window.on_actionNew_triggered)
        menu = QMenu()
        menu.addAction(newDocAction)
        if index.isValid():
            obj = index.data(Qt.UserRole)

            def newPage():
                obj.doc.appendNewPage()
                self.onDocsChanged()

            def newSym():
                obj.doc.appendNewSymbol()
                self.onDocsChanged()

            addPageAction = QAction("Add page", None)
            addPageAction.triggered.connect(newPage)
            addSymAction = QAction("Add symbol", None)
            addSymAction.triggered.connect(newSym)
            rmPageAction = QAction("Delete page", None)
            menu.addAction(addPageAction)
            menu.addAction(addSymAction)
            menu.addSeparator()
            if obj.page is not None:
                menu.addAction(rmPageAction)
        menu.exec(self.ui.projectTree.mapToGlobal(pt))