From c852447716c2efeeab07b22ebd11ebd30d302c27 Mon Sep 17 00:00:00 2001
From: Louis <spalax@gresille.org>
Date: Mon, 25 Mar 2013 23:36:05 +0100
Subject: [PATCH] Analyse syntaxique de \beginsong
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- Correction de problèmes dans l'analyse du titre
- Analyse des arguments
---
 songbook.py      | 29 ++++++++---------------------
 utils/plastex.py |  5 ++++-
 utils/songs.py   | 23 +++++++++++++++++++----
 3 files changed, 31 insertions(+), 26 deletions(-)

diff --git a/songbook.py b/songbook.py
index 887799be..80117f4e 100755
--- a/songbook.py
+++ b/songbook.py
@@ -14,19 +14,15 @@ import platform
 from utils import recursiveFind
 from utils.plastex import parsetex
 
-reArtist = re.compile('(?<=by=)(.(?<![,\\]\\}]))+')
-reAlbum  = re.compile('(?<=album=)(.(?<![,\\]\\}]))+')
-
 class Song:
-    def __init__(self, titles, artist, album, path, languages):
+    def __init__(self, path, languages, titles, args):
         self.titles  = titles
-        self.artist = artist
-        self.album  = album
+        self.args   = args
         self.path   = path
         self.languages = languages
 
     def __repr__(self):
-        return repr((self.titles, self.artist, self.album, self.path))
+        return repr((self.titles, self.args, self.path))
 
 if platform.system() == "Linux":
 	from xdg.BaseDirectory import *
@@ -79,19 +75,10 @@ class SongsList:
         album, etc.).
         """
         path = os.path.join(self._library, 'songs', filename)
-        with open(path, 'r+') as f:
-            data   = f.read()
-            artist = reArtist.search(data.replace("{","")).group(0)
-            match  = reAlbum.search(data.replace("{",""))
-            if match:
-                album = match.group(0)
-            else:
-                album = ''
-
-            # Exécution de PlasTeX
-            data = parsetex(path)
+        # Exécution de PlasTeX
+        data = parsetex(path)
 
-            self.songs.append(Song(artist = artist, album = album, path = path, **data))
+        self.songs.append(Song(path, data['languages'], data['titles'], data['args']))
 
     def append_list(self, filelist):
         """Ajoute une liste de chansons à la liste
@@ -108,8 +95,8 @@ class SongsList:
         configurer l'ordre de ce tri.
         """
         self.songs = sorted(self.songs, key=lambda x: locale.strxfrm(unprefixed(x.titles[0], self._prefixes)))
-        self.songs = sorted(self.songs, key=lambda x: locale.strxfrm(x.album))
-        self.songs = sorted(self.songs, key=lambda x: locale.strxfrm(x.artist))
+        self.songs = sorted(self.songs, key=lambda x: locale.strxfrm(x.args.get("album", "")))
+        self.songs = sorted(self.songs, key=lambda x: locale.strxfrm(x.args.get("by", "")))
 
     def latex(self):
         """Renvoie le code LaTeX nécessaire pour intégrer la liste de chansons.
diff --git a/utils/plastex.py b/utils/plastex.py
index ebb9b526..426b4c76 100755
--- a/utils/plastex.py
+++ b/utils/plastex.py
@@ -33,7 +33,9 @@ def parsetex(filename):
     Renvoie un dictionnaire contenant les métadonnées lues dans le fichier. Les
     clefs sont :
     - languages: l'ensemble des langages utilisés (recherche des
-      \selectlanguages{}).
+      \selectlanguages{}) ;
+    - titles: la liste des titres ;
+    - args: le dictionnaire des paramètres passés à \\beginsong.
     """
     # Analyse syntaxique
     doc = SongParser.parse(filename)
@@ -47,5 +49,6 @@ def parsetex(filename):
             data["languages"].add(node.attributes['lang'])
         if node.nodeName == "beginsong":
             data["titles"] = node.attributes["titles"]
+            data["args"] = node.attributes["args"]
 
     return data
diff --git a/utils/songs.py b/utils/songs.py
index 2f376f0e..f8694e03 100644
--- a/utils/songs.py
+++ b/utils/songs.py
@@ -3,6 +3,19 @@
 
 import plasTeX
 
+def split_linebreak(texlist):
+    return_list = []
+    current = []
+    for token in texlist:
+        if token.nodeName == '\\':
+            return_list.append(current)
+            current = []
+        else:
+            current.append(token.textContent.encode('utf-8'))
+    if current:
+        return_list.append(current)
+    return return_list
+
 class beginsong(plasTeX.Command):
     args = '{titles}[ args:dict ]'
     def invoke(self, tex):
@@ -10,10 +23,12 @@ class beginsong(plasTeX.Command):
 
         # Parsing title
         titles = []
-        for token in self.attributes['titles'].allChildNodes:
-            if token.nodeName != '\\':
-                titles.append(token.source.encode('utf-8'))
+        for tokens in split_linebreak(self.attributes['titles'].allChildNodes):
+            titles.append("".join(tokens))
         self.attributes['titles'] = titles
 
         # Parsing keyval arguments
-        pass
+        args = {}
+        for (key, val) in self.attributes['args'].iteritems():
+            args[key] = val.textContent.encode('utf-8')
+        self.attributes['args'] = args