mirror of
https://github.com/jrnl-org/jrnl.git
synced 2025-05-20 13:08:31 +02:00
58 lines
2 KiB
Python
58 lines
2 KiB
Python
#!/usr/bin/env python
|
|
# encoding: utf-8
|
|
|
|
from .json_exporter import JSONExporter
|
|
from .util import get_tags_count
|
|
from xml.dom import minidom
|
|
|
|
|
|
class XMLExporter(JSONExporter):
|
|
"""This Exporter can convert entries and journals into XML."""
|
|
names = ["xml"]
|
|
extension = "xml"
|
|
|
|
@classmethod
|
|
def export_entry(cls, entry, doc=None):
|
|
"""Returns an XML representation of a single entry."""
|
|
doc_el = doc or minidom.Document()
|
|
entry_el = doc_el.createElement('entry')
|
|
for key, value in cls.entry_to_dict(entry).items():
|
|
elem = doc_el.createElement(key)
|
|
elem.appendChild(doc_el.createTextNode(value))
|
|
entry_el.appendChild(elem)
|
|
if not doc:
|
|
doc_el.appendChild(entry_el)
|
|
return doc_el.toprettyxml()
|
|
else:
|
|
return entry_el
|
|
|
|
@classmethod
|
|
def entry_to_xml(cls, entry, doc):
|
|
entry_el = doc.createElement('entry')
|
|
entry_el.setAttribute('date', entry.date.isoformat())
|
|
if hasattr(entry, "uuid"):
|
|
entry_el.setAttribute('uuid', entry.uuid)
|
|
entry_el.setAttribute('starred', entry.starred)
|
|
entry_el.appendChild(doc.createTextNode(entry.fulltext))
|
|
return entry_el
|
|
|
|
@classmethod
|
|
def export_journal(cls, journal):
|
|
"""Returns an XML representation of an entire journal."""
|
|
tags = get_tags_count(journal)
|
|
doc = minidom.Document()
|
|
xml = doc.createElement('journal')
|
|
tags_el = doc.createElement('tags')
|
|
entries_el = doc.createElement('entries')
|
|
for count, tag in tags:
|
|
tag_el = doc.createElement('tag')
|
|
tag_el.setAttribute('name', tag)
|
|
count_node = doc.createTextNode(str(count))
|
|
tag_el.appendChild(count_node)
|
|
tags_el.appendChild(tag_el)
|
|
for entry in journal.entries:
|
|
entries_el.appendChild(cls.entry_to_xml(entry, doc))
|
|
xml.appendChild(entries_el)
|
|
xml.appendChild(tags_el)
|
|
doc.appendChild(xml)
|
|
return doc.toprettyxml()
|