deroshkin
/
novel-stats-fork
forked from witten/novel-stats
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add title to the output

This commit is contained in:
Dmytro Yeroshkin 2021-11-01 10:01:31 +01:00
parent e2935dbddf
commit 1c53c94b9c
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
novel_stats/novel_stats.py
View File

@ -10,6 +10,7 @@ STATUS_MARKER = '[status]: # '
ACT_MARKER = '[act]: # ' ACT_MARKER = '[act]: # '
# Standard markdown comment marker, supported by Pandoc and Calibre's ebook-convert. # Standard markdown comment marker, supported by Pandoc and Calibre's ebook-convert.
COMMENT_MARKER = '[//]: # ' COMMENT_MARKER = '[//]: # '
TITLE_MARKER = '# '
def count_words(line): def count_words(line):
@ -74,6 +75,7 @@ def main():
word_count_by_act = collections.defaultdict(int) word_count_by_act = collections.defaultdict(int)
status_by_chapter = {} status_by_chapter = {}
current_status = None current_status = None
title = None
for line in mdfile.readlines(): for line in mdfile.readlines():
if line.startswith(CHAPTER_MARKER): if line.startswith(CHAPTER_MARKER):
@ -97,6 +99,10 @@ def main():
elif line.startswith(ACT_MARKER): elif line.startswith(ACT_MARKER):
act_heading = line[len(ACT_MARKER) :].strip('()\n') act_heading = line[len(ACT_MARKER) :].strip('()\n')
word_count_by_act[act_heading] = count_words(act_heading) word_count_by_act[act_heading] = count_words(act_heading)
elif line.startswith(TITLE_MARKER):
title = line[len(TITLE_MARKER):].strip()
line_word_count = count_words(line)
word_count_by_chapter[chapter_heading] += line_word_count
elif line.startswith(COMMENT_MARKER): # Don't count the words in a comment. elif line.startswith(COMMENT_MARKER): # Don't count the words in a comment.
pass pass
else: else:
@ -113,6 +119,9 @@ def main():
word_count_by_act[act_heading] += word_count_by_chapter[chapter_heading] word_count_by_act[act_heading] += word_count_by_chapter[chapter_heading]
total_word_count += word_count_by_chapter[chapter_heading] total_word_count += word_count_by_chapter[chapter_heading]
if title:
print(f'Novel Stats for {title.upper()}')
# -c or --chapter to give a chapter-by-chapter word count summary. # -c or --chapter to give a chapter-by-chapter word count summary.
if arguments.chapter: if arguments.chapter:
for chapter_heading, chapter_word_count in word_count_by_chapter.items(): for chapter_heading, chapter_word_count in word_count_by_chapter.items():