forked from varia/varia.website
many many many Varia's websites, work in progress: https://many.vvvvvvaria.org
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
1.3 KiB
43 lines
1.3 KiB
7 years ago
|
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=-", r"--no-headers"]
|
||
|
|
||
|
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)
|