Veronica K. B. Olsen
84 lines
2.1 KiB
Python
84 lines
2.1 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""novelWriter GUI Document Word Counter
|
||
|
||
novelWriter – GUI Document Word Counter
|
||
===================================
|
||
A thread for counting words and characters in a document
|
||
|
||
File History:
|
||
Created: 2019-04-22 [0.0.1]
|
||
|
||
"""
|
||
|
||
import logging
|
||
import nw
|
||
|
||
from time import time
|
||
|
||
from PyQt5.QtCore import QThread
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
class WordCounter(QThread):
|
||
|
||
def __init__(self, theParent):
|
||
QThread.__init__(self, theParent)
|
||
self.theParent = theParent
|
||
self.charCount = 0
|
||
self.wordCount = 0
|
||
self.paraCount = 0
|
||
return
|
||
|
||
def run(self):
|
||
|
||
self.charCount = 0
|
||
self.wordCount = 0
|
||
self.paraCount = 0
|
||
|
||
prevEmpty = True
|
||
|
||
for n in range(self.theParent.theDoc.blockCount()):
|
||
|
||
theBlock = self.theParent.theDoc.findBlockByNumber(n)
|
||
if not theBlock.isValid():
|
||
continue
|
||
|
||
countPara = True
|
||
theText = theBlock.text()
|
||
theLen = len(theText)
|
||
|
||
if theLen == 0:
|
||
prevEmpty = True
|
||
continue
|
||
if theText[0] == "@" or theText[0] == "%":
|
||
prevEmpty = True
|
||
continue
|
||
|
||
if theText[0:5] == "#### ":
|
||
self.wordCount -= 1
|
||
self.charCount -= 5
|
||
countPara = False
|
||
elif theText[0:4] == "### ":
|
||
self.wordCount -= 1
|
||
self.charCount -= 4
|
||
countPara = False
|
||
elif theText[0:3] == "## ":
|
||
self.wordCount -= 1
|
||
self.charCount -= 3
|
||
countPara = False
|
||
elif theText[0:2] == "# ":
|
||
self.wordCount -= 1
|
||
self.charCount -= 2
|
||
countPara = False
|
||
|
||
theBuff = theText.replace("–"," ").replace("—"," ")
|
||
self.wordCount += len(theBuff.split())
|
||
self.charCount += theLen
|
||
if countPara and prevEmpty:
|
||
self.paraCount += 1
|
||
prevEmpty = countPara == False
|
||
|
||
pass
|
||
|
||
## END Class _WordCounter
|