示例#1
0
    def __init__(self, parent=None):
        QtGui.QWidget.__init__(self, parent)
        self.setupUi()

        self.findForm = Find(self)

        self.treeWidget.setStyleSheet('''
            .QTreeWidget::item::selected {
                background-color: rgb(15,95,95);
                font: bold 24px;
                }
            ''')

        self.treeWidget.customContextMenuRequested.connect(self.contextMenuEvent)
        self.treeWidget.currentItemChanged.connect(self.showItemContent)

        self.textEdit.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
        self.textEdit.customContextMenuRequested.connect(self.textMenu)

        self.textEdit.setMarginWidth(1, 20)
        self.textEdit.setMarginsBackgroundColor(QtGui.QColor("#333333"))
        self.textEdit.setMarginsForegroundColor(QtGui.QColor("#CCCCCC"))
        self.textEdit.markerDefine(QsciScintilla.RightTriangle)
        self.textEdit.setMarkerBackgroundColor(QtGui.QColor("#ACF39D"))

        self.actionNew.triggered.connect(self.newDb)
        self.actionOpen.triggered.connect(self.openFile)

        self.treeAddNodeAction.triggered.connect(self.addParentItem)
        self.treeAddSubNodeAction.triggered.connect(self.addSubItem)

        self.textEdit.setFocusPolicy(QtCore.Qt.StrongFocus)
        self.textEdit.installEventFilter(self)

        self.editUndoAction.triggered.connect(self.txtUndo)
        self.editRedoAction.triggered.connect(self.txtRedo)

        self.searchAction.triggered.connect(self.findInNode)

        self.aboutAction.triggered.connect(self.about)

        self.wordWrapModeAction.triggered.connect(self.wordWrap)
        self.showLineNumbersAction.triggered.connect(self.showLineNumbers)

        self.fontBytes = c_char_p()
        self.fontBytes.value = b'Courier New'
        self.textEdit.SendScintilla(QsciScintilla.SCI_STYLESETSIZE, 0, 12)
        self.textEdit.SendScintilla(QsciScintilla.SCI_STYLESETFONT, self.fontBytes.value)
        self.textEdit.SendScintilla(QsciScintilla.SCI_SETCODEPAGE, 65001)
示例#2
0
 def findInNode(self):
     self.findForm = Find(self)
     self.findForm.show()
示例#3
0
class MyForm(QtGui.QMainWindow,  TakeNote):

    def __init__(self, parent=None):
        QtGui.QWidget.__init__(self, parent)
        self.setupUi()

        self.findForm = Find(self)

        self.treeWidget.setStyleSheet('''
            .QTreeWidget::item::selected {
                background-color: rgb(15,95,95);
                font: bold 24px;
                }
            ''')

        self.treeWidget.customContextMenuRequested.connect(self.contextMenuEvent)
        self.treeWidget.currentItemChanged.connect(self.showItemContent)

        self.textEdit.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
        self.textEdit.customContextMenuRequested.connect(self.textMenu)

        self.textEdit.setMarginWidth(1, 20)
        self.textEdit.setMarginsBackgroundColor(QtGui.QColor("#333333"))
        self.textEdit.setMarginsForegroundColor(QtGui.QColor("#CCCCCC"))
        self.textEdit.markerDefine(QsciScintilla.RightTriangle)
        self.textEdit.setMarkerBackgroundColor(QtGui.QColor("#ACF39D"))

        self.actionNew.triggered.connect(self.newDb)
        self.actionOpen.triggered.connect(self.openFile)

        self.treeAddNodeAction.triggered.connect(self.addParentItem)
        self.treeAddSubNodeAction.triggered.connect(self.addSubItem)

        self.textEdit.setFocusPolicy(QtCore.Qt.StrongFocus)
        self.textEdit.installEventFilter(self)

        self.editUndoAction.triggered.connect(self.txtUndo)
        self.editRedoAction.triggered.connect(self.txtRedo)

        self.searchAction.triggered.connect(self.findInNode)

        self.aboutAction.triggered.connect(self.about)

        self.wordWrapModeAction.triggered.connect(self.wordWrap)
        self.showLineNumbersAction.triggered.connect(self.showLineNumbers)

        self.fontBytes = c_char_p()
        self.fontBytes.value = b'Courier New'
        self.textEdit.SendScintilla(QsciScintilla.SCI_STYLESETSIZE, 0, 12)
        self.textEdit.SendScintilla(QsciScintilla.SCI_STYLESETFONT, self.fontBytes.value)
        self.textEdit.SendScintilla(QsciScintilla.SCI_SETCODEPAGE, 65001)

    ### SEARCH START ###
    def findInNode(self):
        self.findForm = Find(self)
        self.findForm.show()

    def searchScintilla(self, pcharSearchValue, searchStart, searchEnd):
        ###Clear styling start ###
        self.textEdit.SendScintilla(QsciScintilla.SCI_STARTSTYLING, 0, 0)
        self.textEdit.SendScintilla(QsciScintilla.SCI_SETSTYLING, len(self.textEdit.text()), 0)
        ###Clear styling end ###

        self.textEdit.SendScintilla(QsciScintilla.SCI_MARKERDELETEALL)
        cursorPosition = self.textEdit.getCursorPosition()
        res = 0
        while res != -1:
            self.textEdit.SendScintilla(QsciScintilla.SCI_SETTARGETSTART, res)
            self.textEdit.SendScintilla(QsciScintilla.SCI_SETTARGETEND, len(self.textEdit.text()))
            res = self.textEdit.SendScintilla(QsciScintilla.SCI_SEARCHINTARGET, len(pcharSearchValue.value), pcharSearchValue.value)

            if res != -1:
                self.textEdit.SendScintilla(QsciScintilla.SCI_SETSELECTIONSTART, res)
                self.textEdit.SendScintilla(QsciScintilla.SCI_SETSELECTIONEND, res+len(pcharSearchValue.value))
                self.textEdit.SendScintilla(QsciScintilla.SCI_MARKERADD, self.textEdit.lineIndexFromPosition(res)[0])
                res += len(pcharSearchValue.value)

                self.textEdit.SendScintilla(QsciScintilla.SCI_STYLESETSIZE, 12, 12)
                self.textEdit.SendScintilla(QsciScintilla.SCI_STYLESETFONT, 12, self.fontBytes.value)
                self.textEdit.SendScintilla(QsciScintilla.SCI_STYLESETBACK, 12, 2042550)
                self.textEdit.SendScintilla(QsciScintilla.SCI_STYLESETFORE, 12, 000)
                self.textEdit.SendScintilla(QsciScintilla.SCI_STARTSTYLING, res-len(pcharSearchValue.value), 12)
                self.textEdit.SendScintilla(QsciScintilla.SCI_SETSTYLING, len(pcharSearchValue.value), 12)

        self.textEdit.setCursorPosition(cursorPosition[0], cursorPosition[1])
        self.textEdit.SendScintilla(QsciScintilla.SCI_SETTARGETSTART, searchStart)
        self.textEdit.SendScintilla(QsciScintilla.SCI_SETTARGETEND, searchEnd)
        res = self.textEdit.SendScintilla(QsciScintilla.SCI_SEARCHINTARGET, len(pcharSearchValue.value), pcharSearchValue.value)

        if res != -1:
            self.textEdit.SendScintilla(QsciScintilla.SCI_SETSELECTIONSTART, res)
            self.textEdit.SendScintilla(QsciScintilla.SCI_SETSELECTIONEND, res+len(pcharSearchValue.value))

        else:
            print('no more')
    ### SEARCH STOP ###

     ### EXTENDING THE STANDART CONTEXT MENU START ###
    def textMenu(self, position):
        editMenu = self.textEdit.createStandardContextMenu()
        editMenu.addSeparator()
        upperAction = editMenu.addAction(self.tr("To UPPER Case"))
        lowerAction = editMenu.addAction(self.tr("To lower case"))
        textAction = editMenu.exec_(self.textEdit.viewport().mapToGlobal(position))
        if textAction == upperAction:
            self.upperCase()
        if textAction == lowerAction:
            self.lowerCase()
    ### EXTENDING THE STANDART CONTEXT MENU STOP ###

    ### UPPER/LOWER CASE START ###
    def upperCase(self):
        self.textEdit.SendScintilla(QsciScintillaBase.SCI_UPPERCASE)

    def lowerCase(self):
        self.textEdit.SendScintilla(QsciScintillaBase.SCI_LOWERCASE)
    ### UPPER/LOWER CASE STOP ###

    ### SET WORD WRAP START ###
    def wordWrap(self):
        if self.wordWrapModeAction.isChecked():
            self.textEdit.SendScintilla(QsciScintillaBase.SCI_SETWRAPMODE, 1)
        else:
            self.textEdit.SendScintilla(QsciScintillaBase.SCI_SETWRAPMODE, 0)
    ### SET WORD WRAP STOP ###

    ### SET SHOW LINE NUMBERS START ###
    def showLineNumbers(self):
        if self.showLineNumbersAction.isChecked():
            ## Margins colors
            # line numbers margin
            self.textEdit.setMarginWidth(1, 35)
            self.textEdit.setMarginLineNumbers(1, True)
        else:
            self.textEdit.setMarginWidth(1, 20)
            self.textEdit.setMarginLineNumbers(1, False)
    ### SET SHOW LINE NUMBERS STOP ###

    ### CREATING TREE WIDGET CONTEXT MENU START ###
    def contextMenuEvent(self, eventmenu):
        menu = QMenu(self)
        addNodeAction = menu.addAction("Add Node")
        addSubNodeAction = menu.addAction("Add Subnode")
        renameAction = menu.addAction("Rename")
        deleteAction = menu.addAction("Delete")
        action = menu.exec_(self.mapToGlobal(eventmenu.pos()))
        if action == addNodeAction:
            self.addParentItem()
        if action == addSubNodeAction:
            self.addSubItem()
        if action == renameAction:
            self.renameItem()
        if action == deleteAction:
            self.deleteItem()
    ### CREATING TREE WIDGET CONTEXT MENU STOP ###

    ### DB VARIABLES START ###
    def tables(self, sqliteUrl):
        db = create_engine(sqliteUrl, echo=False)
        self.metadata = MetaData(db)
        self.nodes = Table('node', self.metadata,
                            Column('node_id', Integer, unique=True),
                            Column('name', UnicodeText),
                            Column('txt', UnicodeText),
                            )
        self.children = Table('children', self.metadata,
                            Column('node_id', Integer, unique=True),
                            Column('name', UnicodeText),
                            Column('txt', UnicodeText),
                            Column('father_id', Integer),
                            )
    ### DB VARIABLES STOP ###

    ### CREATING NEW DATABASE START ###
    def newDb(self):
        files_types = "DB (*.db)"
        self.filename, filter = QtGui.QFileDialog.getSaveFileNameAndFilter(self, 'Save file', '.db', files_types)
        sqliteUrl = ('sqlite:///%s' % self.filename)
        self.tables(sqliteUrl)
        self.metadata.create_all()
        self.redraw()
        self.setWindowTitle('%s - TakeNote' % self.filename)
        parserVal = MainCfg()
        parserVal.writeCfg(sectionValue=self.filename)
    ### CREATING NEW DATABASE STOP ###

    ### OPEN DB START ###
    def openFile(self):
        files_types = "DB (*.db)"
        self.filename, filter = QtGui.QFileDialog.getOpenFileNameAndFilter(self, 'Save file', '.db', files_types)
        sqliteUrl = ('sqlite:///%s' % self.filename)
        self.tables(sqliteUrl)
        self.redraw()
        self.setWindowTitle('%s - TakeNote' % self.filename)
        parseVal = MainCfg()
        parseVal.writeCfgUpdateValue('MAIN', 'default_db', self.filename)
        self.treeWidget.setCurrentItem(self.treeWidget.topLevelItem(0))
        self.textEdit.hasFocus()
    ### OPEN DB STOP ###

    ### REDRAW TREE START ###
    def redraw(self):
        self.treeWidget.clear()
        s = select([self.nodes])
        result = s.execute()
        for row in result:
            self.parentItem = QTreeWidgetItem()
            self.treeWidget.addTopLevelItem(self.parentItem)
            self.parentItem.setText(0, str(row[1]))
            self.parentItem.setStatusTip(1, str(row[0]))
            self.parentItem.setFlags(QtCore.Qt.ItemIsSelectable|QtCore.Qt.ItemIsEnabled)
            s1 = select([self.children], self.children.c.father_id == row[0])
            result1 = s1.execute()
            for k in result1:
                self.childItem = QTreeWidgetItem()
                self.childItem = QTreeWidgetItem(self.parentItem)
                self.childItem.setText(0, k[1])
                self.childItem.setStatusTip(1, str(k[0]))
                self.childItem.setFlags(QtCore.Qt.ItemIsSelectable|QtCore.Qt.ItemIsEnabled)
                self.parentItem.setExpanded(0)
                self.indexOfChildItem = self.parentItem.indexOfChild(self.childItem)

    ### REDRAW TREE START ###

    ### SHOW ITEM CONTENT START ###
    def showItemContent(self):
        try:
            self.treeWidget.currentItem().parent().indexOfChild(self.treeWidget.currentItem())
        except:
            parentUniqueID = self.treeWidget.currentItem().statusTip(1)
            s1 = select([self.nodes.c.txt], self.nodes.c.node_id == parentUniqueID)
            currentParentItemContent = s1.execute()
            self.textEdit.setText(str(currentParentItemContent.fetchone()[0]))

            ### CURRENT LINE POS START ###
            for x in currentLinePos.keys():
                if self.treeWidget.currentItem().text(0) == x:
                    self.textEdit.setCursorPosition(currentLinePos[x], 0)
            ### CURRENT LINE POS END ###
        else:
            parentUniqueID2 = self.treeWidget.currentItem().parent().statusTip(1)
            childIndex = self.treeWidget.currentItem().statusTip(1)
            s = select([self.children.c.txt], self.children.c.node_id == childIndex and self.children.c.father_id == parentUniqueID2)
            currentChildItemContent = s.execute()
            self.textEdit.setText(str(currentChildItemContent.fetchone()[0]))

            ### CURRENT LINE POS START ###
            for x in currentLinePos.keys():
                if self.treeWidget.currentItem().text(0) == x:
                    self.textEdit.setCursorPosition(currentLinePos[x], 0)
            ### CURRENT LINE POS END ###

    ### SHOW ITEM CONTENT STOP ###

    ### EVENT FILTER START ###
    def eventFilter(self, obj, event):
        if event.type() == QtCore.QEvent.FocusOut:

            ### CURRENT LINE POS COLLECT START ###
            cursorPos = self.textEdit.getCursorPosition()
            currentLinePos[self.treeWidget.currentItem().text(0)] = cursorPos[0]
            ### CURRENT LINE POS COLLECT END ###

            try:
                self.treeWidget.currentItem().parent().indexOfChild(self.treeWidget.currentItem())
            except:
                parentUniqueID = self.treeWidget.currentItem().statusTip(1)
                upd = update(self.nodes).where(self.nodes.c.node_id == parentUniqueID).values(txt=self.textEdit.text())
                upd.execute()
            else:
                parentUniqueID2 = self.treeWidget.currentItem().parent().statusTip(1)
                childIndex = self.treeWidget.currentItem().statusTip(1)
                upd1 = update(self.children).where(self.children.c.node_id == childIndex and self.children.c.father_id == parentUniqueID2).values(txt=self.textEdit.text())
                upd1.execute()
            return True
        return False
    ### EVENT FILTER STOP ###

    ### DEFINING NODE/SUBNODE NAME START ###
    def nodeName(self, textToShow):
        text, ok = QtGui.QInputDialog.getText(self, 'Name ', 'Node Name: ', QtGui.QLineEdit.Normal, textToShow)
        if ok and text != '':
                self.inputNodeName = text
        return self.inputNodeName
    ### DEFINING NODE/SUBNODE NAME STOP ###

    ### ADD PARENT ITEM START ###
    def addParentItem(self):
        self.nodeName(textToShow='')
        self.label = self.inputNodeName
        maxIdTmp = select([func.max(self.nodes.c.node_id)])
        run = maxIdTmp.execute()
        fetchTmp = run.fetchone()[0]
        if type(fetchTmp) is int:
            maxId = fetchTmp+1
        else:
            maxId = 0
        self.nodes.insert().execute(name=self.label, node_id=maxId, txt="")
        self.redraw()
        self.treeWidget.setCurrentItem(self.parentItem)
        self.textEdit.clear()
        self.textEdit.setFocus()
        del self.inputNodeName
    ### ADD PARENT ITEM STOP ###

    ### ADD SUB ITEM START ###
    def addSubItem(self):
        if self.treeWidget.currentItem().parent() is None:
            parentId = self.treeWidget.currentItem().statusTip(1)
            indexOfSelectedParent = self.treeWidget.indexFromItem(self.treeWidget.currentItem()).row()
        else:
            parentId = self.treeWidget.currentItem().parent().statusTip(1)
            indexOfSelectedParent = self.treeWidget.indexFromItem(self.treeWidget.currentItem().parent()).row()

        self.nodeName(textToShow='')
        self.label = self.inputNodeName
        maxIdTmp = select([func.max(self.children.c.node_id)])
        run = maxIdTmp.execute()
        fetchTmp = run.fetchone()[0]
        if type(fetchTmp) is int:
            maxId = fetchTmp+1
        else:
            maxId = 0
        self.children.insert().execute(name=self.label, node_id=maxId, txt="", father_id=parentId)

        model = self.treeWidget.model()
        self.redraw()
        indexItem = model.index(indexOfSelectedParent, 0)
        childNumber = self.treeWidget.itemFromIndex(indexItem).childCount()
        self.treeWidget.setCurrentItem(self.treeWidget.itemFromIndex(indexItem).child(childNumber-1))
        self.textEdit.setFocus()
        self.textEdit.clear()
        del self.inputNodeName
    ### ADD SUB ITEM STOP ###

    ### RENAME ITEM START ###
    def renameItem(self):
        self.nodeName(textToShow=self.treeWidget.currentItem().text(0))
        try:
            self.treeWidget.currentItem().parent().indexOfChild(self.treeWidget.currentItem())
        except:
            parentUniqueID = self.treeWidget.currentItem().statusTip(1)
            renNode = update(self.nodes).where(self.nodes.c.node_id == parentUniqueID).values(name=self.inputNodeName)
            renNode.execute()
            self.redraw()
        else:
            parentUniqueID = self.treeWidget.currentItem().parent().statusTip(0)
            childIndex = self.treeWidget.currentItem().statusTip(1)
            renSubNode = update(self.children).where(self.children.c.node_id == childIndex and self.children.c.father_id == parentUniqueID).values(name=self.inputNodeName)
            renSubNode.execute()
            self.redraw()
        del self.inputNodeName
    ### RENAME ITEM STOP ###

    ### DELETE ITEM START ###
    def deleteItem(self):
        try:
            self.treeWidget.currentItem().parent().indexOfChild(self.treeWidget.currentItem())
            parentIndexOfDeletedSubitem = self.treeWidget.indexFromItem(self.treeWidget.currentItem().parent()).row()
            deletedSubitemModel = self.treeWidget.model()
        except:
            parentUniqueID = self.treeWidget.currentItem().statusTip(1)
            delNode1 = self.nodes.delete().where(self.nodes.c.node_id == parentUniqueID)
            delNode2 = self.children.delete().where(self.children.c.father_id == parentUniqueID)
            delNode1.execute()
            delNode2.execute()
            self.textEdit.clear()
            self.redraw()
            self.treeWidget.setCurrentItem(self.treeWidget.topLevelItem(0))
        else:
            parentUniqueID = self.treeWidget.currentItem().parent().statusTip(1)
            childIndex = self.treeWidget.currentItem().statusTip(1)
            delSub = self.children.delete().where(self.children.c.node_id == childIndex and self.children.c.father_id == parentUniqueID)
            delSub.execute()
            self.textEdit.clear()
            self.redraw()
            indexDeletedSubitem = deletedSubitemModel.index(parentIndexOfDeletedSubitem, 0)
            self.treeWidget.setCurrentItem(self.treeWidget.itemFromIndex(indexDeletedSubitem))
            self.treeWidget.expand(indexDeletedSubitem)
    ### DELETE ITEM STOP ###

    ### UNDO REDO START ###
    def txtUndo(self):
        self.textEdit.undo()
    def txtRedo(self):
        self.textEdit.redo()
    ### UNDO REDO STOP ###

    ### ABOUT START ###
    def about(self):
        QMessageBox.about(self, self.tr("About TakeNote"),
            self.tr("TakeNote\n\n"
                    "%s\n"
                    "%s\n"
                    "%s" % (__author__, __version__, __date__)))
    ### ABOUT END ###

    def closeEvent(self, event):
        parseVal = MainCfg()
        sizeHeight = self.size().height()
        sizeWidth = self.size().width()
        posX = self.pos().x()
        posY = self.pos().y()
        parseVal.writeCfgUpdateValue('MAIN', 'geometry', '%s, %s, %s, %s' % (posX, posY, sizeWidth, sizeHeight))