Initial Commit
This commit is contained in:
parent
c7bb77db7d
commit
bdb2d54159
|
@ -0,0 +1,146 @@
|
||||||
|
import argparse
|
||||||
|
from docxtpl import DocxTemplate, RichText
|
||||||
|
import markdown
|
||||||
|
from novel_stats.novel_stats import count_words
|
||||||
|
import tempfile
|
||||||
|
import MarkdownPP
|
||||||
|
import json
|
||||||
|
|
||||||
|
TITLE_MARKER = '# '
|
||||||
|
AUTHOR_MARKER = '### '
|
||||||
|
CHAPTER_MARKER = '## '
|
||||||
|
STATUS_MARKER = '[status]: # '
|
||||||
|
ACT_MARKER = '[act]: # '
|
||||||
|
COMMENT_MARKER = '[//]: # ' # Strandard markdown comment marker, supported by pandoc and calibre's ebook-convert
|
||||||
|
|
||||||
|
class Chapter:
|
||||||
|
def __init__(self, heading):
|
||||||
|
self.heading = heading
|
||||||
|
self.paragraphs = []
|
||||||
|
|
||||||
|
def md_re_parser(md_paragraph):
|
||||||
|
# Correct xml tags
|
||||||
|
pre = '<w:r><w:t xml:space="preserve">'
|
||||||
|
post = '</w:t></w:r>'
|
||||||
|
it_pre = '<w:r><w:rPr><w:i/></w:rPr><w:t xml:space="preserve">'
|
||||||
|
bf_pre = '<w:r><w:rPr><w:b/></w:rPr><w:t xml:space="preserve">'
|
||||||
|
bfit_pre = '<w:r><w:rPr><w:b/><w:i/></w:rPr><w:t xml:space="preserve">'
|
||||||
|
|
||||||
|
# Tag replacement
|
||||||
|
html = markdown.markdown(md_paragraph)
|
||||||
|
html = html.replace('<p>', pre)
|
||||||
|
html = html.replace('</p>', post)
|
||||||
|
html = html.replace('<strong><em>', post+bfit_pre)
|
||||||
|
html = html.replace('</strong></em>', post+pre)
|
||||||
|
html = html.replace('<em>', post+it_pre)
|
||||||
|
html = html.replace('</em>', post+pre)
|
||||||
|
html = html.replace('<strong>', post+bf_pre)
|
||||||
|
html = html.replace('</strong>', post+pre)
|
||||||
|
|
||||||
|
# xml cleanup
|
||||||
|
while pre+post in html:
|
||||||
|
html = html.replace(pre+post,'')
|
||||||
|
|
||||||
|
# convert to a rich text paragraph
|
||||||
|
par = RichText()
|
||||||
|
par.xml = html
|
||||||
|
if len(html) == 0:
|
||||||
|
print(md_paragraph)
|
||||||
|
return par
|
||||||
|
|
||||||
|
def novel_parser(source_file, context = None):
|
||||||
|
if not context:
|
||||||
|
context = {'author_address': 'Street\nTown, State ZIP\nCountry',
|
||||||
|
'author_email': 'name@email.com',
|
||||||
|
'author_phone': 'PhoneNumber(s)',
|
||||||
|
'author_website': 'https://www.author.com'}
|
||||||
|
context['chapters'] = []
|
||||||
|
|
||||||
|
wc = 0
|
||||||
|
chapter = Chapter('')
|
||||||
|
for line in source_file:
|
||||||
|
if line.startswith(TITLE_MARKER):
|
||||||
|
title = line[len(TITLE_MARKER):].strip('()\n')
|
||||||
|
context['project_title'] = title
|
||||||
|
wc += count_words(title)
|
||||||
|
elif line.startswith(AUTHOR_MARKER):
|
||||||
|
author_name = line[len(AUTHOR_MARKER):].strip('()\n')
|
||||||
|
if 'author_name' not in context:
|
||||||
|
context['author_name'] = author_name
|
||||||
|
context['penname'] = author_name
|
||||||
|
elif line.startswith(CHAPTER_MARKER):
|
||||||
|
if chapter.heading or chapter.paragraphs:
|
||||||
|
context['chapters'].append(chapter)
|
||||||
|
chapter = Chapter(line[len(CHAPTER_MARKER):].strip('()\n'))
|
||||||
|
wc += count_words(chapter.heading)
|
||||||
|
elif line.startswith(STATUS_MARKER) or line.startswith(ACT_MARKER) or line.startswith(COMMENT_MARKER):
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
stripped = line.strip()
|
||||||
|
if stripped:
|
||||||
|
wc += count_words(stripped)
|
||||||
|
chapter.paragraphs.append(md_re_parser(stripped))
|
||||||
|
context['chapters'].append(chapter)
|
||||||
|
|
||||||
|
source_file.close()
|
||||||
|
|
||||||
|
context['wc_1000'] = f'{(wc//1000)*1000:,}'
|
||||||
|
|
||||||
|
if 'header' not in context:
|
||||||
|
context['header'] = context['author_name'].split()[-1] + ' - ' + context['project_title'] + ' - '
|
||||||
|
|
||||||
|
return context
|
||||||
|
|
||||||
|
def main():
|
||||||
|
parser = argparse.ArgumentParser()
|
||||||
|
parser.add_argument(
|
||||||
|
'markdown_file',
|
||||||
|
type=argparse.FileType('r'),
|
||||||
|
help='The markdown file for the novel, main file if a multi-file novel',
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
'-pp',
|
||||||
|
action='store_true',
|
||||||
|
help='run markdown pre-processor, this allows for a multi-file input (e.g. each chapter in its own file), but requires the MarkdownPP python library',
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
'--settings',
|
||||||
|
'-s',
|
||||||
|
type=argparse.FileType('r'),
|
||||||
|
help='setting json file',
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
'--template',
|
||||||
|
'-t',
|
||||||
|
type=argparse.FileType('r'),
|
||||||
|
help='template docx file',
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
'--output',
|
||||||
|
'-o',
|
||||||
|
type=argparse.FileType('w'),
|
||||||
|
help='output docx file',
|
||||||
|
)
|
||||||
|
arguments = parser.parse_args()
|
||||||
|
|
||||||
|
arguments.template.close()
|
||||||
|
doc = DocxTemplate(arguments.template.name)
|
||||||
|
if arguments.pp:
|
||||||
|
mdfile = tempfile.TemporaryFile(mode='w+')
|
||||||
|
MarkdownPP.MarkdownPP(
|
||||||
|
input=arguments.markdown_file, output=mdfile, modules=list(MarkdownPP.modules)
|
||||||
|
)
|
||||||
|
mdfile.seek(0)
|
||||||
|
else:
|
||||||
|
mdfile = arguments.markdown_file
|
||||||
|
if arguments.settings:
|
||||||
|
context = json.load(arguments.settings)
|
||||||
|
else:
|
||||||
|
context = None
|
||||||
|
context = novel_parser(mdfile, context = context)
|
||||||
|
doc.render(context)
|
||||||
|
arguments.output.close()
|
||||||
|
doc.save(arguments.output.name)
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
main()
|
|
@ -0,0 +1,6 @@
|
||||||
|
{
|
||||||
|
"author_address": "Address",
|
||||||
|
"author_email": "Email",
|
||||||
|
"author_phone": "Phone number",
|
||||||
|
"author_website": "Website"
|
||||||
|
}
|
Binary file not shown.
Loading…
Reference in New Issue