From 3fd446dd1eb5dc3ce345d38ab1dae65e40b68b95 Mon Sep 17 00:00:00 2001
From: Olivier Keshavjee <olivier.keshavjee@gmail.com>
Date: Sat, 5 Mar 2016 12:55:56 +0100
Subject: [PATCH] Working on backward compatibility

---
 manuskript/load_save/version_0.py   | 24 ++++++++++++--
 manuskript/load_save/version_1.py   | 51 +++++++++++++++++++++++++++--
 manuskript/models/characterModel.py |  2 +-
 3 files changed, 71 insertions(+), 6 deletions(-)

diff --git a/manuskript/load_save/version_0.py b/manuskript/load_save/version_0.py
index 4023539e..1874b050 100644
--- a/manuskript/load_save/version_0.py
+++ b/manuskript/load_save/version_0.py
@@ -14,6 +14,7 @@ from lxml import etree as ET
 
 from manuskript import settings
 from manuskript.functions import iconColor, iconFromColorString, mainWindow
+from manuskript.models.characterModel import Character
 
 try:
     import zlib  # Used with zipfile for compression
@@ -130,8 +131,7 @@ def loadProject(project):
         errors.append("flatModel.xml")
 
     if "perso.xml" in files:
-        loadStandardItemModelXML(mw.mdlCharacter,
-                                 files["perso.xml"], fromString=True)
+        loadStandardItemModelXMLForCharacters(mw.mdlCharacter, files["perso.xml"])
     else:
         errors.append("perso.xml")
 
@@ -246,4 +246,22 @@ def loadItem(root, mdl, parent=QModelIndex()):
 
             if len(col) != 0:
                 # loadItem(col, mdl, mdl.index(r, c, parent))
-                loadItem(col, mdl, mdl.indexFromItem(item))
\ No newline at end of file
+                loadItem(col, mdl, mdl.indexFromItem(item))
+
+
+def loadStandardItemModelXMLForCharacters(mdl, xml):
+    mdl = mainWindow().mdlCharacter
+    root = ET.fromstring(xml)
+    data = root.find("data")
+    for row in data:
+        char = Character(mdl)
+        for col in row:
+            c = int(col.attrib["col"])
+            # Value
+            if col.text:
+                char._data[c] = col.text
+            # Color
+            if "color" in col.attrib:
+                char.setColor(QColor(col.attrib["color"]))
+            # TODO: infos
+        mdl.characters.append(char)
\ No newline at end of file
diff --git a/manuskript/load_save/version_1.py b/manuskript/load_save/version_1.py
index ca6c986c..ede22ddb 100644
--- a/manuskript/load_save/version_1.py
+++ b/manuskript/load_save/version_1.py
@@ -12,6 +12,7 @@ from PyQt5.QtCore import Qt
 from PyQt5.QtGui import QColor
 
 from manuskript import settings
+from manuskript.enums import Character
 from manuskript.functions import mainWindow, iconColor
 
 try:
@@ -24,6 +25,17 @@ except:
 
 cache = {}
 
+
+def formatMetaData(name, value, tabLength=10):
+
+    # TODO: escape ":" in name
+    return "{name}:{spaces}{value}\n".format(
+        name=name,
+        spaces=" " * (tabLength - len(name)),
+        value=value
+    )
+
+
 def saveProject(zip=None):
     """
     Saves the project. If zip is False, the project is saved as a multitude of plain-text files for the most parts
@@ -35,7 +47,7 @@ def saveProject(zip=None):
     @return: Nothing
     """
     if zip is None:
-        zip = True
+        zip = False
         # Fixme
 
 
@@ -118,7 +130,42 @@ def saveProject(zip=None):
     # Characters (self.mdlCharacter)
     # In a character folder
 
-    # TODO
+    path = os.path.join("characters", "{name}.txt")  # Not sure wheter os.path allows this
+    _map = [
+        (Character.name, "Name"),
+        (Character.ID,   "ID"),
+        (Character.importance, "Importance"),
+        (Character.motivation, "Motivation"),
+        (Character.goal, "Goal"),
+        (Character.conflict, "Conflict"),
+        (Character.summarySentence, "Phrase Summary"),
+        (Character.summaryPara, "Paragraph Summary"),
+        (Character.summaryFull, "Full Summary"),
+        (Character.notes, "Notes"),
+    ]
+    mdl = mw.mdlCharacter
+    for c in mdl.characters:
+        content = ""
+        LENGTH = 20
+        for m, name in _map:
+            val = mdl.data(c.index(m.value)).strip()
+            if val:
+                # Multiline formatting
+                if len(val.split("\n")) > 1:
+                    val = "\n".join([" " * (LENGTH + 1) + l for l in val.split("\n")])[LENGTH + 1:]
+
+                content += formatMetaData(name, val, LENGTH)
+
+        for info in c.infos:
+            content += formatMetaData(info.description, info.value, LENGTH)
+
+        name = "{ID}-{slugName}".format(
+            ID=c.ID(),
+            slugName="FIXME"
+        )
+        files.append((
+            path.format(name=name),
+            content))
 
     # Texts
     # In an outline folder
diff --git a/manuskript/models/characterModel.py b/manuskript/models/characterModel.py
index a9881f94..0a922bd7 100644
--- a/manuskript/models/characterModel.py
+++ b/manuskript/models/characterModel.py
@@ -213,7 +213,7 @@ class characterModel(QAbstractItemModel):
 ###############################################################################
 
 class Character():
-    def __init__(self, model, name):
+    def __init__(self, model, name="No name"):
         self._model = model
 
         self._data = {}