Add feature to vote on tags (alastair#240)

Signed-off-by: Thomas Brockmöller <tho.b.j@gmx.de>

Author: gorgobacka

musicbrainzngs/mbxml.py

@@ -775,10 +775,17 @@ def make_tag_request(**kwargs):
                 e_xml = ET.SubElement(e_list, "{%s}%s" % (NS, entity_type.replace('_', '-')))
                 e_xml.set("{%s}id" % NS, e)
                 taglist = ET.SubElement(e_xml, "{%s}user-tag-list" % NS)
-                for tag in tags:
-                    usertag_xml = ET.SubElement(taglist, "{%s}user-tag" % NS)
-                    name_xml = ET.SubElement(usertag_xml, "{%s}name" % NS)
-                    name_xml.text = tag
+                if isinstance(tags, dict):
+                    for tag, vote in tags.items():
+                        usertag_xml = ET.SubElement(taglist, "{%s}user-tag" % NS)
+                        usertag_xml.set('vote', vote)
+                        name_xml = ET.SubElement(usertag_xml, "{%s}name" % NS)
+                        name_xml.text = tag
+                else:
+                    for tag in tags:
+                        usertag_xml = ET.SubElement(taglist, "{%s}user-tag" % NS)
+                        name_xml = ET.SubElement(usertag_xml, "{%s}name" % NS)
+                        name_xml.text = tag
     if kwargs.keys():
         raise TypeError("make_tag_request() got an unexpected keyword argument '%s'" % kwargs.popitem()[0])
 

musicbrainzngs/musicbrainz.py

@@ -1259,8 +1259,10 @@ def submit_tags(**kwargs):
     """Submit user tags.
     Takes parameters named e.g. 'artist_tags', 'recording_tags', etc.,
     and of the form:
+    {entity_id1: {tag1: vote, ...}, ...}
+    If you don't want to vote, you can use a list of tags instead:
     {entity_id1: [tag1, ...], ...}
-    If you only have one tag for an entity you can use a string instead
+    If you only have one tag for an entity, you can use a string instead
     of a list.
 
     The user's tags for each entity will be set to that list, adding or
@@ -1269,7 +1271,7 @@ def submit_tags(**kwargs):
     """
     for k, v in kwargs.items():
         for id, tags in v.items():
-            kwargs[k][id] = tags if isinstance(tags, list) else [tags]
+            kwargs[k][id] = tags if isinstance(tags, (list, dict)) else [tags]
 
     query = mbxml.make_tag_request(**kwargs)
     return _do_mb_post("tag", query)