Guido van Rossum | 1ce7c6f | 1997-01-15 19:19:19 +0000 | [diff] [blame] | 1 | # This file implements a class which forms an interface to the .cddb |
| 2 | # directory that is maintained by SGI's cdman program. |
| 3 | # |
| 4 | # Usage is as follows: |
| 5 | # |
| 6 | # import readcd |
| 7 | # r = readcd.Readcd() |
| 8 | # c = Cddb(r.gettrackinfo()) |
| 9 | # |
| 10 | # Now you can use c.artist, c.title and c.track[trackno] (where trackno |
| 11 | # starts at 1). When the CD is not recognized, all values will be the empty |
| 12 | # string. |
| 13 | # It is also possible to set the above mentioned variables to new values. |
| 14 | # You can then use c.write() to write out the changed values to the |
| 15 | # .cdplayerrc file. |
| 16 | |
| 17 | import string, posix, os |
| 18 | |
| 19 | _cddbrc = '.cddb' |
| 20 | _DB_ID_NTRACKS = 5 |
| 21 | _dbid_map = '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ@_=+abcdefghijklmnopqrstuvwxyz' |
| 22 | def _dbid(v): |
| 23 | if v >= len(_dbid_map): |
| 24 | return string.zfill(v, 2) |
| 25 | else: |
| 26 | return _dbid_map[v] |
| 27 | |
| 28 | def tochash(toc): |
| 29 | if type(toc) == type(''): |
| 30 | tracklist = [] |
| 31 | for i in range(2, len(toc), 4): |
| 32 | tracklist.append((None, |
Eric S. Raymond | dcd3a87 | 2001-02-09 16:45:10 +0000 | [diff] [blame] | 33 | (int(toc[i:i+2]), |
| 34 | int(toc[i+2:i+4])))) |
Guido van Rossum | 1ce7c6f | 1997-01-15 19:19:19 +0000 | [diff] [blame] | 35 | else: |
| 36 | tracklist = toc |
| 37 | ntracks = len(tracklist) |
| 38 | hash = _dbid((ntracks >> 4) & 0xF) + _dbid(ntracks & 0xF) |
| 39 | if ntracks <= _DB_ID_NTRACKS: |
| 40 | nidtracks = ntracks |
| 41 | else: |
| 42 | nidtracks = _DB_ID_NTRACKS - 1 |
| 43 | min = 0 |
| 44 | sec = 0 |
| 45 | for track in tracklist: |
| 46 | start, length = track |
| 47 | min = min + length[0] |
| 48 | sec = sec + length[1] |
| 49 | min = min + sec / 60 |
| 50 | sec = sec % 60 |
| 51 | hash = hash + _dbid(min) + _dbid(sec) |
| 52 | for i in range(nidtracks): |
| 53 | start, length = tracklist[i] |
| 54 | hash = hash + _dbid(length[0]) + _dbid(length[1]) |
| 55 | return hash |
| 56 | |
| 57 | class Cddb: |
| 58 | def __init__(self, tracklist): |
| 59 | if os.environ.has_key('CDDB_PATH'): |
| 60 | path = os.environ['CDDB_PATH'] |
Eric S. Raymond | dcd3a87 | 2001-02-09 16:45:10 +0000 | [diff] [blame] | 61 | cddb_path = path.split(',') |
Guido van Rossum | 1ce7c6f | 1997-01-15 19:19:19 +0000 | [diff] [blame] | 62 | else: |
| 63 | home = os.environ['HOME'] |
| 64 | cddb_path = [home + '/' + _cddbrc] |
| 65 | |
| 66 | self._get_id(tracklist) |
| 67 | |
| 68 | for dir in cddb_path: |
| 69 | file = dir + '/' + self.id + '.rdb' |
| 70 | try: |
| 71 | f = open(file, 'r') |
| 72 | self.file = file |
| 73 | break |
| 74 | except IOError: |
| 75 | pass |
Eric S. Raymond | dcd3a87 | 2001-02-09 16:45:10 +0000 | [diff] [blame] | 76 | ntracks = int(self.id[:2], 16) |
Guido van Rossum | 1ce7c6f | 1997-01-15 19:19:19 +0000 | [diff] [blame] | 77 | self.artist = '' |
| 78 | self.title = '' |
| 79 | self.track = [None] + [''] * ntracks |
| 80 | self.trackartist = [None] + [''] * ntracks |
| 81 | self.notes = [] |
| 82 | if not hasattr(self, 'file'): |
| 83 | return |
Guido van Rossum | 9694fca | 1997-10-22 21:00:49 +0000 | [diff] [blame] | 84 | import re |
| 85 | reg = re.compile(r'^([^.]*)\.([^:]*):[\t ]+(.*)') |
Guido van Rossum | 1ce7c6f | 1997-01-15 19:19:19 +0000 | [diff] [blame] | 86 | while 1: |
| 87 | line = f.readline() |
| 88 | if not line: |
| 89 | break |
Guido van Rossum | 9694fca | 1997-10-22 21:00:49 +0000 | [diff] [blame] | 90 | match = reg.match(line) |
| 91 | if not match: |
Guido van Rossum | 1ce7c6f | 1997-01-15 19:19:19 +0000 | [diff] [blame] | 92 | print 'syntax error in ' + file |
| 93 | continue |
Guido van Rossum | 9694fca | 1997-10-22 21:00:49 +0000 | [diff] [blame] | 94 | name1, name2, value = match.group(1, 2, 3) |
Guido van Rossum | 1ce7c6f | 1997-01-15 19:19:19 +0000 | [diff] [blame] | 95 | if name1 == 'album': |
| 96 | if name2 == 'artist': |
| 97 | self.artist = value |
| 98 | elif name2 == 'title': |
| 99 | self.title = value |
| 100 | elif name2 == 'toc': |
| 101 | if not self.toc: |
| 102 | self.toc = value |
| 103 | if self.toc != value: |
| 104 | print 'toc\'s don\'t match' |
| 105 | elif name2 == 'notes': |
| 106 | self.notes.append(value) |
| 107 | elif name1[:5] == 'track': |
| 108 | try: |
Eric S. Raymond | dcd3a87 | 2001-02-09 16:45:10 +0000 | [diff] [blame] | 109 | trackno = int(name1[5:]) |
| 110 | except ValueError: |
Guido van Rossum | 1ce7c6f | 1997-01-15 19:19:19 +0000 | [diff] [blame] | 111 | print 'syntax error in ' + file |
| 112 | continue |
| 113 | if trackno > ntracks: |
| 114 | print 'track number ' + `trackno` + \ |
| 115 | ' in file ' + file + \ |
| 116 | ' out of range' |
| 117 | continue |
| 118 | if name2 == 'title': |
| 119 | self.track[trackno] = value |
| 120 | elif name2 == 'artist': |
| 121 | self.trackartist[trackno] = value |
| 122 | f.close() |
| 123 | for i in range(2, len(self.track)): |
| 124 | track = self.track[i] |
| 125 | # if track title starts with `,', use initial part |
| 126 | # of previous track's title |
| 127 | if track and track[0] == ',': |
| 128 | try: |
Eric S. Raymond | dcd3a87 | 2001-02-09 16:45:10 +0000 | [diff] [blame] | 129 | off = self.track[i - 1].index(',') |
| 130 | except ValueError: |
Guido van Rossum | 1ce7c6f | 1997-01-15 19:19:19 +0000 | [diff] [blame] | 131 | pass |
| 132 | else: |
| 133 | self.track[i] = self.track[i-1][:off] \ |
| 134 | + track |
| 135 | |
| 136 | def _get_id(self, tracklist): |
| 137 | # fill in self.id and self.toc. |
| 138 | # if the argument is a string ending in .rdb, the part |
| 139 | # upto the suffix is taken as the id. |
| 140 | if type(tracklist) == type(''): |
| 141 | if tracklist[-4:] == '.rdb': |
| 142 | self.id = tracklist[:-4] |
| 143 | self.toc = '' |
| 144 | return |
| 145 | t = [] |
| 146 | for i in range(2, len(tracklist), 4): |
| 147 | t.append((None, \ |
Eric S. Raymond | dcd3a87 | 2001-02-09 16:45:10 +0000 | [diff] [blame] | 148 | (int(tracklist[i:i+2]), \ |
| 149 | int(tracklist[i+2:i+4])))) |
Guido van Rossum | 1ce7c6f | 1997-01-15 19:19:19 +0000 | [diff] [blame] | 150 | tracklist = t |
| 151 | ntracks = len(tracklist) |
| 152 | self.id = _dbid((ntracks >> 4) & 0xF) + _dbid(ntracks & 0xF) |
| 153 | if ntracks <= _DB_ID_NTRACKS: |
| 154 | nidtracks = ntracks |
| 155 | else: |
| 156 | nidtracks = _DB_ID_NTRACKS - 1 |
| 157 | min = 0 |
| 158 | sec = 0 |
| 159 | for track in tracklist: |
| 160 | start, length = track |
| 161 | min = min + length[0] |
| 162 | sec = sec + length[1] |
| 163 | min = min + sec / 60 |
| 164 | sec = sec % 60 |
| 165 | self.id = self.id + _dbid(min) + _dbid(sec) |
| 166 | for i in range(nidtracks): |
| 167 | start, length = tracklist[i] |
| 168 | self.id = self.id + _dbid(length[0]) + _dbid(length[1]) |
| 169 | self.toc = string.zfill(ntracks, 2) |
| 170 | for track in tracklist: |
| 171 | start, length = track |
| 172 | self.toc = self.toc + string.zfill(length[0], 2) + \ |
| 173 | string.zfill(length[1], 2) |
| 174 | |
| 175 | def write(self): |
| 176 | import posixpath |
| 177 | if os.environ.has_key('CDDB_WRITE_DIR'): |
| 178 | dir = os.environ['CDDB_WRITE_DIR'] |
| 179 | else: |
| 180 | dir = os.environ['HOME'] + '/' + _cddbrc |
| 181 | file = dir + '/' + self.id + '.rdb' |
| 182 | if posixpath.exists(file): |
| 183 | # make backup copy |
| 184 | posix.rename(file, file + '~') |
| 185 | f = open(file, 'w') |
| 186 | f.write('album.title:\t' + self.title + '\n') |
| 187 | f.write('album.artist:\t' + self.artist + '\n') |
| 188 | f.write('album.toc:\t' + self.toc + '\n') |
| 189 | for note in self.notes: |
| 190 | f.write('album.notes:\t' + note + '\n') |
| 191 | prevpref = None |
| 192 | for i in range(1, len(self.track)): |
| 193 | if self.trackartist[i]: |
| 194 | f.write('track'+`i`+'.artist:\t'+self.trackartist[i]+'\n') |
| 195 | track = self.track[i] |
| 196 | try: |
Eric S. Raymond | dcd3a87 | 2001-02-09 16:45:10 +0000 | [diff] [blame] | 197 | off = track.index(',') |
| 198 | except ValueError: |
Guido van Rossum | 1ce7c6f | 1997-01-15 19:19:19 +0000 | [diff] [blame] | 199 | prevpref = None |
| 200 | else: |
| 201 | if prevpref and track[:off] == prevpref: |
| 202 | track = track[off:] |
| 203 | else: |
| 204 | prevpref = track[:off] |
| 205 | f.write('track' + `i` + '.title:\t' + track + '\n') |
| 206 | f.close() |