def guess_album(trackinfo): for rid,trackmap in guess_album2(trackinfo): release = lookups.get_release_by_releaseid(rid) albumartist=release.artist if musicbrainz2.model.Release.TYPE_SOUNDTRACK in release.types: directoryname = "Soundtrack" else: directoryname = albumartist.name releaseevents=release.getReleaseEvents() tracks=release.getTracks() trackdata=[] for tracknum,fileid in trackmap.items(): trk=tracks[tracknum-1] musicfile = trackinfo[fileid] if trk.artist is None: artist=albumartist.name sortartist=albumartist.sortName artistid=albumartist.id else: artist=trk.artist.name sortartist=trk.artist.sortName artistid=trk.artist.id trackdata.append(( tracknum, artist, sortartist, trk.title, musicfile.getDuration(), musicfile.getFilename(), artistid, trk.id)) asin = lookups.get_asin_from_release(release) albuminfo = ( directoryname, release.title, rid+".html", [x.date for x in releaseevents], asin, trackdata, albumartist, release.id, ) yield albuminfo
def guess_album(trackinfo):
releasedata={}
for rid in guess_album2(trackinfo):
release = lookups.get_release_by_releaseid(rid)
albumartist=release.artist
if musicbrainz2.model.Release.TYPE_SOUNDTRACK in release.types:
directoryname = "Soundtrack"
else:
directoryname = albumartist.name
#print albumartist.name,":",release.title+" ("+rid+".html)"
releaseevents=release.getReleaseEvents()
#print "Release dates:"
#for ev in releaseevents:
# print " ",ev.date
#print "Track:"
tracks=release.getTracks()
trackdata=[]
for tracknum in range(len(tracks)):
trk=tracks[tracknum]
(fname,artist,trackname,dur,trackprints,puid) = trackinfo[tracknum+1]
if trk.artist is None:
artist=albumartist.name
sortartist=albumartist.sortName
artistid=albumartist.id
else:
artist=trk.artist.name
sortartist=trk.artist.sortName
artistid=trk.artist.id
#print " ",tracknum+1,"-",artist,"-",trk.title,"%2d:%06.3f" % (int(dur/60000),(dur%6000)/1000),`fname`
trackdata.append((tracknum+1,artist,sortartist,trk.title,dur,fname,artistid,trk.id))
asin = lookups.get_asin_from_release(release)
albuminfo = (
directoryname,
release.title,
rid+".html",
[x.date for x in releaseevents],
asin,
trackdata,
albumartist,
release.id,
)
yield albuminfo
def guess_album(trackinfo):
for rid, trackmap in guess_album2(trackinfo):
release = lookups.get_release_by_releaseid(rid)
albumartist = release.artist
if musicbrainz2.model.Release.TYPE_SOUNDTRACK in release.types:
directoryname = "Soundtrack"
else:
directoryname = albumartist.name
releaseevents = release.getReleaseEvents()
tracks = release.getTracks()
trackdata = []
for tracknum, fileid in trackmap.items():
trk = tracks[tracknum - 1]
musicfile = trackinfo[fileid]
if trk.artist is None:
artist = albumartist.name
sortartist = albumartist.sortName
artistid = albumartist.id
else:
artist = trk.artist.name
sortartist = trk.artist.sortName
artistid = trk.artist.id
trackdata.append((tracknum, artist, sortartist, trk.title,
musicfile.getDuration(), musicfile.getFilename(),
artistid, trk.id))
asin = lookups.get_asin_from_release(release)
albuminfo = (
directoryname,
release.title,
rid + ".html",
[x.date for x in releaseevents],
asin,
trackdata,
albumartist,
release.id,
)
yield albuminfo
def main():
if len(sys.argv) < 2:
print_usage()
sys.exit(1)
releaseid = None
embedcovers = True
asin = None
year = None
noact = False
totaldiscs = None
for option in sys.argv[2:]:
if option.startswith("--release-id="):
releaseid = option.split("=")[1].strip()
elif option.startswith("--no-embed-coverart"):
embedcovers = False
elif option.startswith("--release-asin="):
asin = option.split("=", 1)[1].strip()
elif option.startswith("--year="):
year = option.split("=")[1].strip()
elif option.startswith("-n"):
noact = True
elif option.startswith("--total-discs"):
totaldiscs = option.split("=", 1)[1].strip()
srcpath = os.path.abspath(sys.argv[1])
if not os.path.exists(srcpath):
print_usage()
sys.exit(2)
if noact:
print "Performing dry-run"
print "Source path: " + srcpath
tocfilename = ""
if os.path.exists(os.path.join(srcpath, "data.toc")):
tocfilename = "data.toc"
elif os.path.exists(os.path.join(srcpath, "TOC")):
tocfilename = "TOC"
else:
print "No TOC in source path!"
sys.exit(4)
disc = toc.Disc(cdrdaotocfile=os.path.join(srcpath, tocfilename))
disc.tocfilename = tocfilename
disc.discid = discid.generate_musicbrainz_discid(
disc.get_first_track_num(), disc.get_last_track_num(), disc.get_track_offsets()
)
for i in range(len(disc.tracks)):
disc.tracks[i].filename = os.path.join(srcpath, "track" + str(i + 1).zfill(2) + ".flac")
if not os.path.exists(disc.tracks[i].filename):
disc.tracks[i].filename = os.path.join(srcpath, "track" + str(i + 1).zfill(2) + ".cdda.flac")
print "discID: " + disc.discid
if releaseid:
disc.releaseid = releaseid
release = get_musicbrainz_release(disc)
if release is None:
raise Exception("Couldn't find a matching release. Sorry, I tried.")
print "release id: %s.html" % (release.id)
disc.releasetypes = release.getTypes()
disc.set_musicbrainz_tracks(release.getTracks())
disc.releasedate = release.getEarliestReleaseDate()
disc.artist = release.artist.name
disc.album = release.title
if year is not None:
disc.year = year
disc.releasedate = year
elif disc.releasedate is not None:
disc.year = disc.releasedate[0:4]
else:
raise Exception("Unknown year: %s %s " % (` disc.artist `, ` disc.album `))
disc.compilation = 0
disc.number = 0
disc.totalnumber = 0
if asin is not None:
disc.asin = asin
else:
disc.asin = lookups.get_asin_from_release(release, prefer=".co.uk")
# Set the compilation tag appropriately
if musicbrainz2.model.Release.TYPE_COMPILATION in disc.releasetypes:
disc.compilation = 1
# Name the target folder differently for soundtracks
if musicbrainz2.model.Release.TYPE_SOUNDTRACK in disc.releasetypes:
newpath = "Soundtrack - %s - %s" % (disc.year, disc.album)
else:
newpath = "%s - %s - %s" % (mp3names.FixArtist(disc.artist), disc.year, disc.album)
newpath = mp3names.FixFilename(newpath)
newpath = os.path.join(srcpath, "../%s/" % newpath)
newpath = os.path.normpath(newpath)
print "Destination path: " + newpath
if os.path.exists(newpath):
print "Destination path already exists, skipping"
sys.exit(3)
if not noact:
os.mkdir(newpath)
# Get album art
imageurl = lookups.get_album_art_url_for_asin(disc.asin)
# Check for manual image
if os.path.exists(os.path.join(srcpath, "folder.jpg")):
print "Using existing image"
if not noact:
shutil.copyfile(os.path.join(srcpath, "folder.jpg"), os.path.join(newpath, "folder.jpg"))
elif imageurl is not None:
if not noact:
try:
(f, h) = urllib.urlretrieve(imageurl, os.path.join(newpath, "folder.jpg"))
if h.getmaintype() != "image":
print "WARNING: Failed to retrieve coverart (%s)" % imageurl
embedcovers = False
except:
print "WARNING: Failed to retrieve coverart (%s)" % imageurl
embedcovers = False
else:
embedcovers = False
# Deal with disc x of y numbering
(albumname, discnumber, disctitle) = lookups.parse_album_name(disc.album)
if discnumber is None:
disc.number = 1
disc.totalnumber = 1
elif totaldiscs is not None:
disc.totalnumber = totaldiscs
disc.number = int(discnumber)
else:
disc.number = int(discnumber)
discs = lookups.get_all_releases_in_set(release.id)
disc.totalnumber = len(discs)
print "disc " + str(disc.number) + " of " + str(disc.totalnumber)
flacname(disc, release, srcpath, newpath, embedcovers, noact)