NovelEditor.HighlightProcessing

Complexity

Total Complexity

13

Size/Duplication

Total Lines	152
Duplicated Lines	0 %

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	HighlightProcessingClass.highlight()	0	31	3
A	HighlightProcessingClass.line_highlight()	0	23	3
A	HighlightProcessingClass.all_highlight()	0	13	2
A	HighlightProcessingClass.__init__()	0	26	1
A	HighlightProcessingClass.create_tags()	0	33	4

#!/usr/bin/env python3
import re

from janome.tokenizer import Tokenizer

from . import ProcessingMenu
from . import FindProcessing
from . import Definition


class HighlightProcessingClass(Definition.DefinitionClass):
    """ハイライトのクラス.

    ・ハイライトするためのプログラム群

    Args:
        app (instance): MainProcessingClass のインスタンス
        tokenizer (instance): Tokenizer のインスタンス
        locale_var (str): ロケーション
        master (instance): toplevel のインスタンス
    """
    def __init__(self, app, tokenizer, locale_var, master=None):
        super().__init__(locale_var, master)
        self.COLOR = [
            'sky blue',
            'yellow green',
            'gold',
            'salmon',
            'orange',
            'red',
            'hot pink',
            'dark orchid',
            'purple',
            'midnight blue',
            'light slate blue',
            'dodger blue',
            'dark turquoise',
            'cadet blue',
            'maroon',
            'tan1',
            'rosy brown',
            'indian red',
            'orange red',
            'violet red'
        ]
        self.app = app
        self.TOKENIZER = tokenizer

    def all_highlight(self):
        """全てハイライト.

        ・開く処理等の時にすべての行をハイライトする。
        """
        # 全てのテキストを取得
        src = self.app.NovelEditor.get('1.0', 'end - 1c')
        # 全てのハイライトを一度解除する
        for tag in self.app.NovelEditor.tag_names():
            self.app.NovelEditor.tag_remove(tag, '1.0', 'end')

        # ハイライトする
        self.highlight('1.0', src, 'end')

    def line_highlight(self):
        """現在行だけハイライト.

        ・入力等の変更時に現在の行のみをハイライトする。
        """
        start = 'insert linestart'
        end = 'insert lineend'
        # 現在行のテキストを取得
        src = self.app.NovelEditor.get(start, end)
        # その行のハイライトを一度解除する
        for tag in self.app.NovelEditor.tag_names():
            self.app.NovelEditor.tag_remove(tag, start, end)

        # ハイライトする
        self.highlight(start, src, end)
        # 置換処理時に選択する
        if FindProcessing.FindProcessingClass.replacement_check:
            start = FindProcessing.FindProcessingClass.next_pos
            end = '{0} + {1}c'.format(
                start,
                len(FindProcessing.FindProcessingClass.find_text)
            )
            self.app.NovelEditor.tag_add('sel', start, end)

    def highlight(self, start, src, end):
        """ハイライトの共通処理.

        ・ハイライトする文字が見つかったらハイライト処理をする。
        先頭の文字が全角スペースならば、一文字ずらしてハイライトする。

        Args:
            start (str): はじめの位置 例.(1.0)
            src (str): ハイライトする文章
            end (str): 終わりの位置
        """
        self.create_tags()
        self.app.NovelEditor.mark_set('range_start', start)
        space_count = re.match(r"\u3000*", self.app.NovelEditor.get(start, end)).end()
        # 形態素解析を行う
        for token in self.t.tokenize(src):
            content = token.surface
            self.app.NovelEditor.mark_set(
                'range_end', 'range_start+{0}c'
                .format(len(content))
            )
            # 全角スペースの時はずらす
            if space_count > 0:
                self.app.NovelEditor.tag_add(
                    content,
                    'range_start+{0}c'.format(space_count),
                    'range_end+{0}c'.format(space_count)
                )
            else:
                self.app.NovelEditor.tag_add(content, 'range_start', 'range_end')
            self.app.NovelEditor.mark_set('range_start', 'range_end')

    def create_tags(self):
        """タグの作成.

        ・キャラクターの名前をJanomeの形態素解析にかかるようにする。
        キャラクターの名前を色付きにする。
        """
        i = 0
        system_dic = u"喜寛,固有名詞,ヨシヒロ"
        # キャラクターから一覧を作る。
        children = self.app.tree.get_children('data/character')
        for child in children:
            # ユーザー定義辞書の設定
            reading = ""
            childname = self.app.tree.item(child, "text")
            for token in self.TOKENIZER.tokenize(childname):
                reading += token.phonetic
            system_dic += u"\n{0},固有名詞,{1}".format(childname, reading)
            # タグの作成
            self.app.NovelEditor.tag_configure(
                childname,
                foreground=self.COLOR[i % len(self.COLOR)],
                font=(self.app.font, ProcessingMenu.ProcessingMenuClass.font_size, "bold")
            )
            i += 1

        with open("./userdic.csv", mode='w', encoding='utf-8') as f:
            f.write(system_dic)

        # Janomeを使って日本語の形態素解析
        self.t = Tokenizer(
            "./userdic.csv",
            udic_type="simpledic",
            udic_enc="utf8"
        )