Lib/irix5/cddb.py

   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,
  33                                   (string.atoi(toc[i:i+2]),
  34                                    string.atoi(toc[i+2:i+4]))))
  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']
  61                         cddb_path = string.splitfields(path, ',')
  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
  76                 ntracks = string.atoi(self.id[:2], 16)
  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
  84                 import regex
  85                 reg = regex.compile('^\\([^.]*\\)\\.\\([^:]*\\):[\t ]+\\(.*\\)')
  86                 while 1:
  87                         line = f.readline()
  88                         if not line:
  89                                 break
  90                         if reg.match(line) == -1:
  91                                 print 'syntax error in ' + file
  92                                 continue
  93                         name1 = line[reg.regs[1][0]:reg.regs[1][1]]
  94                         name2 = line[reg.regs[2][0]:reg.regs[2][1]]
  95                         value = line[reg.regs[3][0]:reg.regs[3][1]]
  96                         if name1 == 'album':
  97                                 if name2 == 'artist':
  98                                         self.artist = value
  99                                 elif name2 == 'title':
 100                                         self.title = value
 101                                 elif name2 == 'toc':
 102                                         if not self.toc:
 103                                                 self.toc = value
 104                                         if self.toc != value:
 105                                                 print 'toc\'s don\'t match'
 106                                 elif name2 == 'notes':
 107                                         self.notes.append(value)
 108                         elif name1[:5] == 'track':
 109                                 try:
 110                                         trackno = string.atoi(name1[5:])
 111                                 except strings.atoi_error:
 112                                         print 'syntax error in ' + file
 113                                         continue
 114                                 if trackno > ntracks:
 115                                         print 'track number ' + `trackno` + \
 116                                                   ' in file ' + file + \
 117                                                   ' out of range'
 118                                         continue
 119                                 if name2 == 'title':
 120                                         self.track[trackno] = value
 121                                 elif name2 == 'artist':
 122                                         self.trackartist[trackno] = value
 123                 f.close()
 124                 for i in range(2, len(self.track)):
 125                         track = self.track[i]
 126                         # if track title starts with `,', use initial part
 127                         # of previous track's title
 128                         if track and track[0] == ',':
 129                                 try:
 130                                         off = string.index(self.track[i - 1],
 131                                                            ',')
 132                                 except string.index_error:
 133                                         pass
 134                                 else:
 135                                         self.track[i] = self.track[i-1][:off] \
 136                                                         + track
 137
 138         def _get_id(self, tracklist):
 139                 # fill in self.id and self.toc.
 140                 # if the argument is a string ending in .rdb, the part
 141                 # upto the suffix is taken as the id.
 142                 if type(tracklist) == type(''):
 143                         if tracklist[-4:] == '.rdb':
 144                                 self.id = tracklist[:-4]
 145                                 self.toc = ''
 146                                 return
 147                         t = []
 148                         for i in range(2, len(tracklist), 4):
 149                                 t.append((None, \
 150                                           (string.atoi(tracklist[i:i+2]), \
 151                                            string.atoi(tracklist[i+2:i+4]))))
 152                         tracklist = t
 153                 ntracks = len(tracklist)
 154                 self.id = _dbid((ntracks >> 4) & 0xF) + _dbid(ntracks & 0xF)
 155                 if ntracks <= _DB_ID_NTRACKS:
 156                         nidtracks = ntracks
 157                 else:
 158                         nidtracks = _DB_ID_NTRACKS - 1
 159                         min = 0
 160                         sec = 0
 161                         for track in tracklist:
 162                                 start, length = track
 163                                 min = min + length[0]
 164                                 sec = sec + length[1]
 165                         min = min + sec / 60
 166                         sec = sec % 60
 167                         self.id = self.id + _dbid(min) + _dbid(sec)
 168                 for i in range(nidtracks):
 169                         start, length = tracklist[i]
 170                         self.id = self.id + _dbid(length[0]) + _dbid(length[1])
 171                 self.toc = string.zfill(ntracks, 2)
 172                 for track in tracklist:
 173                         start, length = track
 174                         self.toc = self.toc + string.zfill(length[0], 2) + \
 175                                   string.zfill(length[1], 2)
 176
 177         def write(self):
 178                 import posixpath
 179                 if os.environ.has_key('CDDB_WRITE_DIR'):
 180                         dir = os.environ['CDDB_WRITE_DIR']
 181                 else:
 182                         dir = os.environ['HOME'] + '/' + _cddbrc
 183                 file = dir + '/' + self.id + '.rdb'
 184                 if posixpath.exists(file):
 185                         # make backup copy
 186                         posix.rename(file, file + '~')
 187                 f = open(file, 'w')
 188                 f.write('album.title:\t' + self.title + '\n')
 189                 f.write('album.artist:\t' + self.artist + '\n')
 190                 f.write('album.toc:\t' + self.toc + '\n')
 191                 for note in self.notes:
 192                         f.write('album.notes:\t' + note + '\n')
 193                 prevpref = None
 194                 for i in range(1, len(self.track)):
 195                         if self.trackartist[i]:
 196                                 f.write('track'+`i`+'.artist:\t'+self.trackartist[i]+'\n')
 197                         track = self.track[i]
 198                         try:
 199                                 off = string.index(track, ',')
 200                         except string.index_error:
 201                                 prevpref = None
 202                         else:
 203                                 if prevpref and track[:off] == prevpref:
 204                                         track = track[off:]
 205                                 else:
 206                                         prevpref = track[:off]
 207                         f.write('track' + `i` + '.title:\t' + track + '\n')
 208                 f.close()