forked from witten/novel-stats
Added mdpp support
This commit is contained in:
parent
dcfea89ec7
commit
93b3af2c3b
|
@ -3,6 +3,7 @@
|
|||
|
||||
import collections
|
||||
import sys
|
||||
import os
|
||||
|
||||
|
||||
CHAPTER_MARKER = '## '
|
||||
|
@ -26,6 +27,14 @@ def count_words(line):
|
|||
def main():
|
||||
arguments = sys.argv[1:]
|
||||
filename = arguments[0]
|
||||
tmpfilename = None
|
||||
|
||||
if '-pp' in arguments:
|
||||
import MarkdownPP
|
||||
tmpfilename = f'.tmp-{os.getpid}.md'
|
||||
MarkdownPP.MarkdownPP(input=open(filename), output=open(tmpfilename,'w'), modules=list(MarkdownPP.modules))
|
||||
filename = tmpfilename
|
||||
|
||||
chapter_heading = None
|
||||
act_heading = None
|
||||
total_word_count = 0
|
||||
|
@ -102,6 +111,9 @@ def main():
|
|||
|
||||
print('total: {:,} words'.format(total_word_count))
|
||||
|
||||
if tmpfilename:
|
||||
os.remove(tmpfilename)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
|
Loading…
Reference in New Issue