12345678910111213141516171819202122232425262728293031323334353637383940414243 |
- import subprocess
- from pelican import signals
- from pelican.readers import BaseReader
- from pelican.utils import pelican_open
- class Txt2tagsReader(BaseReader):
- enabled = True
- file_extensions = ['t2t', 'txt2tags']
- 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
- t2t_cmd = [r"txt2tags", r"--encoding=utf-8", r"--target=html", r"--infile=-", r"--outfile=-"]
- proc = subprocess.Popen(t2t_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, t2t_cmd)
- return output, metadata
- def add_reader(readers):
- for ext in Txt2tagsReader.file_extensions:
- readers.reader_classes[ext] = Txt2tagsReader
- def register():
- signals.readers_init.connect(add_reader)
|