Added mdpp support

novel_stats/novel_stats.py

@@ -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
@@ -101,6 +110,9 @@ def main():
         print('{}: {:,} words (~{}%)'.format(status, status_word_count, status_word_count * 100 // total_word_count))
 
     print('total: {:,} words'.format(total_word_count))
+ 
+    if tmpfilename:
+        os.remove(tmpfilename)
 
 
 if __name__ == '__main__':