Move the qml/* files to the src folder. Now it is the main version
[mussorgsky] / src / albumModel.py
1 # -*- coding: utf-8 -*-
2 import os
3 import sys
4 from PySide import QtCore
5 from PySide import QtGui
6 from PySide import QtDeclarative
7
8 from albumItem import AlbumItem
9
10 class AlbumModel (QtCore.QAbstractListModel):
11     COLUMNS = ('album', )
12
13     def __init__ (self, data):
14         QtCore.QAbstractListModel.__init__ (self)
15         self._albums = data
16         self.setRoleNames (dict(enumerate(AlbumModel.COLUMNS)))
17
18     def rowCount (self, parent=QtCore.QModelIndex()):
19         return len (self._albums)
20
21     def data (self, index, role):
22         if index.isValid () and role == AlbumModel.COLUMNS.index ('album'):
23             return self._albums[index.row ()]
24         return None
25
26     def getAlbumInRow (self, row):
27         assert row >= 0 and row < len (self._albums)
28         return self._albums [row]
29
30     def updateThumb (self, row, url):
31         print "Changing album_art from",self._albums[row].album_art, "to:", url
32         assert row >= 0 and row < len (self._albums)
33         self._albums[row].album_art = url
34
35     def get_albums (self):
36         return self._albums