def _lookup_cddb(self):
"""
returns a cdinfo() instance containing the information we found in
cddb, or None if that doesn't work. This method doesn't intentionally
throw any exceptions (unlike _lookup_mb) although I'm sure it's
possible.
"""
import DiscID, CDDB
if self.cddevice: dev = DiscID.open(self.cddevice)
else: dev = DiscID.open()
cdid = DiscID.disc_id(dev)
tracks = cdid[1]
(status, info) = CDDB.query(cdid)
if status == 200:
(status, info) = CDDB.read(info['category'], \
info['disc_id'])
elif status == 210 or status == 211:
(status, info) = CDDB.read(info[0]['category'], \
info[0]['disc_id'])
else:
return None
ret = cdinfo()
for key in info.keys():
if key.startswith('TTITLE'):
n = int(re.findall("TTITLE([0-9]+)", key)[0])
ret.titles[n] = info[key].encode('ascii', errors='replace')
elif key == 'DTITLE':
(artist, album) = info[key].split('/')
ret.artist = artist.strip()
ret.album = album.strip()
elif key == 'DYEAR':
ret.releasedate = info[key]
wx.LogMessage("cddb succeeded")
return ret
def _query_cddb(self, disc_id):
try:
(query_status, query_info) = CDDB.query(disc_id)
self.logger.debug("CDDB Query status: %s" % query_status)
except Exception as e:
self.logger.error(e)
query_status = 0
query_info = {}
# Exact match found
try:
if query_status == 200:
(read_status, read_info) = CDDB.read(query_info["category"],
query_info["disc_id"])
self.logger.debug("CDDB Read Status: %s" % read_status)
# Multiple matches found - pick first
elif query_status == 210 or query_status == 211:
(read_status, read_info) = CDDB.read(query_info[0]["category"],
query_info[0]["disc_id"])
self.logger.debug("CDDB Read Status: %s" % read_status)
# No match found
else:
self.logger.info("CD query failed, status: %s " % query_status)
except Exception as e:
self.logger.error(e)
read_status = 0
read_info = {}
if read_status != 210:
self.logger.info("CDDB read failed, status: %s" % read_status)
return read_info
def query(category, discid, xcode='utf8:utf8'):
#from quodlibet's cddb plugin by Michael Urman
discinfo = {}
tracktitles = {}
read, info = CDDB.read(category, discid, **CLIENTINFO)
if read != 210: return None
xf, xt = xcode.split(':')
for key, value in info.iteritems():
try:
value = value.decode('utf-8', 'replace').strip().encode(
xf, 'replace').decode(xt, 'replace')
except AttributeError:
pass
if key.startswith('TTITLE'):
try:
tracktitles[int(key[6:])] = value
except ValueError:
pass
elif key == 'DGENRE':
discinfo['genre'] = value
elif key == 'DTITLE':
dtitle = value.strip().split(' / ', 1)
if len(dtitle) == 2:
discinfo['artist'], discinfo['title'] = dtitle
else:
discinfo['title'] = dtitle[0].strip()
elif key == 'DYEAR':
discinfo['year'] = value
return discinfo, tracktitles
def get_cddb_info(self):
try:
disc = DiscID.open(self.device)
self.info = DiscID.disc_id(disc)
status, info = CDDB.query(self.info)
except IOError:
return
if status in (210, 211):
info = info[0]
status = 200
if status != 200:
return
(status, info) = CDDB.read(info["category"], info["disc_id"])
title = info["DTITLE"].split(" / ")
for i in range(self.info[1]):
tr = self[i]
tr.set_tag_raw("title", info["TTITLE" + str(i)].decode("iso-8859-15", "replace"))
tr.set_tag_raw("album", title[1].decode("iso-8859-15", "replace"))
tr.set_tag_raw("artist", title[0].decode("iso-8859-15", "replace"))
tr.set_tag_raw("year", info["EXTD"].replace("YEAR: ", ""))
tr.set_tag_raw("genre", info["DGENRE"])
self.name = title[1].decode("iso-8859-15", "replace")
event.log_event("cddb_info_retrieved", self, True)
def get_cddb(self):
try:
import CDDB, DiscID
except ImportError:
fatal('You need python-cddb (http://cddb-py.sf.net) to convert cds. Please install it.')
disc_id = DiscID.disc_id(DiscID.open(self.dev))
query_info = CDDB.query(disc_id)[1]
if not query_info:
fatal('The disk is not listed in FreeDB, dir2ogg only supports disk listed in MusicBrainz or FreeDB')
if isinstance(query_info, list):
query_info = query_info[0]
read_info = CDDB.read(query_info['category'], query_info['disc_id'])[1]
for track in range(disc_id[1]):
title = {}
title['discid'] = query_info['disc_id']
title['artist'], title['album'] = (track.strip() for track in query_info['title'].split("/"))
title['genre'] = read_info['DGENRE']
title['date'] = read_info['DYEAR']
title['title'] = read_info['TTITLE' + str(track)]
title['tracktotal'] = str(len(range(disc_id[1])) + 1)
title['ntracknumber'] = '0' * (len(title['tracktotal'] ) - len(str(track+1)) ) + str(track+1)
title['tracknumber'] = str(track+1)
for key, val in title.items():
title[key] = unicode(str(val), "ISO-8859-1")
ConvertTrack(self.dev, self.conf, track+1, title)
def get_cddb_info(self):
try:
disc = DiscID.open(self.device)
self.info = DiscID.disc_id(disc)
status, info = CDDB.query(self.info)
except IOError:
return
if status in (210, 211):
info = info[0]
status = 200
if status != 200:
return
(status, info) = CDDB.read(info['category'], info['disc_id'])
title = info['DTITLE'].split(" / ")
for i in range(self.info[1]):
tr = self[i]
tr.set_tag_raw('title',
info['TTITLE' + `i`].decode('iso-8859-15', 'replace'))
tr.set_tag_raw('album',
title[1].decode('iso-8859-15', 'replace'))
tr.set_tag_raw('artist',
title[0].decode('iso-8859-15', 'replace'))
tr.set_tag_raw('year',
info['EXTD'].replace("YEAR: ", ""))
tr.set_tag_raw('genre',
info['DGENRE'])
self.name = title[1].decode('iso-8859-15', 'replace')
event.log_event('cddb_info_retrieved', self, True)
def get_cddb(self):
try:
import CDDB, DiscID
except ImportError:
fatal('You need python-cddb (http://cddb-py.sf.net) to convert cds. Please install it.')
disc_id = DiscID.disc_id(DiscID.open(self.dev))
query_info = CDDB.query(disc_id)[1]
if not query_info:
fatal('The disk is not listed in FreeDB, dir2ogg only supports disk listed in MusicBrainz or FreeDB')
if isinstance(query_info, list):
query_info = query_info[0]
read_info = CDDB.read(query_info['category'], query_info['disc_id'])[1]
for track in range(disc_id[1]):
title = {}
title['discid'] = query_info['disc_id']
title['artist'], title['album'] = (track.strip() for track in query_info['title'].split("/"))
title['genre'] = read_info['DGENRE']
title['date'] = read_info['DYEAR']
title['title'] = read_info['TTITLE' + str(track)]
title['tracktotal'] = str(len(range(disc_id[1])) + 1)
title['ntracknumber'] = '0' * (len(title['tracktotal'] ) - len(str(track+1)) ) + str(track+1)
title['tracknumber'] = str(track+1)
for key, val in title.items():
title[key] = unicode(str(val), "ISO-8859-1")
ConvertTrack(self.dev, self.conf, track+1, title)
def query(category, discid, xcode='utf8:utf8'):
#from quodlibet's cddb plugin by Michael Urman
discinfo = {}
tracktitles = {}
read, info = CDDB.read(category, discid, **CLIENTINFO)
if read != 210: return None
xf, xt = xcode.split(':')
for key, value in info.iteritems():
try: value = value.decode('utf-8', 'replace').strip().encode(
xf, 'replace').decode(xt, 'replace')
except AttributeError: pass
if key.startswith('TTITLE'):
try: tracktitles[int(key[6:])] = value
except ValueError: pass
elif key == 'DGENRE': discinfo['genre'] = value
elif key == 'DTITLE':
dtitle = value.strip().split(' / ', 1)
if len(dtitle) == 2:
discinfo['artist'], discinfo['title'] = dtitle
else: discinfo['title'] = dtitle[0].strip()
elif key == 'DYEAR': discinfo['year'] = value
return discinfo, tracktitles
def get_cddb_info(self):
try:
disc = DiscID.open(self.device)
self.info = DiscID.disc_id(disc)
status, info = CDDB.query(self.info)
except IOError:
return
if status in (210, 211):
info = info[0]
status = 200
if status != 200:
return
(status, info) = CDDB.read(info['category'], info['disc_id'])
title = info['DTITLE'].split(" / ")
for i in range(self.info[1]):
tr = self[i]
tr.set_tag_raw(
'title', info['TTITLE' + ` i `].decode('iso-8859-15',
'replace'))
tr.set_tag_raw('album', title[1].decode('iso-8859-15', 'replace'))
tr.set_tag_raw('artist', title[0].decode('iso-8859-15', 'replace'))
tr.set_tag_raw('year', info['EXTD'].replace("YEAR: ", ""))
tr.set_tag_raw('genre', info['DGENRE'])
self.name = title[1].decode('iso-8859-15', 'replace')
event.log_event('cddb_info_retrieved', self, True)
def cddbRequest(self, discInfo):
""" Return disc information from online CDDB, None if request fails """
import CDDB, socket
# Make sure to not be blocked by the request
socket.setdefaulttimeout(consts.socketTimeout)
try:
(status, info) = CDDB.query(discInfo)
if status == 200:
disc = info # Success
elif status == 210:
disc = info[0] # Exact multiple matches
elif status == 211:
disc = info[0] # Inexact multiple matches
else:
raise Exception, "Unknown disc (phase 1 returned %u)" % status
(status, info) = CDDB.read(disc["category"], disc["disc_id"])
if status == 210:
return info
else:
raise Exception, "Unknown disc (phase 2 returned %u)" % status
except:
logger.error("[%s] CDDB request failed\n\n%s" % (MOD_INFO[modules.MODINFO_NAME], traceback.format_exc()))
return None
def get_CDDB_tag(self, query_status, query_info):
if self.is_audio_cd:
read_status, read_info = CDDB.read(query_info['category'], query_info['disc_id'])
print "read status: ", read_status
print "read info: ", read_info
else:
return
self.song_list = []
if query_status in [200, 210]:
self.is_audio_cd = True
self.cddb_disc_id_b10 = self.disc_id[0]
self.cddb_disc_id_b16 = query_info['disc_id']
self.total_tracks = self.disc_id[1]
if read_status == 210:
for i in range(self.disc_id[1]):
n = "%.02d" %(i + 1)
title = "%s" %(read_info['TTITLE' + `i`])
frame = self.disc_id[i+2]
self.song_list.append({"track_number":int(n), "title":title, "num_frame":frame})
print "Song_list: ", self.song_list
if read_info['DTITLE']:
disc_title = read_info['DTITLE']
else:
disc_title = query_info[0]['title']
self.artist, self.album = re.split(" / ", disc_title)
if "," in self.artist:
a, b = re.split(", ", self.artist)
self.artist = b.title() + " " + a.title()
try:
self.disc_len = read_info['disc_len']
except:
self.disc_len = self.disc_id[2+self.disc_id[1]]
self.year = read_info['DYEAR']
if read_info['DGENRE']:
self.cddb_genre = read_info['DGENRE']
else:
self.cddb_genre = query_info[0]['category']
elif read_status in [401, 402, 403, 409, 417]:
self.error = read_status
else:
return
elif query_status in [211, 202, 403, 409]:
self.error = query_status
try:
cdaudio.close()
except:
pass
return self.error
else:
cdaudio.close()
return
def query(category, discid, xcode='utf8:utf8'):
discinfo = {}
tracktitles = {}
dump = path.join(path.expanduser("~"), '.cddb', category, discid)
try:
for line in file(dump):
if line.startswith("TTITLE"):
track, title = line.split("=", 1)
try: track = int(track[6:])
except (ValueError): pass
else: tracktitles[track] = \
title.decode('utf-8', 'replace').strip()
elif line.startswith("DGENRE"):
discinfo['genre'] = line.split('=', 1)[1].strip()
elif line.startswith("DTITLE"):
dtitle = line.split('=', 1)[1].strip().split(' / ', 1)
if len(dtitle) == 2:
discinfo['artist'], discinfo['title'] = dtitle
else:
discinfo['title'] = dtitle[0].strip()
elif line.startswith("DYEAR"):
discinfo['year'] = line.split('=', 1)[1].strip()
except EnvironmentError: pass
else: return discinfo, tracktitles
read, info = CDDB.read(category, discid, **CLIENTINFO)
if read != 210: return None
try: os.makedirs(path.join(path.expanduser("~"), '.cddb'))
except EnvironmentError: pass
try:
save = file(dump, 'w')
keys = info.keys()
keys.sort()
for key in keys:
print>>save, "%s=%s" % (key, info[key])
save.close()
except EnvironmentError: pass
xf, xt = xcode.split(':')
for key, value in info.iteritems():
try: value = value.decode('utf-8', 'replace').strip().encode(
xf, 'replace').decode(xt, 'replace')
except AttributeError: pass
if key.startswith('TTITLE'):
try: tracktitles[int(key[6:])] = value
except ValueError: pass
elif key == 'DGENRE': discinfo['genre'] = value
elif key == 'DTITLE':
dtitle = value.strip().split(' / ', 1)
if len(dtitle) == 2:
discinfo['artist'], discinfo['title'] = dtitle
else: discinfo['title'] = dtitle[0].strip()
elif key == 'DYEAR': discinfo['year'] = value
return discinfo, tracktitles
def set_category(self, cat):
self._clear_md()
cache_path = os.path.join(self.cache_dir, '{0}-{1}'.format(self.fingerprint, cat))
for k,v in cached(lambda:CDDB.read(cat, self.fingerprint)[1])(cache_path).items():
if not v or k[0] not in 'DTE':
continue
elif k == 'DTITLE':
self.metadata[0]['artist'],_,self.metadata[0]['title'] = v.partition(' / ')
elif k == 'DGENRE': self.metadata[0]['genre'] = v
elif k == 'DYEAR': self.metadata[0]['year'] = v
elif k.startswith('TTITLE'): self.metadata[int(k[6:])+1]['title'] = v
def query(self):
def fixchars(pathname):
"""Fixes characters that cause escaping problems in string
constructions used to execute commands on the OS or in the DB"""
pathname = pathname.replace(os.sep, "-")
pathname = pathname.replace("\"", "")
pathname = pathname.replace("?", "")
return pathname
def padNum(num):
if int(num) < 10:
return "0" + str(num)
else:
return str(num)
delims = ("/", " - ")
def findTitle(title, count):
"""Tries the delimiters in order until it finds the right one"""
#print title
#print len(delims)
if count < len(delims):
titleparts = title.split(delims[count])
#print titleparts
if len(titleparts) == 2:
self.artistName = fixchars(titleparts[0].strip())
self.albumName = fixchars(titleparts[1].strip())
return True
else:
return findTitle(title, count+1)
return False
self.category = fixchars(self.query_info["category"])
self.disc_id = self.query_info["disc_id"]
self.title = fixchars(self.query_info["title"])
if not findTitle(self.title, 0):
raise "Title not parsed: %s" % self.title
(track_status, track_info) = CDDB.read(self.category, self.disc_id)
self.genre = fixchars(track_info['DGENRE'])
self.year = track_info['DYEAR']
if track_info.has_key('disc_len'):
self.albumlength = str(float(track_info['disc_len']) / 60) + " min."
else:
self.albumlength = ""
self.comments = fixchars(track_info['EXTD'])
self.albumtitle = fixchars(track_info['DTITLE'])
for (tracknum,trackname) in track_info.items():
if tracknum.find("TTITLE") != -1:
num = int(tracknum[len("TTITLE"):]) + 1
self.tracks[padNum(num)] = fixchars(trackname)
def _lookup_cddb(self):
"""
returns a cdinfo() instance containing the information we found in
cddb, or None if that doesn't work. This method doesn't intentionally
throw any exceptions (unlike _lookup_mb) although I'm sure it's
possible.
"""
import DiscID, CDDB
if self.cddevice:
dev = DiscID.open(self.cddevice)
else:
dev = DiscID.open()
cdid = DiscID.disc_id(dev)
tracks = cdid[1]
(status, info) = CDDB.query(cdid)
if status == 200:
(status, info) = CDDB.read(info["category"], info["disc_id"])
elif status == 210 or status == 211:
(status, info) = CDDB.read(info[0]["category"], info[0]["disc_id"])
else:
return None
ret = cdinfo()
for key in info.keys():
if key.startswith("TTITLE"):
n = int(re.findall("TTITLE([0-9]+)", key)[0])
ret.titles[n] = info[key].encode("ascii", errors="replace")
elif key == "DTITLE":
(artist, album) = info[key].split("/")
ret.artist = artist.strip()
ret.album = album.strip()
elif key == "DYEAR":
ret.releasedate = info[key]
wx.LogMessage("cddb succeeded")
return ret
def extractAudioCdInfo(self):
""" extract the CD info (album art + artist + tracks), and construct the UPnP items"""
self.cdrom = DiscID.open(self.device_name)
disc_id = DiscID.disc_id(self.cdrom)
(query_status, query_info) = CDDB.query(disc_id)
if query_status in (210, 211):
query_info = query_info[0]
(read_status, read_info) = CDDB.read(query_info['category'],
query_info['disc_id'])
# print query_info['title']
# print disc_id[1]
# for i in range(disc_id[1]):
# print "Track %.02d: %s" % (i, read_info['TTITLE' + `i`])
track_count = disc_id[1]
disc_id = query_info['disc_id']
self.disc_title = query_info['title'].encode('utf-8')
tracks = {}
for i in range(track_count):
tracks[i +
1] = read_info['TTITLE' +
` i `].decode('ISO-8859-1').encode('utf-8')
self.name = self.disc_title
root_item = Container(None, self.disc_title)
# we will sort the item by "track_number"
def childs_sort(x, y):
return cmp(x.track_number, y.track_number)
root_item.sorting_method = childs_sort
self.set_root_item(root_item)
for number, title in tracks.items():
item = TrackItem(self.device_name, number, "Unknown", title)
external_id = "%s_%d" % (disc_id, number)
root_item.add_child(item, external_id=external_id)
self.info('Sharing audio CD %s' % self.disc_title)
reactor.callLater(2, self.checkIfAudioCdStillPresent)
self.init_completed()
def printDisc(disc_info):
category = disc_info["category"]
disc_id = disc_info["disc_id"]
title = disc_info["title"]
(track_status, track_info) = CDDB.read(category, disc_id)
print "%s\t(%s)" % (title,category)
tracks = {}
for (k,v) in track_info.items():
if k.find("TTITLE") != -1:
num = int(k[len("TTITLE"):]) + 1
tracks[num] = v
tracks.items().sort()
for (k,v) in tracks.items():
print "%s. %s" % (k,v)
def run(self):
(status, info) = CDDB.query(self._disc_id,
client_name="dmusic",
client_version="1.0")
if status in [200, 210, 211]:
if status in [210, 211]:
info = sorted(info, key=lambda w: w["category"])
info = info[0]
else:
self.logdebug('Unable to fetch CDDB info, status=%d', status)
(status, info) = CDDB.read(info['category'], info['disc_id'])
if status != 210:
self.logdebug('Unable to fetch CDDB info, status=%d', status)
return
gobject.idle_add(self._cb, self._disc_id, info)
def main():
dev = None
cdrom = None
if len(sys.argv) >= 2:
dev = sys.argv[1]
if dev:
cdrom = DiscID.open(dev)
else:
cdrom = DiscID.open()
print "Getting disc id in CDDB format...",
disc_id = DiscID.disc_id(cdrom)
print "Disc ID: %08lx Num tracks: %d" % (disc_id[0], disc_id[1])
print "Querying CDDB for info on disc...",
(query_stat, query_info) = CDDB.query(disc_id)
if query_stat == 200:
print("success!\nQuerying CDDB for track info of `%s'... " %
query_info['title']),
(read_stat, read_info) = CDDB.read(query_info['category'],
query_info['disc_id'])
if read_stat == 210:
print "success!"
# Start from 0, not 1
# thanks to bgp for the fix!
for i in range(0, disc_id[1]):
print "Track %.02d: %s" % (i + 1,
read_info['TTITLE' + repr(i)])
else:
print "failure getting track info, status: %i" % read_stat
elif query_stat == 210 or query_stat == 211:
print "multiple matches found! Matches are:"
for i in query_info:
print "ID: %s Category: %s Title: %s" % \
(i['disc_id'], i['category'], i['title'])
else:
print "failure getting disc info, status %i" % query_stat
def extractAudioCdInfo(self):
'''
extract the CD info (album art + artist + tracks),
and construct the UPnP items
'''
self.cdrom = DiscID.open(self.device_name)
disc_id = DiscID.disc_id(self.cdrom)
(query_status, query_info) = CDDB.query(disc_id)
if query_status in (210, 211):
query_info = query_info[0]
(read_status, read_info) = CDDB.read(query_info['category'],
query_info['disc_id'])
# print(query_info['title'])
# print(disc_id[1])
# for i in range(disc_id[1]):
# print(f'Track {i:.02d}: {read_info["TTITLE" + repr(i)]}')
track_count = disc_id[1]
disc_id = query_info['disc_id']
self.disc_title = query_info['title'].encode('utf-8')
tracks = {}
for i in range(track_count):
tracks[i + 1] = (read_info[f'TTITLE{repr(i)}'].decode(
'ISO-8859-1').encode('utf-8'))
self.name = self.disc_title
root_item = Container(None, self.disc_title)
root_item.sorting_method = 'track_number'
self.set_root_item(root_item)
for number, title in list(tracks.items()):
item = TrackItem(self.device_name, number, 'Unknown', title)
external_id = f'{disc_id}_{number:d}'
root_item.add_child(item, external_id=external_id)
self.info('Sharing audio CD %s', self.disc_title)
reactor.callLater(2, self.checkIfAudioCdStillPresent)
self.init_completed()
def refresh(self):
self.tracks = []
try:
self.disc = discid.read()
except:
logger.debug("Cdrom: Unable to read cd")
return
logger.debug("Cdrom: reading cd")
self.n = len(self.disc.tracks)
logger.debug('Cdrom: %d tracks found', self.n)
read_info = {}
try:
self.disc_id = DiscID.disc_id(DiscID.open())
(query_status, query_info) = CDDB.query(self.disc_id)
(read_status, read_info) = CDDB.read(query_info['category'],
query_info['disc_id'])
except:
pass
if 'DYEAR' in read_info:
self.year = read_info['DYEAR']
else:
self.year = ''
if 'DGENRE' in read_info:
self.genre = read_info['DGENRE']
else:
self.genre = 'unknown'
if 'DTITLE' in read_info:
self.albumtitle = self.sanitizeString(read_info['DTITLE'])
else:
self.albumtitle = 'CD'
for track in self.disc.tracks:
number = track.number
duration = track.seconds
key = 'TTITLE' + repr((track.number - 1))
if key in read_info:
name = self.sanitizeString(read_info[key])
else:
name = 'Cdrom Track %s (%s)' % (
number, time.strftime('%H:%M:%S', time.gmtime(duration)))
self.tracks.append((number, name, duration, self.albumtitle,
self.genre, self.year))
def extractAudioCdInfo (self):
""" extract the CD info (album art + artist + tracks), and construct the UPnP items"""
self.cdrom = DiscID.open(self.device_name)
disc_id = DiscID.disc_id(self.cdrom)
(query_status, query_info) = CDDB.query(disc_id)
if query_status in (210, 211):
query_info = query_info[0]
(read_status, read_info) = CDDB.read(query_info['category'], query_info['disc_id'])
# print query_info['title']
# print disc_id[1]
# for i in range(disc_id[1]):
# print "Track %.02d: %s" % (i, read_info['TTITLE' + `i`])
track_count = disc_id[1]
disc_id = query_info['disc_id']
self.disc_title = query_info['title'].encode('utf-8')
tracks = {}
for i in range(track_count):
tracks[i + 1] = read_info['TTITLE' + `i`].decode('ISO-8859-1').encode('utf-8')
self.name = self.disc_title
root_item = Container(None, self.disc_title)
# we will sort the item by "track_number"
def childs_sort(x, y):
return cmp(x.track_number, y.track_number)
root_item.sorting_method = childs_sort
self.set_root_item(root_item)
for number, title in tracks.items():
item = TrackItem(self.device_name, number, "Unknown", title)
external_id = "%s_%d" % (disc_id, number)
root_item.add_child(item, external_id=external_id)
self.info('Sharing audio CD %s', self.disc_title)
reactor.callLater(2, self.checkIfAudioCdStillPresent)
self.init_completed()
def saveDisc(disc_info):
category = disc_info["category"]
disc_id = disc_info["disc_id"]
title = disc_info["title"]
(artistName, albumName) = disc_info["title"].split("/")
artistName = artistName.strip()
albumName = albumName.strip()
#Search for existing artist
artists = Artist.selectBy(name=artistName)
if artists.count() == 0:
#Create artist
artist = Artist(name=artistName,category=category)
else:
#Use existing artist
artist = artists[0]
#Search for existing album
albums = Album.selectBy(name=albumName)
if albums.count() == 0:
#Create album
album = Album(disc_id=disc_id,name=albumName,artist=artist)
#Create tracks
(track_status, track_info) = CDDB.read(category, disc_id)
tracks = {}
for (k,v) in track_info.items():
if k.find("TTITLE") != -1:
num = int(k[len("TTITLE"):]) + 1
tracks[num] = v
tracks.items().sort()
for (k,v) in tracks.items():
track = Track(num=k,name=v,album=album)
print "***Saved in catalogue***"
else:
print "***Already catalogued***"
printDisc(disc_info)
def _get_information_from_result_element(self, result, disc_id):
"""Catch any information from a CDDB result dictionnary"""
title = unicode(result['title'], "iso-8859-1")
category = unicode(result['category'], "iso-8859-1")
disc = unicode(result['disc_id'], "iso-8859-1")
self.artist = title[:title.index(' / ')]
self.title = title[title.index(' / ') + 3:]
# Get track titles
info = CDDB.read(category, disc)[1]
for i in range(disc_id[1]):
if i + 4 == len(disc_id):
# We must calculate last track length in a different way
length = disc_id[len(disc_id) - 1] - self.length
else:
# Calculate track length in seconds
length = (disc_id[i + 3] - disc_id[ i + 2]) / 75
self.length += length
track_title = unicode(info['TTITLE' + str(i)], "iso-8859-1")
self.tracks.append(CompactDiscTrack(i + 1, track_title, length))
def __init__(self):
self.good = True
cdrom = DiscID.open("/dev/sr0")
self.disc_id = DiscID.disc_id(cdrom)
self.disc_info = []
(status,info) = CDDB.query(self.disc_id)
if status == 200:
info = [info]
elif status != 211 and status != 210:
lnp("CDDB: No matches found")
self.good = False
return
for i in info:
(status, read_info) = (None,None)
for j in range(3):
(status,read_info) = CDDB.read(i['category'],i['disc_id'])
if status != 210:
continue
else:
break
if status == 210:
read_info['DISCID'] = i['disc_id']
self.disc_info.append(read_info)
def get_cd_info():
def check_status(status, success_values):
cddb_status_msg = {
200: "Success",
211: "Multiple inexact matches were found",
210: "Multiple exact matches were found",
202: "No match found",
403: "Error: database entry is corrupt",
409: "Error: no handshake. (client-side error?)"
}
debug("CDDB status: %d (%s)"%(status, cddb_status_msg[status]))
if status not in success_values:
s = ", ".join(success_values)
debug("CDDB status error: %d (success: %s)" % (status, s))
return False
return True
cdrom = DiscID.open()
disc_id = DiscID.disc_id(cdrom)
debug("CDDB: Tracks in CD: %d"%disc_id[1])
qstatus, qinfo = CDDB.query(disc_id)
if not check_status(qstatus, (200, 210)):
tracks = ["track%02d"%n for n in xrange(1, disc_id[1]+1)]
return "Unknown - Unknown", tracks
if isinstance(qinfo, list):
qinfo = qinfo[0]
debug("CDDB: CD-ID: %s" % qinfo["disc_id"])
status, info = CDDB.read(qinfo['category'], qinfo['disc_id'])
check_status(status, (200, 210))
year = info.get("DYEAR")
disc_title = year and "%s (%s)"%(qinfo["title"], year) or qinfo["title"]
debug("CDDB: Disc title: %s"%disc_title)
recode = lambda s: s.decode("iso8859-1").encode("utf-8")
get = lambda num: recode(info['TTITLE%d'%num]).title()
title = recode(disc_title).title()
tracks = map(get, range(disc_id[1]))
return title, tracks
def CDDB_lookup():
"""
Queries the CDDB for the album
"""
devices = cdrdao.scan_devices()
discid = cdrdao.get_discid(devices[0][0])
(query_status, query_info) = CDDB.query(discid)
if query_status != 200:
print "Couldn't find match on the freedb"
return None
(read_status, read_info) = CDDB.read(query_info['category'], query_info['disc_id'])
tracks = []
for i in range(int(discid[1])):
tracks.append({'number':i+1, 'title': cddb_normalize(read_info["TTITLE{}".format(i)])})
artist, album = read_info['DTITLE'].split('/',1)
artist = cddb_normalize(artist.strip())
album = cddb_normalize(album.strip())
obj = {'artist': artist, 'album': album, 'tracks':tracks}
return obj
def isDisc(self, device):
if discinfo.DiscInfo.isDisc(self, device) != 1:
return 0
disc_id = DiscID.disc_id(device)
if mmpython.USE_NETWORK:
try:
(query_stat, query_info) = CDDB.query(disc_id)
except:
# Oops no connection
query_stat = 404
else:
query_stat = 404
if query_stat == 210 or query_stat == 211:
# set this to success
query_stat = 200
for i in query_info:
if i['title'] != i['title'].upper():
query_info = i
break
else:
query_info = query_info[0]
elif query_stat != 200:
_debug("failure getting disc info, status %i" % query_stat)
if query_stat == 200:
qi = query_info['title'].split('/')
self.artist = qi[0].strip()
self.title = qi[1].strip()
for type in ('title', 'artist'):
if getattr(self, type) and getattr(self, type)[0] in ('"', '\'') \
and getattr(self, type)[-1] in ('"', '\''):
setattr(self, type, getattr(self, type)[1:-1])
(read_stat, read_info) = CDDB.read(query_info['category'],
query_info['disc_id'])
# id = disc_id + number of tracks
#self.id = '%s_%s' % (query_info['disc_id'], disc_id[1])
if read_stat == 210:
for i in range(0, disc_id[1]):
mi = mediainfo.MusicInfo()
mi.title = read_info['TTITLE' + `i`]
mi.album = self.title
mi.artist = self.artist
mi.genre = query_info['category']
mi.codec = 'PCM'
mi.samplerate = 44.1
mi.trackno = i+1
mi.trackof = disc_id[1]
self.tracks.append(mi)
for type in ('title', 'album', 'artist', 'genre'):
if getattr(mi, type) and getattr(mi, type)[0] in ('"', '\'') \
and getattr(mi, type)[-1] in ('"', '\''):
setattr(mi, type, getattr(mi, type)[1:-1])
else:
_debug("failure getting track info, status: %i" % read_stat)
# set query_stat to somthing != 200
query_stat = 400
if query_stat != 200:
_debug("failure getting disc info, status %i" % query_stat)
self.no_caching = 1
for i in range(0, disc_id[1]):
mi = mediainfo.MusicInfo()
mi.title = 'Track %s' % (i+1)
mi.codec = 'PCM'
mi.samplerate = 44.1
mi.trackno = i+1
mi.trackof = disc_id[1]
self.tracks.append(mi)
# read the tracks to generate the title list
device = open(device)
(first, last) = cdrom.toc_header(device)
lmin = 0
lsec = 0
num = 0
for i in range(first, last + 2):
if i == last + 1:
min, sec, frames = cdrom.leadout(device)
else:
min, sec, frames = cdrom.toc_entry(device, i)
if num:
self.tracks[num-1].length = (min-lmin)*60 + (sec-lsec)
num += 1
lmin, lsec = min, sec
device.close()
# correct bad titles for the tracks, containing also the artist
for t in self.tracks:
if not self.artist or not t.title.startswith(self.artist):
break
else:
for t in self.tracks:
t.title = t.title[len(self.artist):].lstrip('/ \t-_')
# correct bad titles for the tracks, containing also the title
for t in self.tracks:
if not self.title or not t.title.startswith(self.title):
break
else:
for t in self.tracks:
t.title = t.title[len(self.title):].lstrip('/ \t-_')
return 1
def getDiscID(self):
discid = ((self.checksum % 0xff) << 24 | int(self.length - self.PRESEC) << 8 | self.tracks)
return discid
def cddb_sum(self, n):
"Calculate checksum"
sum = 0
while n > 0:
sum = sum + (n % 10)
n = n / 10
return sum
if __name__ == '__main__':
renamer = mp3cddb(sys.argv[1])
disc_id = renamer.getcddbformat()
import CDDB
(status, info) = CDDB.query(disc_id)
status2 = None
info2 = None
if status == 200:
(status2, info2) = CDDB.read(info['category'], info['disc_id'])
elif status == 210 or status == 211:
(status2, info2) = CDDB.read(info[0]['category'], info[0]['disc_id'])
if status == 202:
print "No Match Found!"
if info2:
renamer.updateNames(info2)
# -*- coding: utf-8 -*-
# Convert CDROM content to WAV
import subprocess
import CDDB, DiscID
cdrom = DiscID.open()
disc_id = DiscID.disc_id(cdrom)
(query_status, query_info) = CDDB.query(disc_id)
(read_status, read_info) = CDDB.read(query_info["category"], query_info["disc_id"])
for i in range(disc_id[1]):
name = "%s.wav" % (read_info["TTITLE" + ` i `])
name = name.replace("\xba", ".")
name = name.replace('"', "'")
name = name.replace(",", "")
name = name.replace(":", "")
if i < 9:
mplayer_cmd = 'mplayer cdda://%d -ao pcm:file="0%d - %s" ' % (i + 1, i + 1, name)
else:
mplayer_cmd = 'mplayer cdda://%d -ao pcm:file="%d - %s" ' % (i + 1, i + 1, name)
print mplayer_cmd
# subprocess.call(mplayer_cmd,shell=True)
cdrom = DiscID.open()
print "Getting disc id in CDDB format...",
disc_id = DiscID.disc_id(cdrom)
print "Disc ID: %08lx Num tracks: %d" % (disc_id[0], disc_id[1])
print "Querying CDDB for info on disc...",
(query_stat, query_info) = CDDB.query(disc_id)
if query_stat == 200:
print ("success!\nQuerying CDDB for track info of `%s'... " %
query_info['title']),
(read_stat, read_info) = CDDB.read(query_info['category'],
query_info['disc_id'])
if read_stat == 210:
print "success!"
# Start from 0, not 1
# thanks to bgp for the fix!
for i in range(0, disc_id[1]):
print "Track %.02d: %s" % (i+1, read_info['TTITLE' + `i`])
else:
print "failure getting track info, status: %i" % read_stat
elif query_stat == 210 or query_stat == 211:
print "multiple matches found! Matches are:"
for i in query_info:
print "ID: %s Category: %s Title: %s" % \
(i['disc_id'], i['category'], i['title'])
def get_CDDB_tag(self, query_status, query_info):
if self.is_audio_cd:
read_status, read_info = CDDB.read(query_info['category'],
query_info['disc_id'])
print "read status: ", read_status
print "read info: ", read_info
else:
return
self.song_list = []
if query_status in [200, 210]:
self.is_audio_cd = True
self.cddb_disc_id_b10 = self.disc_id[0]
self.cddb_disc_id_b16 = query_info['disc_id']
self.total_tracks = self.disc_id[1]
if read_status == 210:
for i in range(self.disc_id[1]):
n = "%.02d" % (i + 1)
title = "%s" % (read_info['TTITLE' + ` i `])
frame = self.disc_id[i + 2]
self.song_list.append({
"track_number": int(n),
"title": title,
"num_frame": frame
})
print "Song_list: ", self.song_list
if read_info['DTITLE']:
disc_title = read_info['DTITLE']
else:
disc_title = query_info[0]['title']
self.artist, self.album = re.split(" / ", disc_title)
if "," in self.artist:
a, b = re.split(", ", self.artist)
self.artist = b.title() + " " + a.title()
try:
self.disc_len = read_info['disc_len']
except:
self.disc_len = self.disc_id[2 + self.disc_id[1]]
self.year = read_info['DYEAR']
if read_info['DGENRE']:
self.cddb_genre = read_info['DGENRE']
else:
self.cddb_genre = query_info[0]['category']
elif read_status in [401, 402, 403, 409, 417]:
self.error = read_status
else:
return
elif query_status in [211, 202, 403, 409]:
self.error = query_status
try:
cdaudio.close()
except:
pass
return self.error
else:
cdaudio.close()
return
#!/usr/bin/python
# vim: fileencoding=utf-8
import CDDB, DiscID
cdrom = DiscID.open()
disc_id = DiscID.disc_id(cdrom)
#disc_id = [2417058606, 46, 150, 7370, 13269, 19460, 25947, 32305, 39009, 46232, 52252, 57642, 65161, 71936, 78422, 84966, 92375, 99055, 108136, 114731, 119543, 126023, 134664, 141327, 150150, 160145, 167527, 175022, 181651, 186421, 195267, 202583, 209452, 216932, 222989, 225177, 227313, 232838, 240578, 247447, 258438, 264953, 269940, 277132, 286799, 296798, 305398, 310924, 4453]
(query_status, query_info) = CDDB.query(disc_id)
category = query_info['category']
discid = query_info['disc_id']
(read_status, read_info) = CDDB.read(category, discid)
print read_info
exit(0)
(category, disc_title) = read_info['DTITLE'].split(' / ')
year = read_info['DYEAR']
genre = read_info['DGENRE']
for i in range(disc_id[1]):
(title, vocal) = read_info['TTITLE' + `i`].split(' / ')
print "%.02d|%s|%s|%s|%s|%s" % (i+1, title, vocal,disc_title,year,genre)
def getCddbDiscInfoRemote(disc_id):
if debug:
print "Querying CDDB for info on disc...",
(query_stat, query_info) = CDDB.query(disc_id)
disc_info = None
track_info = None
if query_stat == 210 or query_stat == 211:
disc_info = query_info[0]
#print "multiple matches found! "
if debug:
print "multiple matches found! Matches are:"
for i in query_info:
print "ID: %s Category: %s Title: %s" % \
(i['disc_id'], i['category'], i['title'])
elif query_stat == 200:
if debug:
print ("success, found single disc entry!")
disc_info = query_info
else:
if debug:
print "failure getting disc info, status %i" % query_stat
#print repr(disc_info['title'])
#print cleanPath(unicode(disc_info['title'],ENCODING))
info = {}
info['type'] = "CD"
info['discId'] = "%x" % disc_id[0]
info['nrTracks'] = disc_id[1]
if not disc_info:
info['discIdCddb'] = ''
info['category'] = ''
info['artist'] = info['discId']
info['album'] = 'unknown'
info['year'] = ""
info['genre'] = ""
for i in range(0, disc_id[1]):
info ["artist%d" % i] = info['discId']
info ["track%d" % i] = "Track %02d" % (i+1)
else:
info['discIdCddb'] = disc_info['disc_id']
info['category'] = disc_info['category']
#m = re.match(u"(.*)\s+/(\s+(.*))?", disc_info['title'])
#artist = m.group(1)
#try:
# album = m.group(3)
#except:
# album = "unknown"
m = disc_info['title'].find("/")
if m != -1:
artist = disc_info['title'][:m].strip()
album = disc_info['title'][m+1:].strip()
else:
artist = disc_info['title'].strip()
album = "unknown"
info['artist'] = artist
info['album'] = album
info['year'] = ""
info['genre'] = ""
if debug:
print "USING ID: %s, Category: %s, Title: %s" % \
(info['discIdCddb'], info['category'],
disc_info['title'])
track_stat = None
track_info = None
try:
(track_stat, track_info) = CDDB.read(disc_info['category'],
disc_info['disc_id'])
except:
trac_stat = -1
if track_stat == 210:
if debug:
print "success getting track info!"
try:
info['year'] = track_info['DYEAR']
except:
pass
try:
info['genre'] = track_info['DGENRE']
except:
pass
for i in range(0, disc_id[1]):
# update info structure and handle samplers where title
# encode actual artist & title
oTitle = track_info['TTITLE' + `i`]
m = oTitle.find("/")
if m != -1:
artist = oTitle[:m].strip()
title = oTitle[m+1:].strip()
else:
artist = info['artist']
title = oTitle
info["artist%d" % i] = artist
info["track%d" % i] = title
if len(info["track%d" % i]) == 0:
info["track%d" % i] = "Track %02d" % (i+1)
else:
if debug:
print "failure getting track info, status: %i" % track_stat
for i in range(0, disc_id[1]):
info ["artist%d" % i] = info['artist']
info ["track%d" % i] = "Track %02d" % (i+1)
track_info = None
if debug:
print info
print ""
return info
if queryStatus is 202:
print ' Can\'t find CD data in CDDB.'
artist = raw_input('Artist name: ')
album = raw_input('Album name: ')
numberOfTracks = raw_input('Number of tracks: ')
for i in range(0, int(numberOfTracks)):
readInfo['TTITLE' + `i`] = raw_input('Track ' + `i + 1` + ' name: ')
elif queryStatus is 500:
print ' 500 error.'
exit(1);
elif queryStatus is 200 or queryStatus is 210:
if isinstance(queryInfo, list):
print 'Chose album name:'
count = 1;
for element in queryInfo:
(readStatus, elementInfo) = CDDB.read(element['category'], element['disc_id'], user = '******', host = 'localhost', client_name = 'CDDB-py', client_version = '5')
print "\t", count, ")", element['title'], elementInfo['DYEAR'];
count += 1;
chosen = raw_input('Your choose: ')
queryInfo = queryInfo[int(chosen) - 1]
(readStatus, readInfo) = CDDB.read(queryInfo['category'], queryInfo['disc_id'], user = '******', host = 'localhost', client_name = 'CDDB-py', client_version = '5')
album = queryInfo['title'].replace('(', '-').replace(')', '')
splitAt = album.index('/')
artist = album[:splitAt - 1].replace('(', '-').replace(')', '')
album = album[splitAt + 2:].replace('(', '-').replace(')', '')
print ' DONE.'
try:
cddb_file = ConfigObj(cddb_fullpath)
disk_title = info2['DTITLE'].split(' / ')
for i in range(num_tracks):
track_title[i] = track_prefix[i] + cddb_file['TTITLE%s' % (i)]
print("Used local cddb data.")
except (ConfigObjError):
print("Error in local cddb file")
cddb_local = False
if not cddb_local:
(status1, info1) = CDDB.query(disc_id)
print("query status = %s" % status1)
if status1 == 200: # Entry exist
print(info1['title'] + " // " + info1['category'])
(status2, info2) = CDDB.read(info1['category'], info1['disc_id'])
print("read status = %s" % status2)
readtracks = True
elif status1 == 210: # Various entries exist, choose blindly the first
print(info1[0]['title'] + " // " + info1[0]['category'])
(status2, info2) = CDDB.read(info1[0]['category'], info1[0]['disc_id'])
print("read status = %s" % status2)
if status2 == 210: # Start reading tracks
disk_title = info2['DTITLE'].split(' / ')
for i in range(num_tracks):
track_title[i] = track_prefix[i] + info2['TTITLE%s' % (i)]
# Makes playlist
playlist = open(playlist_fullpath, 'w')
playlist.write('<?xml version="1.0" encoding="UTF-8"?>\n')
playlist.write('<playlist version="1" xmlns="http://xspf.org/ns/0/">\n')
def query(category, discid, xcode='utf8:utf8'):
discinfo = {}
tracktitles = {}
dump = path.join(expanduser("~"), '.cddb', category, discid)
try:
for line in file(dump):
if line.startswith("TTITLE"):
track, title = line.split("=", 1)
try:
track = int(track[6:])
except (ValueError):
pass
else:
tracktitles[track] = \
title.decode('utf-8', 'replace').strip()
elif line.startswith("DGENRE"):
discinfo['genre'] = line.split('=', 1)[1].strip()
elif line.startswith("DTITLE"):
dtitle = line.split('=', 1)[1].strip().split(' / ', 1)
if len(dtitle) == 2:
discinfo['artist'], discinfo['title'] = dtitle
else:
discinfo['title'] = dtitle[0].strip()
elif line.startswith("DYEAR"):
discinfo['year'] = line.split('=', 1)[1].strip()
except EnvironmentError:
pass
else:
return discinfo, tracktitles
read, info = CDDB.read(category, discid, **CLIENTINFO)
if read != 210:
return None
try:
os.makedirs(path.join(expanduser("~"), '.cddb'))
except EnvironmentError:
pass
try:
save = file(dump, 'w')
keys = info.keys()
keys.sort()
for key in keys:
print>>save, "%s=%s" % (key, info[key])
save.close()
except EnvironmentError:
pass
xf, xt = xcode.split(':')
for key, value in info.iteritems():
try:
value = value.decode('utf-8', 'replace').strip().encode(
xf, 'replace').decode(xt, 'replace')
except AttributeError:
pass
if key.startswith('TTITLE'):
try:
tracktitles[int(key[6:])] = value
except ValueError:
pass
elif key == 'DGENRE':
discinfo['genre'] = value
elif key == 'DTITLE':
dtitle = value.strip().split(' / ', 1)
if len(dtitle) == 2:
discinfo['artist'], discinfo['title'] = dtitle
else:
discinfo['title'] = dtitle[0].strip()
elif key == 'DYEAR':
discinfo['year'] = value
return discinfo, tracktitles
