|
@@ -0,0 +1,42 @@
|
|
|
+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)
|