Lib/plat-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                                   (int(toc[i:i+2]),
  34                                    int(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 = path.split(',')
  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 = int(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 re
  85                 reg = re.compile(r'^([^.]*)\.([^:]*):[\t ]+(.*)')
  86                 while 1:
  87                         line = f.readline()
  88                         if not line:
  89                                 break
  90                         match = reg.match(line)
  91                         if not match:
  92                                 print 'syntax error in ' + file
  93                                 continue
  94                         name1, name2, value = match.group(1, 2, 3)
  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:
 109                                         trackno = int(name1[5:])
 110                                 except strings.atoi_error:
 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:
 129                                         off = self.track[i - 1].index(',')
 130                                 except ValueError:
 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, \
 148                                           (int(tracklist[i:i+2]), \
 149                                            int(tracklist[i+2:i+4]))))
 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:
 197                                 off = track.index(',')
 198                         except ValuError:
 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()