# -*- coding: utf-8 -*-
PLUGIN_NAME = u"Add Cluster As Release"
PLUGIN_AUTHOR = u"Lukáš Lalinský, Philip Jägenstedt"
PLUGIN_DESCRIPTION = "Adds a plugin context menu option to clusters to help you quickly add a release into the MusicBrainz\
database via the website by pre-populating artists, track names and times."
PLUGIN_VERSION = "0.4"
PLUGIN_API_VERSIONS = ["1.0.0"]
from picard.cluster import Cluster
from picard.util import webbrowser2
from picard.ui.itemviews import BaseAction, register_cluster_action
import codecs
import os
import tempfile
HTML_HEAD = """
Add Cluster As Release
"""
HTML_ATTR_ESCAPE = {
"&": "&",
'"': """
}
class AddClusterAsRelease(BaseAction):
NAME = "Add Cluster As Release..."
def callback(self, objs):
if len(objs) != 1 or not isinstance(objs[0], Cluster):
return
cluster = objs[0]
(fd, fp) = tempfile.mkstemp(suffix=".html")
f = codecs.getwriter("utf-8")(os.fdopen(fd, "w"))
def esc(s):
return "".join(HTML_ATTR_ESCAPE.get(c, c) for c in s)
# add a global (release-level) name-value
def nv(n, v):
f.write(HTML_INPUT % (esc(n), esc(v)))
f.write(HTML_HEAD)
nv("artist_credit.names.0.artist.name", cluster.metadata["albumartist"])
nv("name", cluster.metadata["album"])
discnumber_shift = -1
for i, file in enumerate(cluster.files):
try:
i = int(file.metadata["tracknumber"]) - 1
except:
pass
# As per https://musicbrainz.org/doc/Development/Release_Editor_Seeding#Tracklists_and_Mediums
# the medium numbers ("m") must be starting with 0.
# Maybe the existing tags don't have disc numbers in them or
# they're starting with something smaller than or equal to 0, so try
# to produce a sane disc number.
try:
m = int(file.metadata.get("discnumber", 1))
if m <= 0:
# A disc number was smaller than or equal to 0 - all other
# disc numbers need to be changed to accommodate that.
discnumber_shift = max(discnumber_shift, 0 - m)
m = m + discnumber_shift
except Exception as e:
# The most likely reason for an exception at this point is a
# ValueError because the disc number in the tags was not a
# number. Just log the exception and assume the medium number
# is 0.
file.log.info("Trying to get the disc number of %s caused the following error: %s; assuming 0",
file.filename, e)
m = 0
# add a track-level name-value
def tnv(n, v):
nv("mediums.%d.track.%d.%s" % (m, i, n), v)
tnv("name", file.metadata["title"])
if file.metadata["artist"] != cluster.metadata["albumartist"]:
tnv("artist_credit.names.0.name", file.metadata["artist"])
tnv("length", str(file.metadata.length))
f.write(HTML_TAIL)
f.close()
webbrowser2.open("file://"+fp)
register_cluster_action(AddClusterAsRelease())