40d157895c
* Add PANDOC_EXTENSIONS configuration variable, allowing one to enable or disable Pandoc's markdown extensions individually. * Remove dependency on pypandoc. * Don't change the working directory. * More efficient metadata extraction.
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
import subprocess
|
|
|
|
from pelican import signals
|
|
from pelican.readers import BaseReader
|
|
from pelican.utils import pelican_open
|
|
|
|
class PandocReader(BaseReader):
|
|
enabled = True
|
|
file_extensions = ['md', 'markdown', 'mkd', 'mdown']
|
|
|
|
def read(self, filename):
|
|
with pelican_open(filename) as fp:
|
|
text = list(fp.splitlines())
|
|
|
|
metadata = {}
|
|
for i, line in enumerate(text):
|
|
kv = line.split(':', 1)
|
|
if len(kv) == 2:
|
|
name, value = kv[0].lower(), kv[1].strip()
|
|
metadata[name] = self.process_metadata(name, value)
|
|
else:
|
|
content = "\n".join(text[i:])
|
|
break
|
|
|
|
extra_args = self.settings.get('PANDOC_ARGS', [])
|
|
extensions = self.settings.get('PANDOC_EXTENSIONS', '')
|
|
if isinstance(extensions, list):
|
|
extensions = ''.join(extensions)
|
|
|
|
pandoc_cmd = ["pandoc", "--from=markdown" + extensions, "--to=html5"]
|
|
pandoc_cmd.extend(extra_args)
|
|
|
|
proc = subprocess.Popen(pandoc_cmd,
|
|
stdin = subprocess.PIPE,
|
|
stdout = subprocess.PIPE)
|
|
|
|
output = proc.communicate(content.encode('utf-8'))[0].decode('utf-8')
|
|
status = proc.wait()
|
|
if status:
|
|
raise subprocess.CalledProcessError(status, pandoc_cmd)
|
|
|
|
return output, metadata
|
|
|
|
def add_reader(readers):
|
|
readers.reader_classes['md'] = PandocReader
|
|
|
|
def register():
|
|
signals.readers_init.connect(add_reader)
|