ref: cb3049652da75b7ac00a865cd518f29ff05b3155
dir: /sys/lib/python/plat-irix5/cddb.py/
# This file implements a class which forms an interface to the .cddb # directory that is maintained by SGI's cdman program. # # Usage is as follows: # # import readcd # r = readcd.Readcd() # c = Cddb(r.gettrackinfo()) # # Now you can use c.artist, c.title and c.track[trackno] (where trackno # starts at 1). When the CD is not recognized, all values will be the empty # string. # It is also possible to set the above mentioned variables to new values. # You can then use c.write() to write out the changed values to the # .cdplayerrc file. import string, posix, os _cddbrc = '.cddb' _DB_ID_NTRACKS = 5 _dbid_map = '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ@_=+abcdefghijklmnopqrstuvwxyz' def _dbid(v): if v >= len(_dbid_map): return string.zfill(v, 2) else: return _dbid_map[v] def tochash(toc): if type(toc) == type(''): tracklist = [] for i in range(2, len(toc), 4): tracklist.append((None, (int(toc[i:i+2]), int(toc[i+2:i+4])))) else: tracklist = toc ntracks = len(tracklist) hash = _dbid((ntracks >> 4) & 0xF) + _dbid(ntracks & 0xF) if ntracks <= _DB_ID_NTRACKS: nidtracks = ntracks else: nidtracks = _DB_ID_NTRACKS - 1 min = 0 sec = 0 for track in tracklist: start, length = track min = min + length[0] sec = sec + length[1] min = min + sec / 60 sec = sec % 60 hash = hash + _dbid(min) + _dbid(sec) for i in range(nidtracks): start, length = tracklist[i] hash = hash + _dbid(length[0]) + _dbid(length[1]) return hash class Cddb: def __init__(self, tracklist): if os.environ.has_key('CDDB_PATH'): path = os.environ['CDDB_PATH'] cddb_path = path.split(',') else: home = os.environ['HOME'] cddb_path = [home + '/' + _cddbrc] self._get_id(tracklist) for dir in cddb_path: file = dir + '/' + self.id + '.rdb' try: f = open(file, 'r') self.file = file break except IOError: pass ntracks = int(self.id[:2], 16) self.artist = '' self.title = '' self.track = [None] + [''] * ntracks self.trackartist = [None] + [''] * ntracks self.notes = [] if not hasattr(self, 'file'): return import re reg = re.compile(r'^([^.]*)\.([^:]*):[\t ]+(.*)') while 1: line = f.readline() if not line: break match = reg.match(line) if not match: print 'syntax error in ' + file continue name1, name2, value = match.group(1, 2, 3) if name1 == 'album': if name2 == 'artist': self.artist = value elif name2 == 'title': self.title = value elif name2 == 'toc': if not self.toc: self.toc = value if self.toc != value: print 'toc\'s don\'t match' elif name2 == 'notes': self.notes.append(value) elif name1[:5] == 'track': try: trackno = int(name1[5:]) except strings.atoi_error: print 'syntax error in ' + file continue if trackno > ntracks: print 'track number %r in file %r out of range' % (trackno, file) continue if name2 == 'title': self.track[trackno] = value elif name2 == 'artist': self.trackartist[trackno] = value f.close() for i in range(2, len(self.track)): track = self.track[i] # if track title starts with `,', use initial part # of previous track's title if track and track[0] == ',': try: off = self.track[i - 1].index(',') except ValueError: pass else: self.track[i] = self.track[i-1][:off] \ + track def _get_id(self, tracklist): # fill in self.id and self.toc. # if the argument is a string ending in .rdb, the part # upto the suffix is taken as the id. if type(tracklist) == type(''): if tracklist[-4:] == '.rdb': self.id = tracklist[:-4] self.toc = '' return t = [] for i in range(2, len(tracklist), 4): t.append((None, \ (int(tracklist[i:i+2]), \ int(tracklist[i+2:i+4])))) tracklist = t ntracks = len(tracklist) self.id = _dbid((ntracks >> 4) & 0xF) + _dbid(ntracks & 0xF) if ntracks <= _DB_ID_NTRACKS: nidtracks = ntracks else: nidtracks = _DB_ID_NTRACKS - 1 min = 0 sec = 0 for track in tracklist: start, length = track min = min + length[0] sec = sec + length[1] min = min + sec / 60 sec = sec % 60 self.id = self.id + _dbid(min) + _dbid(sec) for i in range(nidtracks): start, length = tracklist[i] self.id = self.id + _dbid(length[0]) + _dbid(length[1]) self.toc = string.zfill(ntracks, 2) for track in tracklist: start, length = track self.toc = self.toc + string.zfill(length[0], 2) + \ string.zfill(length[1], 2) def write(self): import posixpath if os.environ.has_key('CDDB_WRITE_DIR'): dir = os.environ['CDDB_WRITE_DIR'] else: dir = os.environ['HOME'] + '/' + _cddbrc file = dir + '/' + self.id + '.rdb' if posixpath.exists(file): # make backup copy posix.rename(file, file + '~') f = open(file, 'w') f.write('album.title:\t' + self.title + '\n') f.write('album.artist:\t' + self.artist + '\n') f.write('album.toc:\t' + self.toc + '\n') for note in self.notes: f.write('album.notes:\t' + note + '\n') prevpref = None for i in range(1, len(self.track)): if self.trackartist[i]: f.write('track%r.artist:\t%s\n' % (i, self.trackartist[i])) track = self.track[i] try: off = track.index(',') except ValuError: prevpref = None else: if prevpref and track[:off] == prevpref: track = track[off:] else: prevpref = track[:off] f.write('track%r.title:\t%s\n' % (i, track)) f.close()