2
0
mirror of https://github.com/lnbook/lnbook synced 2024-11-15 00:15:05 +00:00
lnbook/tools/readme_wordcount.py

42 lines
1.1 KiB
Python
Raw Normal View History

2021-07-20 13:12:20 +00:00
#!/usr/bin/env python3
from __future__ import print_function
import glob
import re
markup_files = glob.glob('*.asciidoc')
wordcount = {}
wordcount_sum = 0
for markup_file in markup_files:
markup_f = open(markup_file, 'r')
markup_contents = markup_f.read()
markup_f.close()
wc = len(markup_contents.strip().split())
wordcount_sum += wc
wordcount[markup_file] = wc
print(wc, "\t", markup_file)
print(wordcount_sum)
readme_f = open('README.md','r')
readme = readme_f.read()
readme_f.close()
wc_tag_re = re.compile("\| +(\[.*\])\((.*asciidoc)\) +\| +[\#]+ +\|(.*)$")
wc_total_re = re.compile("Total Word Count:")
readme_f = open('README.md','w')
for line in readme.splitlines():
match_re = wc_tag_re.match(line)
match_total = wc_total_re.match(line)
if match_re:
if match_re.group(2) in wordcount:
wordcount_bar = "#" * ((wordcount[match_re.group(2)]//500) + 1)
line = match_re.expand("| \g<1>(\g<2>) | " + wordcount_bar + " |\g<3>")
if match_total:
line = f"Total Word Count: {wordcount_sum}"
readme_f.write(line+"\n")
print(line)
readme_f.close()