mirror of
https://github.com/jrnl-org/jrnl.git
synced 2025-05-10 16:48:31 +02:00
9d8d6a83ae
fstring wip Run pyupgrade fix broken pyupgrade fstring run pyupgrade on plugin dir fixup! remove py2 remnants and use mocks in tests small print bugfix The file=sys.stderr was part of the format(), so an error got printed to stdout Drop use of codecs package Use builtins.open() instead fixup! remove py2 remnants and use mocks in tests
63 lines
2.5 KiB
Python
63 lines
2.5 KiB
Python
#!/usr/bin/env python
|
|
|
|
from .util import ERROR_COLOR, RESET_COLOR
|
|
from .util import slugify
|
|
from .plugins.template import Template
|
|
import os
|
|
|
|
|
|
class Exporter:
|
|
"""This Exporter can convert entries and journals into text files."""
|
|
def __init__(self, format):
|
|
with open("jrnl/templates/" + format + ".template") as f:
|
|
front_matter, body = f.read().strip("-\n").split("---", 2)
|
|
self.template = Template(body)
|
|
|
|
def export_entry(self, entry):
|
|
"""Returns a string representation of a single entry."""
|
|
return str(entry)
|
|
|
|
def _get_vars(self, journal):
|
|
return {
|
|
'journal': journal,
|
|
'entries': journal.entries,
|
|
'tags': journal.tags
|
|
}
|
|
|
|
def export_journal(self, journal):
|
|
"""Returns a string representation of an entire journal."""
|
|
return self.template.render_block("journal", **self._get_vars(journal))
|
|
|
|
def write_file(self, journal, path):
|
|
"""Exports a journal into a single file."""
|
|
try:
|
|
with open(path, "w", encoding="utf-8") as f:
|
|
f.write(self.export_journal(journal))
|
|
return f"[Journal exported to {path}]"
|
|
except OSError as e:
|
|
return f"[{ERROR_COLOR}ERROR{RESET_COLOR}: {e.filename} {e.strerror}]"
|
|
|
|
def make_filename(self, entry):
|
|
return entry.date.strftime("%Y-%m-%d_{}.{}".format(slugify(entry.title), self.extension))
|
|
|
|
def write_files(self, journal, path):
|
|
"""Exports a journal into individual files for each entry."""
|
|
for entry in journal.entries:
|
|
try:
|
|
full_path = os.path.join(path, self.make_filename(entry))
|
|
with open(full_path, "w", encoding="utf-8") as f:
|
|
f.write(self.export_entry(entry))
|
|
except OSError as e:
|
|
return f"[{ERROR_COLOR}ERROR{RESET_COLOR}: {e.filename} {e.strerror}]"
|
|
return f"[Journal exported to {path}]"
|
|
|
|
def export(self, journal, format="text", output=None):
|
|
"""Exports to individual files if output is an existing path, or into
|
|
a single file if output is a file name, or returns the exporter's
|
|
representation as string if output is None."""
|
|
if output and os.path.isdir(output): # multiple files
|
|
return self.write_files(journal, output)
|
|
elif output: # single file
|
|
return self.write_file(journal, output)
|
|
else:
|
|
return self.export_journal(journal)
|