mirror of
https://github.com/olivierkes/manuskript.git
synced 2024-09-30 01:07:24 +13:00
a33249fc9d
Signed-off-by: TheJackiMonster <thejackimonster@gmail.com>
126 lines
3.7 KiB
Python
126 lines
3.7 KiB
Python
#!/usr/bin/env python
|
|
# --!-- coding: utf8 --!--
|
|
|
|
from PyQt5.QtWidgets import qApp, QMessageBox
|
|
from manuskript.models import outlineItem
|
|
from manuskript.enums import Outline
|
|
from lxml import etree as ET
|
|
from manuskript.functions import mainWindow, safeTranslate
|
|
from manuskript.importer.abstractImporter import abstractImporter
|
|
|
|
|
|
class opmlImporter(abstractImporter):
|
|
|
|
name = "OPML"
|
|
description = ""
|
|
fileFormat = "OPML Files (*.opml *.xml)"
|
|
icon = "text-x-opml+xml"
|
|
|
|
@classmethod
|
|
def isValid(cls):
|
|
return True
|
|
|
|
@classmethod
|
|
def startImport(cls, filePath, parentItem, settingsWidget, fromString=None):
|
|
"""
|
|
Import/export outline cards in OPML format.
|
|
"""
|
|
ret = False
|
|
|
|
if filePath != "":
|
|
# We have a filePath, so we read the file
|
|
try:
|
|
with open(filePath, 'rb') as opmlFile:
|
|
#opmlContent = cls.saveNewlines(opmlFile.read())
|
|
opmlContent = opmlFile.read()
|
|
except:
|
|
QMessageBox.critical(settingsWidget,
|
|
safeTranslate(qApp, "Import", "OPML Import"),
|
|
safeTranslate(qApp, "Import", "File open failed."))
|
|
return None
|
|
|
|
elif fromString == "":
|
|
# We have neither filePath nor fromString, so we leave
|
|
return None
|
|
|
|
else:
|
|
# We load from string
|
|
opmlContent = bytes(fromString, "utf-8")
|
|
|
|
parsed = ET.fromstring(opmlContent)
|
|
|
|
opmlNode = parsed
|
|
bodyNode = opmlNode.find("body")
|
|
items = []
|
|
|
|
if bodyNode != None:
|
|
outlineEls = bodyNode.findall("outline")
|
|
|
|
if outlineEls != None:
|
|
for element in outlineEls:
|
|
items.extend(cls.parseItems(element, parentItem))
|
|
ret = True
|
|
|
|
if not ret:
|
|
QMessageBox.critical(
|
|
settingsWidget,
|
|
safeTranslate(qApp, "Import", "OPML Import"),
|
|
safeTranslate(qApp, "Import", "This does not appear to be a valid OPML file."))
|
|
|
|
return None
|
|
|
|
return items
|
|
|
|
@classmethod
|
|
def parseItems(cls, underElement, parentItem=None):
|
|
items = []
|
|
title = underElement.get('text')
|
|
if title != None:
|
|
card = outlineItem(parent=parentItem, title=title)
|
|
items.append(card)
|
|
|
|
body = ""
|
|
note = underElement.get('_note')
|
|
|
|
if note != None and not cls.isWhitespaceOnly(note):
|
|
#body = cls.restoreNewLines(note)
|
|
body = note
|
|
|
|
children = underElement.findall('outline')
|
|
|
|
if children != None and len(children) > 0:
|
|
for el in children:
|
|
items.extend(cls.parseItems(el, card))
|
|
else:
|
|
card.setData(Outline.type, 'md')
|
|
card.setData(Outline.text, body)
|
|
|
|
return items
|
|
|
|
@classmethod
|
|
def saveNewlines(cls, inString):
|
|
"""
|
|
Since XML parsers are notorious for stripping out significant newlines,
|
|
save them in a form we can restore after the parse.
|
|
"""
|
|
inString = inString.replace("\r\n", "\n")
|
|
inString = inString.replace("\n", "{{lf}}")
|
|
|
|
return inString
|
|
|
|
@classmethod
|
|
def restoreNewLines(cls, inString):
|
|
"""
|
|
Restore any significant newlines
|
|
"""
|
|
return inString.replace("{{lf}}", "\n")
|
|
|
|
@classmethod
|
|
def isWhitespaceOnly(cls, inString):
|
|
"""
|
|
Determine whether or not a string only contains whitespace.
|
|
"""
|
|
s = cls.restoreNewLines(inString)
|
|
s = ''.join(s.split())
|
|
|
|
return len(s) == 0
|