Ajout Amsterdam.
[minwii.git] / src / minwii / widgets / songfilebrowser.py
1 # -*- coding: utf-8 -*-
2 """
3 Boîte de dialogue pour sélection des chansons.
4
5 $Id$
6 $URL$
7 """
8
9 from pgu.gui import FileDialog
10 import os
11 import tempfile
12 from xml.etree import ElementTree
13 from minwii.musicxml import musicXml2Song
14
15 INDEX_TXT = 'index.txt'
16
17 class FileOpenDialog(FileDialog):
18
19
20
21 def __init__(self, path):
22 FileDialog.__init__(self,
23 title_txt="Ouvrir une chanson",
24 button_txt="Ouvrir",
25 path=path,
26 )
27 self.list.style.width = 700
28 self.list.style.height = 250
29
30 def _list_dir_(self):
31 self.input_dir.value = self.curdir
32 self.input_dir.pos = len(self.curdir)
33 self.input_dir.vpos = 0
34 dirs = []
35 files = []
36 try:
37 for i in os.listdir(self.curdir):
38 if i.startswith('.') : continue
39 if os.path.isdir(os.path.join(self.curdir, i)): dirs.append(i)
40 else: files.append(i)
41 except:
42 self.input_file.value = "Dossier innacessible !"
43
44 dirs.sort()
45 dirs.insert(0, '..')
46
47 files.sort()
48 for i in dirs:
49 self.list.add(i, image=self.dir_img, value=i)
50
51 xmlFiles = []
52 for i in files:
53 if not i.endswith('.xml') :
54 continue
55 filepath = os.path.join(self.curdir, i)
56 xmlFiles.append(filepath)
57 # self.list.add(FileOpenDialog.getSongTitle(filepath), value=i)
58
59 if xmlFiles :
60 printableLines = self.getPrintableLines(xmlFiles)
61 for l in printableLines :
62 self.list.add(l[0], value = l[1])
63
64 self.list.set_vertical_scroll(0)
65
66 def getPrintableLines(self, xmlFiles) :
67 index = self.getUpdatedIndex(xmlFiles)
68
69 printableLines = []
70 for l in index :
71 l = l.strip()
72 l = l.split('\t')
73 printableLines.append(('%s - %s / %s' % (l[2], l[3], l[4]), l[0]))
74
75 return printableLines
76
77
78 @staticmethod
79 def getSongTitle(file) :
80 it = ElementTree.iterparse(file, ['start', 'end'])
81 creditFound = False
82 title = os.path.basename(file)
83
84 for evt, el in it :
85 if el.tag == 'credit' :
86 creditFound = True
87 if el.tag == 'credit-words' and creditFound:
88 title = el.text
89 break
90 if el.tag == 'part-list' :
91 # au delà de ce tag : aucune chance de trouver un titre
92 break
93 return title
94
95 @staticmethod
96 def getSongMetadata(file) :
97 metadata = {}
98 metadata['title'] = FileOpenDialog.getSongTitle(file).encode('iso-8859-1')
99 metadata['mtime'] = str(os.stat(file).st_mtime)
100 metadata['file'] = os.path.basename(file)
101 song = musicXml2Song(file)
102 metadata['distinctNotes'] = len(song.distinctNotes)
103
104 histo = song.intervalsHistogram
105 coeffInter = reduce(lambda a, b : a + b,
106 [abs(k) * v for k, v in histo.items()])
107
108 totInter = reduce(lambda a, b: a+b, histo.values())
109 totInter = totInter - histo.get(0, 0)
110 difficulty = int(round(float(coeffInter) / totInter, 0))
111 metadata['difficulty'] = difficulty
112
113 return metadata
114
115 def getUpdatedIndex(self, xmlFiles) :
116 indexTxtPath = os.path.join(self.curdir, INDEX_TXT)
117 index = []
118
119 if not os.path.exists(indexTxtPath) :
120 musicXmlFound = False
121 tmp = tempfile.TemporaryFile(mode='r+')
122 for file in xmlFiles :
123 try :
124 metadata = FileOpenDialog.getSongMetadata(file)
125 musicXmlFound = True
126 except ValueError, e :
127 print e
128 if e.args and e.args[0] == 'not a musicxml file' :
129 continue
130
131 line = '%(file)s\t%(mtime)s\t%(title)s\t%(distinctNotes)d\t%(difficulty)d\n' % metadata
132 index.append(line)
133 tmp.write(line)
134
135 if musicXmlFound :
136 tmp.seek(0)
137 indexFile = open(indexTxtPath, 'w')
138 indexFile.write(tmp.read())
139 indexFile.close()
140 tmp.close()
141 else :
142 indexedFiles = {}
143 indexTxt = open(indexTxtPath, 'r')
144
145 # check if index is up to date, and update entries if so.
146 for l in filter(None, indexTxt.readlines()) :
147 parts = l.split('\t')
148 fileBaseName, modificationTime = parts[0], parts[1]
149 filePath = os.path.join(self.curdir, fileBaseName)
150
151 if not os.path.exists(filePath) :
152 continue
153
154 indexedFiles[fileBaseName] = l
155 currentMtime = str(os.stat(filePath).st_mtime)
156
157 # check modification time missmatch
158 if currentMtime != modificationTime :
159 try :
160 metadata = FileOpenDialog.getSongMetadata(filePath)
161 musicXmlFound = True
162 except ValueError, e :
163 print e
164 if e.args and e.args[0] == 'not a musicxml file' :
165 continue
166
167 metadata = FileOpenDialog.getSongMetadata(filePath)
168 line = '%(file)s\t%(mtime)s\t%(title)s\t%(distinctNotes)d\t%(difficulty)d\n' % metadata
169 indexedFiles[fileBaseName] = line
170
171 # check for new files.
172 for file in xmlFiles :
173 fileBaseName = os.path.basename(file)
174 if not indexedFiles.has_key(fileBaseName) :
175 try :
176 metadata = FileOpenDialog.getSongMetadata(filePath)
177 musicXmlFound = True
178 except ValueError, e :
179 print e
180 if e.args and e.args[0] == 'not a musicxml file' :
181 continue
182
183 metadata = FileOpenDialog.getSongMetadata(file)
184 line = '%(file)s\t%(mtime)s\t%(title)s\t%(distinctNotes)d\t%(difficulty)d\n' % metadata
185 indexedFiles[fileBaseName] = line
186
187 # ok, the index is up to date !
188
189 index = indexedFiles.values()
190 index.sort()
191
192
193 return index
194