summaryrefslogtreecommitdiff
path: root/nephilim/song.py
blob: 3bd319cfa504338e1117ba94479b7ef6ffe6ccb4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
from PyQt4 import QtCore
import os

from misc import sec2min

class Song:
    """The Song class offers an abstraction of a song."""
    _data = None

    def __init__(self, data):
        self._data = data
        if 'id' in self._data:
            self._data['id'] = int(self._data['id'])
        if 'track' in self._data:
            # make sure the track is a valid number!
            t=self._data['track']
            for i in xrange(len(t)):
                if ord(t[i])<ord('0') or ord(t[i])>ord('9'):
                    try:
                        self._data['track']=int(t[0:i])
                    except TypeError:
                        self._data['track']=-1
                    break
            self._data['track']=int(self._data['track'])

        # ensure all string-values are utf-8 encoded
        for tag in self._data.keys():
            if isinstance(self._data[tag], str):
                self._data[tag] = self._data[tag].decode('utf-8')
        if 'time' in self._data:
            self._data['time']   = int(self._data['time'])
            self._data['timems'] = '%i:%i'%(self._data['time'] / 60, self._data['time'] % 60)
            self._data['length'] = sec2min(self._data['time'])

    def id(self):
        """Get the song's playlist ID. (-1 if not in playlist)."""
        return self.tag('id', -1)

    def title(self):
        """Get the song's title (or filename if it has no title)."""
        return self.tag('title', self._data['file'])

    def artist(self):
        """Get the song's artist."""
        return self.tag('artist')

    def track(self):
        """Get the song's track number."""
        return self.tag('track')

    def album(self):
        """Get the album."""
        return self.tag('album')

    def filepath(self):
        """Get the filepath."""
        return self._data['file']

    def tag(self, tag, default=''):
        """Get a tag. If it doesn't exist, return default."""
        if tag in self._data:
            return self._data[tag]
        if tag=='song':
            return self.__str__()

        return default

    def expand_tags(self, str):
        """Expands tags in form $tag in str."""
        ret = str
        ret = ret.replace('$title', self.title())    #to ensure that it is set to at least filename
        for tag in self._data:
            ret = ret.replace('$' + tag, unicode(self._data[tag]))
        ret = ret.replace('$songdir', os.path.dirname(self.filepath()))
        return ret