Added new data model for old revisions and statuses

Signed-off-by: TheJackiMonster <thejackimonster@gmail.com>

bin/test_io.py

@@ -24,3 +24,16 @@ plots = data.Plots(path)
 
 plots.load()
 plots.save()
+
+revs = data.Revisions(path)
+
+revs.load()
+
+statusHost = data.StatusHost(path)
+
+statusHost.load()
+
+for status in statusHost:
+    print("--" + str(status))
+
+statusHost.save()

manuskript/data/__init__.py

@@ -1,6 +1,9 @@
 #!/usr/bin/env python
 # --!-- coding: utf8 --!--
 
-from manuskript.data.unique_id import UniqueIDHost, UniqueID
+
 from manuskript.data.plots import Plots, PlotLine, PlotStep
+from manuskript.data.revisions import Revisions
 from manuskript.data.settings import Settings
+from manuskript.data.status import StatusHost, Status
+from manuskript.data.unique_id import UniqueIDHost, UniqueID

manuskript/data/revisions.py

@@ -0,0 +1,82 @@
+#!/usr/bin/env python
+# --!-- coding: utf8 --!--
+
+import os
+
+from lxml import etree
+from manuskript.io.xmlFile import XmlFile
+
+
+class RevisionEntry:
+
+    def __init__(self, outline, timestamp: int, text: str):
+        self.outline = outline
+
+        self.timestamp = timestamp
+        self.text = text
+
+    def load(self):
+        self.outline.load()
+
+
+class RevisionOutline:
+
+    def __init__(self, revisions, ID: int):
+        self.revisions = revisions
+
+        self.ID = ID
+        self.entries = list()
+
+    def __iter__(self):
+        return self.entries.__iter__()
+
+    def load(self):
+        self.revisions.load()
+
+
+class Revisions:
+
+    def __init__(self, path):
+        self.file = XmlFile(os.path.join(path, "revisions.xml"))
+        self.outline = dict()
+
+    def __iter__(self):
+        return self.outline.values().__iter__()
+
+    @classmethod
+    def loadRevisionEntry(cls, revisions, ID: int, element: etree.Element):
+        timestamp = element.get("timestamp")
+        text = element.get("text")
+
+        if (timestamp is None) or (text is None):
+            return
+
+        revOutline = revisions.outline.get(ID, None)
+
+        if revOutline is None:
+            revOutline = RevisionOutline(revisions, ID)
+            revisions.outline[ID] = revOutline
+
+        revOutline.entries.append(RevisionEntry(revOutline, timestamp, text))
+
+    @classmethod
+    def loadRevisionOutline(cls, revisions, element: etree.Element, parent: etree.Element):
+        if element.tag == "revision":
+            if parent is None:
+                return
+
+            ID = int(parent.get("ID"))
+
+            cls.loadRevisionEntry(revisions, ID, element)
+        elif element.tag == "outlineItem":
+            ID = element.get("ID")
+
+            if ID is None:
+                return
+
+            for child in element:
+                cls.loadRevisionOutline(revisions, child, element)
+
+    def load(self):
+        tree = self.file.load()
+        Revisions.loadRevisionOutline(self, tree.getroot(), None)

manuskript/data/status.py

@@ -0,0 +1,66 @@
+#!/usr/bin/env python
+# --!-- coding: utf8 --!--
+
+import os
+
+from manuskript.io.textFile import TextFile
+
+
+class Status:
+
+    def __init__(self, host, name: str):
+        self.host = host
+
+        self.name = name
+
+    def __str__(self):
+        return self.name
+
+    def load(self):
+        self.host.load()
+
+    def save(self):
+        self.host.save()
+
+
+class StatusHost:
+
+    def __init__(self, path):
+        self.file = TextFile(os.path.join(path, "status.txt"))
+        self.statuses = dict()
+
+    def addStatus(self, name: str):
+        self.statuses[name] = Status(self, name)
+
+    def removeStatus(self, name: str):
+        self.statuses.pop(name)
+
+    def renameStatus(self, oldName: str, newName: str):
+        status = self.statuses.get(oldName)
+        status.name = newName
+        self.statuses[newName] = status
+        self.statuses.pop(oldName)
+
+    def getStatus(self, name: str):
+        return self.statuses.get(name)
+
+    def __iter__(self):
+        return self.statuses.values().__iter__()
+
+    def load(self):
+        text = self.file.load()
+        self.statuses.clear()
+
+        if len(text) <= 1:
+            return
+
+        text = text[:-1]
+
+        if len(text) <= 0:
+            return
+
+        for name in text.split("\n"):
+            self.addStatus(name)
+
+    def save(self):
+        self.file.save("\n".join(self.statuses.keys()) + "\n")