def __init__(self):
cdaudio = DiscID.open()
print "CD: ", cdaudio
try:
self.disc_id = DiscID.disc_id(cdaudio)
print "disc_id: ", self.disc_id
self.is_audio_cd = True
except:
print "No CD"
disc_id = None
self.is_audio_cd = False
self.song_list = []
cdaudio.close()
raise
if self.is_audio_cd:
try:
self.query_status, self.query_info = CDDB.query(self.disc_id)
print "query_status: ", self.query_status
print "query_info: ", self.query_info
except:
print 'IOError'
self.song_list = []
cdaudio.close()
self.query_status = 409
self.query_info = {"disc_id": "", "category": "", "title":""}
pass
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 cd_rip(self, device):
self.log.debug(device.device_node)
self.log.debug(device.device_path)
if self.is_cd_ready():
try:
# cddb
cd_title = None
cdrom = DiscID.open()
disc_id = DiscID.disc_id(cdrom)
(query_status, query_info) = CDDB.query(disc_id)
if query_status == 200:
cd_title = query_info['title'][:20]
self.log.info("Starting CD ripping for %s" % cd_title)
rip_workdir = get_config(self.args.conf, 'Ripping', 'workdir')
ripper = Worker(rip_workdir)
cd2cloud_cfg = get_config(self.args.conf, 'Ripping',
'abcde_conf')
cmd = '/usr/bin/abcde -c %s -j 5 -N' % cd2cloud_cfg
#ripper.thread(cmd)
ripper.run_cmd(cmd)
self.log.info("Complete CD ripping for %s" % cd_title)
self.notify(cd_title)
except:
msg = 'FAILED %s' % cd_title
self.notify(msg)
def __init__(self, disc_id):
# Assume that no results will be found.
self.artist = _("Unknown artist")
self.length = 0
self.title = _("Unknown title")
self.tracks = []
try:
(query_status, query_info) = CDDB.query(disc_id)
except IOError:
# Set query_status to 0 to act like an unknown CD.
query_status = 0
# See CDDB documentation for more information.
# http://cddb-py.sourceforge.net/CDDB/README
# query_info variable's type depends on query_status
if query_status == 200:
# On 200, a dictionary containing three items is returned
self._get_information_from_result_element(query_info, disc_id)
elif query_status == 210 or query_status == 211:
# On 211 or 210, a list will be returned as the second item.
# Each element of the list will be a dictionary containing
# three items, exactly the same as a single 200 success return.
self._get_information_from_result_element(query_info[0], disc_id)
else:
# No metadata found for this disc
for i in range(disc_id[1]):
self.tracks.append(CompactDiscTrack(i + 1,
_("Unknown track %(num)s.") % {'num': str(i + 1)}, 0))
def __init__(self):
cdaudio = DiscID.open()
print "CD: ", cdaudio
try:
self.disc_id = DiscID.disc_id(cdaudio)
print "disc_id: ", self.disc_id
self.is_audio_cd = True
except:
print "No CD"
disc_id = None
self.is_audio_cd = False
self.song_list = []
cdaudio.close()
raise
if self.is_audio_cd:
try:
self.query_status, self.query_info = CDDB.query(self.disc_id)
print "query_status: ", self.query_status
print "query_info: ", self.query_info
except:
print 'IOError'
self.song_list = []
cdaudio.close()
self.query_status = 409
self.query_info = {"disc_id": "", "category": "", "title": ""}
pass
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 read(self): try: device = DiscID.open() disc_info = DiscID.disc_id(device) (query_status, self.query_info) = CDDB.query(disc_info) except cdrom.error, e: raise cdrom.error, e
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 _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 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 _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 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(self, device):
cdromfd = cdrom.audiocd_open(device)
disc_id = cdrom.audiocd_id(cdromfd)
if kaa.metadata.USE_NETWORK:
try:
(query_stat, query_info) = CDDB.query(disc_id)
except Exception, e:
# Oops no connection
query_stat = 404
def query(self, device):
cdromfd = cdrom.audiocd_open(device)
disc_id = cdrom.audiocd_id(cdromfd)
if kaa.metadata.USE_NETWORK:
try:
(query_stat, query_info) = CDDB.query(disc_id)
except Exception, e:
# Oops no connection
query_stat = 404
def cddb_query(discid, cache_dir=os.path.expanduser('~/.cache/cddb')):
cache_path = os.path.join(cache_dir, '%08x' % discid[0])
query = cached(lambda:CDDB.query(discid)[1])(cache_path, discid)
categories = {}
if query is None:
raise Exception('Server has no data for this disc')
if isinstance(query, dict):
query = [query]
if isinstance(query, list):
for q in query:
categories[q['category']] = q['title']
return categories
def getCDDB(self, cddbdiscid):
"""
@param cddbdiscid: list of id, tracks, offsets, seconds
@rtype: str
"""
# FIXME: convert to nonblocking?
import CDDB
code, md = CDDB.query(cddbdiscid)
self.debug('CDDB query result: %r, %r', code, md)
if code == 200:
return md['title']
return None
def getCDDB(self, cddbdiscid):
"""
@param cddbdiscid: list of id, tracks, offsets, seconds
@rtype: str
"""
# FIXME: convert to nonblocking?
import CDDB
code, md = CDDB.query(cddbdiscid)
self.debug('CDDB query result: %r, %r', code, md)
if code == 200:
return md['title']
return None
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 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 CDDB_info(CDINFO, DISCID):
# Get some information
cdrom = DiscID.open(CDINFO, )
disc_id = DiscID.disc_id(cdrom)
# Query CDDB
print "Querying CDDB ... "
(query_status, query_info) = CDDB.query(disc_id)
info = False
# Check for status messages
if (query_status == 200):
# 200: success
print " success."
info = True
elif (query_status == 210):
# 210: Multiple exact matches
if (DISCID is not None):
query_info = query_info[DISCID]
info = True
else:
multiple_matches("Multiple exact matches found:", query_info)
elif (query_status == 211):
# 211: Multiple inexact matches
if (DISCID > -1):
query_info = query_info[DISCID]
info = True
else:
multiple_matches("Multiple inexact matches found:", query_info)
elif (query_status == 202):
# 202: No match found
print " failed. No match found."
else:
# Something else went horribly wrong
print " failed with status " + str(query_status) + ":\n"
print "Using: "
print query_info
return (info, query_info, disc_id)
def CDDB_info(CDINFO, DISCID):
# Get some information
cdrom = DiscID.open(CDINFO,)
disc_id = DiscID.disc_id(cdrom)
# Query CDDB
print "Querying CDDB ... "
(query_status, query_info) = CDDB.query(disc_id)
info = False
# Check for status messages
if (query_status == 200):
# 200: success
print " success."
info = True
elif (query_status == 210):
# 210: Multiple exact matches
if (DISCID is not None):
query_info = query_info[DISCID]
info = True
else:
multiple_matches("Multiple exact matches found:", query_info)
elif (query_status == 211):
# 211: Multiple inexact matches
if (DISCID > -1):
query_info = query_info[DISCID]
info = True
else:
multiple_matches("Multiple inexact matches found:", query_info)
elif (query_status == 202):
# 202: No match found
print " failed. No match found."
else:
# Something else went horribly wrong
print " failed with status "+str(query_status)+":\n"
print "Using: "
print query_info
return (info, query_info, disc_id)
def getCDDB(self, cddbdiscid):
"""
@param cddbdiscid: list of id, tracks, offsets, seconds
@rtype: str
"""
# FIXME: convert to nonblocking?
import CDDB
try:
code, md = CDDB.query(cddbdiscid)
self.debug('CDDB query result: %r, %r', code, md)
if code == 200:
return md['title']
except IOError, e:
# FIXME: for some reason errno is a str ?
if e.errno == 'socket error':
self._stdout.write("Warning: network error: %r\n" % (e, ))
else:
raise
def getCDDB(self, cddbdiscid):
"""
@param cddbdiscid: list of id, tracks, offsets, seconds
@rtype: str
"""
# FIXME: convert to nonblocking?
import CDDB
try:
code, md = CDDB.query(cddbdiscid)
self.debug('CDDB query result: %r, %r', code, md)
if code == 200:
return md['title']
except IOError, e:
# FIXME: for some reason errno is a str ?
if e.errno == 'socket error':
self._stdout.write("Warning: network error: %r\n" % (e, ))
else:
raise
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(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 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 __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 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 _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 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
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' + `i`])
else:
print "failure getting track info, status: %i" % read_stat
def search_by_id(discid):
try:
stat, discs = CDDB.query(discid, **CLIENTINFO)
except EnvironmentError, e:
raise RetrievalError(e.strerror)
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 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
baseDir = '/home/lock/music/'
encoderPath = '/home/lock/usr/bin/mppenc'
def camelCase(value):
return "".join([capitalize(w) for w in re.split(re.compile("[ _]?"), value)])
print 'Getting CDDB info ...',
stdout.flush()
cdrom = DiscID.open()
discId = DiscID.disc_id(cdrom)
artist = ''
album = ''
readInfo = {}
(queryStatus, queryInfo) = CDDB.query(discId, user = '******', host = 'localhost', client_name = 'CDDB-py', client_version = '1.4')
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;
def plugin_album(self, album):
discid = calculate_discid(album)
try:
stat, discs = CDDB.query(discid, **CLIENTINFO)
except IOError:
ErrorMessage(None, _("Timeout"), _(
"Query could not be executed, connection timed out")).run()
return
if stat in (200, 211):
xcode = 'utf8:utf8'
dlg = Gtk.Dialog(title=_('Select an album'))
dlg.set_border_width(6)
dlg.set_resizable(False)
dlg.add_buttons(Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL)
dlg.add_buttons(Gtk.STOCK_SAVE, Gtk.ResponseType.OK)
dlg.vbox.set_spacing(6)
dlg.set_default_response(Gtk.ResponseType.CANCEL)
model = Gtk.ListStore(str, str, str, str, str, str)
for disc in discs:
model.append(
[disc[s] for s in ('title', 'category', 'disc_id')] * 2)
box = Gtk.ComboBox(model=model)
box.set_active(0)
for i in range(3):
crt = Gtk.CellRendererText()
box.pack_start(crt, True)
box.add_attribute(crt, "text", i)
discinfo = Gtk.Label()
crosscode = Gtk.ListStore(str)
crosscode.append(['utf8:utf8'])
crosscode.append(['latin1:latin2'])
crosscode.append(['latin1:cp1251'])
crosscode.append(['latin1:sjis'])
crosscode.append(['latin1:euc-jp'])
cbo = Gtk.ComboBox(
model=crosscode, entry_text_column=0, has_entry=True)
cbo.set_active(0)
def update_discinfo(combo):
xcode = cbo.get_child().get_text()
model = combo.get_model()
t, c, d, title, cat, discid = model[box.get_active()]
info = query(cat, discid, xcode=xcode)
discinfo.set_markup(
make_info_label(info, album, discs[0]['disc_id']))
def crosscode_cddbinfo(combo):
try:
xf, xt = combo.get_child().get_text().split(':')
for row in model:
for show, store in zip(range(0, 3), range(3, 6)):
row[show] = row[store].encode(
xf, 'replace').decode(xt, 'replace')
except:
for row in model:
for show, store in zip(range(0, 3), range(3, 6)):
row[show] = row[store]
update_discinfo(box)
cbo.connect('changed', crosscode_cddbinfo)
box.connect('changed', update_discinfo)
update_discinfo(box)
dlg.vbox.pack_start(Gtk.Label(
_("Select the album you wish to retrieve.")), True, True, 0)
dlg.vbox.pack_start(box, True, True, 0)
dlg.vbox.pack_start(discinfo, True, True, 0)
dlg.vbox.pack_start(cbo, True, True, 0)
dlg.vbox.show_all()
resp = dlg.run()
xcode = cbo.get_child().get_text()
if resp == Gtk.ResponseType.OK:
t, c, d, title, cat, discid = model[box.get_active()]
(disc, track) = query(cat, discid, xcode=xcode)
keys = track.keys()
keys.sort()
for key, song in zip(keys, album):
if 'artist' in disc:
song['artist'] = disc['artist']
if 'title' in disc:
song['album'] = disc['title']
if 'year' in disc:
song['date'] = disc['year']
if 'genre' in disc:
song['genre'] = disc['genre']
s = track[key].split("/")
if len(s) == 2:
song['artist'] = s[0]
song['title'] = s[1]
else:
song['title'] = track[key]
song['tracknumber'] = '%d/%d' % (key + 1, len(album))
dlg.destroy()
else:
n = len(album)
albumname = album[0]('album')
if not albumname:
albumname = ngettext('%d track', '%d tracks', n) % n
ErrorMessage(None, _("CDDB lookup failed (%s)" % stat),
ngettext(u"%(title)s and %(count)d more…",
u"%(title)s and %(count)d more…", n - 1) % {
'title': album[0]('~basename'), 'count':
n - 1}).run()
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)
def search_by_id(discid):
try:
stat, discs = CDDB.query(discid, **CLIENTINFO)
except EnvironmentError, e:
raise RetrievalError(e.strerror)
for line in fileinput.input(cddb_fullpath, inplace=1):
if line != '.\n':
print(line, end="")
# reads local cddb file
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)]
def plugin_album(self, album):
discid = calculate_discid(album)
try:
stat, discs = CDDB.query(discid, **CLIENTINFO)
except IOError:
ErrorMessage(None, _("Timeout"),_(
"Query could not be executed, connection timed out")).run()
return
info = None
if stat in (200,211):
xcode = 'utf8:utf8'
dlg = gtk.Dialog(_('Select an album'))
dlg.set_border_width(6)
dlg.set_has_separator(False)
dlg.set_resizable(False)
dlg.add_buttons(gtk.STOCK_OK, gtk.RESPONSE_OK)
dlg.vbox.set_spacing(6)
dlg.set_default_response(gtk.RESPONSE_OK)
model = gtk.ListStore(str, str, str, str, str, str)
for disc in discs:
model.append(
[disc[s] for s in ('title','category','disc_id')] * 2)
box = gtk.ComboBox(model)
box.set_active(0)
for i in range(3):
crt = gtk.CellRendererText()
box.pack_start(crt)
box.set_attributes(crt, text=i)
discinfo = gtk.Label()
crosscode = gtk.ListStore(str)
crosscode.append(['utf8:utf8'])
crosscode.append(['latin1:latin2'])
crosscode.append(['latin1:cp1251'])
crosscode.append(['latin1:sjis'])
crosscode.append(['latin1:euc-jp'])
cbo = gtk.ComboBoxEntry(crosscode, column=0)
cbo.set_active(0)
def update_discinfo(combo):
xcode = cbo.child.get_text()
t,c,d, title, cat, discid = combo.get_model()[box.get_active()]
info = query(cat, discid, xcode=xcode)
discinfo.set_markup(
make_info_label(info, album, discs[0]['disc_id']))
def crosscode_cddbinfo(combo):
try:
xf, xt = combo.child.get_text().split(':')
for row in model:
for show, store in zip(range(0,3), range(3,6)):
row[show] = row[store].encode(
xf, 'replace').decode(xt, 'replace')
except:
for row in model:
for show, store in zip(range(0,3), range(3,6)):
row[show] = row[store]
update_discinfo(box)
cbo.connect('changed', crosscode_cddbinfo)
box.connect('changed', update_discinfo)
update_discinfo(box)
dlg.vbox.pack_start(gtk.Label(
_("Select the album you wish to retrieve.")))
dlg.vbox.pack_start(box)
dlg.vbox.pack_start(discinfo)
dlg.vbox.pack_start(cbo)
dlg.vbox.show_all()
resp = dlg.run()
xcode = cbo.child.get_text()
if resp == gtk.RESPONSE_OK:
t,c,d, title, cat, discid = model[box.get_active()]
(disc, track) = query(cat, discid, xcode=xcode)
keys = track.keys()
keys.sort()
for key, song in zip(keys, album):
if 'artist' in disc: song['artist'] = disc['artist']
if 'title' in disc: song['album'] = disc['title']
if 'year' in disc: song['date'] = disc['year']
if 'genre' in disc: song['genre'] = disc['genre']
s = track[key].split("/")
if len(s) == 2:
song['artist'] = s[0]
song['title'] = s[1]
else:
song['title'] = track[key]
song['tracknumber'] = '%d/%d' % (key+1, len(album))
dlg.destroy()
else:
n = len(album)
albumname = album[0]('album')
if not albumname:
albumname = ngettext('%d track', '%d tracks', n) % n
ErrorMessage(None, _("CDDB lookup failed (%s)" % stat),
ngettext("%(title)s and %(count)d more...",
"%(title)s and %(count)d more...", n-1) % {
'title': album[0]('~basename'), 'count':
n-1}).run()
from argparse import ArgumentParser
import CDDB, DiscID
import audiotools
if __name__ == '__main__':
ap = ArgumentParser()
ns = ap.parse_args()
while True:
cdrom = DiscID.open()
disc_id = DiscID.disc_id(cdrom)
query_status, results = CDDB.query(disc_id)
stored_results = {}
print query_status, results
# Hack because I suck at coding
if isinstance(results, dict):
results = [results]
for i, query_info in enumerate(results):
print " == RESULT %d ==" % i
read_status, read_info = CDDB.read(query_info['category'], query_info['disc_id'])
print read_status, read_info
stored_results[i] = read_info
for tn in range(disc_id[1]):
print "Track %.02d: %s" % (tn+1, read_info['TTITLE%d' % tn])
