TextFormat.py

import re

from PyQt5.QtCore import Qt
from PyQt5.QtGui import QTextCharFormat
from PyQt5.QtWidgets import QMessageBox

from functools import partial

class TextFormat(QTextCharFormat):
    def __init__(self, name, type):
        super().__init__()
        self.name = name
        self.wordList = []
        self.regexList = []
        self.type = type  # level or pos

        self.counter = 0
        self.counterDict = {}

        # Tab
        self.tabHBox = None
        self.nameLabel = None
        self.editButton = None
        self.colorButton = None
        self.listButton = None
        self.ruleButton = None
        self.removeButton = None
        # Counter
        self.counterNameLabel = None
        self.counterHBox = None

    def setupWordList(self, listPath, lines=None):
        self.wordList = []

        if lines:
            lines = lines
        else:
            with open(listPath, encoding='utf-8') as f:
                lines = f.readlines()

        for line in lines:
            word = line.rstrip('\r\n')
            if word:
                # add a tsek where missing
                if not word.endswith('་'):
                    word += '་'
                self.wordList.append(word)

    def setupRegexList(self, regexPath, lines=None):
        self.regexList = []

        if lines:
            lines = lines
        else:
            try:
                with open(regexPath, encoding='utf-8') as f:
                    lines = f.readlines()
            except UnicodeDecodeError as e:
                self.msg = QMessageBox()
                self.msg.setIcon(QMessageBox.Warning)
                self.msg.setText("The rule file should be UTF-8 encoding.")
                self.msg.show()
                return False


        for i, line in enumerate(lines):
            regex = line.rstrip('\r\n')
            if regex:
                try:
                    re.compile(regex)
                    self.regexList.append(regex)
                except Exception as e:
                    self.msg = QMessageBox()
                    self.msg.setIcon(QMessageBox.Warning)
                    self.msg.setText("Invalid regex in line: {}\n{}".format(
                        i + 1, str(e)))
                    self.msg.show()
                    return False
        else:
            return True


    def getColorRgbaCss(self):
        return 'rgba({}, {}, {})'.format(
            *(self.foreground().color().getRgb()))

    def getColorRgba(self):
        return self.foreground().color().getRgb()

    def setColor(self, qtColor):
        self.setForeground(qtColor)

    def getColor(self):
        return self.foreground().color()


class TextFormatManager:
    def __init__(self, parent):
        self.parent = parent
        self._formats = []

        self._defaultTextFormat = TextFormat('default', 'level')
        self._partOfSpeechFormat = TextFormat('partOfSpeech', 'level')
        self._partOfSpeechFormat.setColor(Qt.lightGray)

    def insert(self, textFormat):
        self._formats.append(textFormat)
        self.parent.tabWidget.addTextFormat(textFormat)
        self.parent.counterWidget.addTextFormat(textFormat)

    def remove(self, textFormat):
        self.parent.tabWidget.removeTextFormat(textFormat)
        self.parent.counterWidget.removeTextFormat(textFormat)
        self._formats.remove(textFormat)
        self.parent.highlightViewpoint()

    def clear(self):
        for textFormat in self._formats:
            self.parent.tabWidget.removeTextFormat(textFormat)
            self.parent.counterWidget.removeTextFormat(textFormat)
        self._formats = []

    def getFormats(self):
        return self._formats

    def getWordFormatDict(self):
        wordFormatDict = {}
        for textFormat in self._formats:
            for word in textFormat.wordList:
                wordFormatDict[word] = textFormat

        return wordFormatDict

    def getDefaultFormat(self):
        return self._defaultTextFormat

    def getPartOfSpeechFormat(self):
        return self._partOfSpeechFormat