def delete_frames(deletes, filenames):
try:
deletes = frame_from_fsnative(deletes)
except ValueError as err:
print_(text_type(err), file=sys.stderr)
frames = deletes.split(",")
for filename in filenames:
with _sig.block():
if verbose:
print_(u"deleting %s from" % deletes, filename,
file=sys.stderr)
try:
id3 = mutagen.id3.ID3(filename)
except mutagen.id3.ID3NoHeaderError:
if verbose:
print_(u"No ID3 header found; skipping.", file=sys.stderr)
except Exception as err:
print_(text_type(err), file=sys.stderr)
raise SystemExit(1)
else:
for frame in frames:
id3.delall(frame)
id3.save()
def delete_frames(deletes, filenames):
try:
deletes = frame_from_fsnative(deletes)
except ValueError as err:
print_(text_type(err), file=sys.stderr)
frames = deletes.split(",")
for filename in filenames:
with _sig.block():
if verbose:
print_(u"deleting %s from" % deletes,
filename,
file=sys.stderr)
try:
id3 = mutagen.id3.ID3(filename)
except mutagen.id3.ID3NoHeaderError:
if verbose:
print_(u"No ID3 header found; skipping.", file=sys.stderr)
except Exception as err:
print_(text_type(err), file=sys.stderr)
raise SystemExit(1)
else:
for frame in frames:
id3.delall(frame)
id3.save()
def call2(self, *args):
for arg in args:
self.assertTrue(isinstance(arg, fsnative))
old_stdout = sys.stdout
old_stderr = sys.stderr
try:
out = StringIO()
err = StringIO()
sys.stdout = out
sys.stderr = err
try:
ret = self._main([fsnative(self.TOOL_NAME)] + list(args))
except SystemExit as e:
ret = e.code
ret = ret or 0
out_val = out.getvalue()
err_val = err.getvalue()
if os.name == "nt" and PY2:
encoding = getattr(sys.stdout, "encoding", None) or "mbcs"
out_val = text_type(out_val, encoding)
err_val = text_type(err_val, encoding)
return (ret, out_val, err_val)
finally:
sys.stdout = old_stdout
sys.stderr = old_stderr
def call2(self, *args):
for arg in args:
self.assertTrue(isinstance(arg, fsnative))
old_stdout = sys.stdout
old_stderr = sys.stderr
try:
out = StringIO()
err = StringIO()
sys.stdout = out
sys.stderr = err
try:
ret = self._main([fsnative(self.TOOL_NAME)] + list(args))
except SystemExit as e:
ret = e.code
ret = ret or 0
out_val = out.getvalue()
err_val = err.getvalue()
if os.name == "nt" and PY2:
encoding = getattr(sys.stdout, "encoding", None) or "mbcs"
out_val = text_type(out_val, encoding)
err_val = text_type(err_val, encoding)
return (ret, out_val, err_val)
finally:
sys.stdout = old_stdout
sys.stderr = old_stderr
def update_to_v24(self):
"""Convert older tags into an ID3v2.4 tag.
This updates old ID3v2 frames to ID3v2.4 ones (e.g. TYER to
TDRC). If you intend to save tags, you must call this function
at some point; it is called by default when loading the tag.
"""
self.__update_common()
# TDAT, TYER, and TIME have been turned into TDRC.
try:
date = text_type(self.get("TYER", ""))
if date.strip(u"\x00"):
self.pop("TYER")
dat = text_type(self.get("TDAT", ""))
if dat.strip("\x00"):
self.pop("TDAT")
date = "%s-%s-%s" % (date, dat[2:], dat[:2])
time = text_type(self.get("TIME", ""))
if time.strip("\x00"):
self.pop("TIME")
date += "T%s:%s:00" % (time[:2], time[2:])
if "TDRC" not in self:
self.add(TDRC(encoding=0, text=date))
except UnicodeDecodeError:
# Old ID3 tags have *lots* of Unicode problems, so if TYER
# is bad, just chuck the frames.
pass
# TORY can be the first part of a TDOR.
if "TORY" in self:
f = self.pop("TORY")
if "TDOR" not in self:
try:
self.add(TDOR(encoding=0, text=str(f)))
except UnicodeDecodeError:
pass
# IPLS is now TIPL.
if "IPLS" in self:
f = self.pop("IPLS")
if "TIPL" not in self:
self.add(TIPL(encoding=f.encoding, people=f.people))
# These can't be trivially translated to any ID3v2.4 tags, or
# should have been removed already.
for key in ["RVAD", "EQUA", "TRDA", "TSIZ", "TDAT", "TIME"]:
if key in self:
del(self[key])
# Recurse into chapters
for f in self.getall("CHAP"):
f.sub_frames.update_to_v24()
for f in self.getall("CTOC"):
f.sub_frames.update_to_v24()
def update_to_v24(self):
"""Convert older tags into an ID3v2.4 tag.
This updates old ID3v2 frames to ID3v2.4 ones (e.g. TYER to
TDRC). If you intend to save tags, you must call this function
at some point; it is called by default when loading the tag.
"""
self.__update_common()
# TDAT, TYER, and TIME have been turned into TDRC.
try:
date = text_type(self.get("TYER", ""))
if date.strip(u"\x00"):
self.pop("TYER")
dat = text_type(self.get("TDAT", ""))
if dat.strip("\x00"):
self.pop("TDAT")
date = "%s-%s-%s" % (date, dat[2:], dat[:2])
time = text_type(self.get("TIME", ""))
if time.strip("\x00"):
self.pop("TIME")
date += "T%s:%s:00" % (time[:2], time[2:])
if "TDRC" not in self:
self.add(TDRC(encoding=0, text=date))
except UnicodeDecodeError:
# Old ID3 tags have *lots* of Unicode problems, so if TYER
# is bad, just chuck the frames.
pass
# TORY can be the first part of a TDOR.
if "TORY" in self:
f = self.pop("TORY")
if "TDOR" not in self:
try:
self.add(TDOR(encoding=0, text=str(f)))
except UnicodeDecodeError:
pass
# IPLS is now TIPL.
if "IPLS" in self:
f = self.pop("IPLS")
if "TIPL" not in self:
self.add(TIPL(encoding=f.encoding, people=f.people))
# These can't be trivially translated to any ID3v2.4 tags, or
# should have been removed already.
for key in ["RVAD", "EQUA", "TRDA", "TSIZ", "TDAT", "TIME"]:
if key in self:
del (self[key])
# Recurse into chapters
for f in self.getall("CHAP"):
f.sub_frames.update_to_v24()
for f in self.getall("CTOC"):
f.sub_frames.update_to_v24()
def test_ASFGUIDAttribute_dunder(self):
attr = ASFGUIDAttribute(data=b"\xff")
self.assertEqual(bytes(attr), b"\xff")
if PY3:
self.assertEqual(text_type(attr), u"b'\\xff'")
self.assertEqual(repr(attr), "ASFGUIDAttribute(b'\\xff')")
else:
self.assertEqual(text_type(attr), u"'\\xff'")
self.assertEqual(repr(attr), "ASFGUIDAttribute('\\xff')")
self.assertRaises(TypeError, int, attr)
def list_tags_raw(filenames):
for filename in filenames:
print_("Raw IDv2 tag info for", filename)
try:
id3 = mutagen.id3.ID3(filename, translate=False)
except mutagen.id3.ID3NoHeaderError:
print_(u"No ID3 header found; skipping.")
except Exception as err:
print_(text_type(err), file=sys.stderr)
raise SystemExit(1)
else:
for frame in id3.values():
print_(text_type(repr(frame)))
def list_tags_raw(filenames):
for filename in filenames:
print_("Raw IDv2 tag info for", filename)
try:
id3 = mutagen.id3.ID3(filename, translate=False)
except mutagen.id3.ID3NoHeaderError:
print_(u"No ID3 header found; skipping.")
except Exception as err:
print_(text_type(err), file=sys.stderr)
raise SystemExit(1)
else:
for frame in id3.values():
print_(text_type(repr(frame)))
def __setitem__(self, key, values):
"""Set a key's value or values.
Setting a value overwrites all old ones. The value may be a
list of Unicode or UTF-8 strings, or a single Unicode or UTF-8
string.
"""
if not isinstance(values, list):
values = [values]
try:
del(self[key])
except KeyError:
pass
for value in values:
if key in _standard_attribute_names:
value = text_type(value)
elif not isinstance(value, ASFBaseAttribute):
if isinstance(value, string_types):
value = ASFUnicodeAttribute(value)
elif isinstance(value, bool):
value = ASFBoolAttribute(value)
elif isinstance(value, int):
value = ASFDWordAttribute(value)
elif isinstance(value, long):
value = ASFQWordAttribute(value)
self.append((key, value))
def test_types(self):
if PY3:
self.assertRaises(TypeError, ID3TimeStamp, b"blah")
self.assertEquals(
text_type(ID3TimeStamp(u"2000-01-01")), u"2000-01-01")
self.assertEquals(
bytes(ID3TimeStamp(u"2000-01-01")), b"2000-01-01")
def __setitem__(self, key, values):
"""Set a key's value or values.
Setting a value overwrites all old ones. The value may be a
list of Unicode or UTF-8 strings, or a single Unicode or UTF-8
string.
"""
if not isinstance(values, list):
values = [values]
try:
del (self[key])
except KeyError:
pass
for value in values:
if key in _standard_attribute_names:
value = text_type(value)
elif not isinstance(value, ASFBaseAttribute):
if isinstance(value, string_types):
if PY2 or isinstance(value, text_type):
value = ASFUnicodeAttribute(value)
elif isinstance(value, bool):
value = ASFBoolAttribute(value)
elif isinstance(value, int):
value = ASFDWordAttribute(value)
elif isinstance(value, long):
value = ASFQWordAttribute(value)
self.append((key, value))
def test_ASFByteArrayAttribute_dunder(self):
attr = ASFByteArrayAttribute(data=b"\xff")
self.assertEqual(bytes(attr), b"\xff")
self.assertEqual(text_type(attr), u"[binary data (1 bytes)]")
if PY3:
self.assertEqual(repr(attr), r"ASFByteArrayAttribute(b'\xff')")
else:
self.assertEqual(repr(attr), r"ASFByteArrayAttribute('\xff')")
self.assertRaises(TypeError, int, attr)
def copy(src, dst, merge, write_v1=True, excluded_tags=None, verbose=False):
"""Returns 0 on success"""
if excluded_tags is None:
excluded_tags = []
try:
id3 = mutagen.id3.ID3(src, translate=False)
except mutagen.id3.ID3NoHeaderError:
print_(u"No ID3 header found in ", src, file=sys.stderr)
return 1
except Exception as err:
print_(str(err), file=sys.stderr)
return 1
if verbose:
print_(u"File", src, u"contains:", file=sys.stderr)
print_(id3.pprint(), file=sys.stderr)
for tag in excluded_tags:
id3.delall(tag)
if merge:
try:
target = mutagen.id3.ID3(dst, translate=False)
except mutagen.id3.ID3NoHeaderError:
# no need to merge
pass
except Exception as err:
print_(str(err), file=sys.stderr)
return 1
else:
for frame in id3.values():
target.add(frame)
id3 = target
# if the source is 2.3 save it as 2.3
if id3.version < (2, 4, 0):
id3.update_to_v23()
v2_version = 3
else:
id3.update_to_v24()
v2_version = 4
try:
id3.save(dst, v1=(2 if write_v1 else 0), v2_version=v2_version)
except Exception as err:
print_(u"Error saving",
dst,
u":\n%s" % text_type(err),
file=sys.stderr)
return 1
else:
if verbose:
print_(u"Successfully saved", dst, file=sys.stderr)
return 0
def update(options, filenames):
encoding = options.encoding or getpreferredencoding()
verbose = options.verbose
noupdate = options.noupdate
force_v1 = options.force_v1
remove_v1 = options.remove_v1
def conv(uni):
return uni.encode('iso-8859-1').decode(encoding)
for filename in filenames:
with _sig.block():
if verbose != "quiet":
print_(u"Updating", filename)
if has_id3v1(filename) and not noupdate and force_v1:
mutagen.id3.delete(filename, False, True)
try:
id3 = mutagen.id3.ID3(filename)
except mutagen.id3.ID3NoHeaderError:
if verbose != "quiet":
print_(u"No ID3 header found; skipping...")
continue
except Exception as err:
print_(text_type(err), file=sys.stderr)
continue
for tag in filter(lambda t: t.startswith(("T", "COMM")), id3):
frame = id3[tag]
if isinstance(frame, mutagen.id3.TimeStampTextFrame):
# non-unicode fields
continue
try:
text = frame.text
except AttributeError:
continue
try:
text = [conv(x) for x in frame.text]
except (UnicodeError, LookupError):
continue
else:
frame.text = text
if not text or min(map(isascii, text)):
frame.encoding = 3
else:
frame.encoding = 1
if verbose == "debug":
print_(id3.pprint())
if not noupdate:
if remove_v1:
id3.save(filename, v1=False)
else:
id3.save(filename)
def update(options, filenames):
encoding = options.encoding or getpreferredencoding()
verbose = options.verbose
noupdate = options.noupdate
force_v1 = options.force_v1
remove_v1 = options.remove_v1
def conv(uni):
return uni.encode('iso-8859-1').decode(encoding)
for filename in filenames:
with _sig.block():
if verbose != "quiet":
print_(u"Updating", filename)
if has_id3v1(filename) and not noupdate and force_v1:
mutagen.id3.delete(filename, False, True)
try:
id3 = mutagen.id3.ID3(filename)
except mutagen.id3.ID3NoHeaderError:
if verbose != "quiet":
print_(u"No ID3 header found; skipping...")
continue
except Exception as err:
print_(text_type(err), file=sys.stderr)
continue
for tag in filter(lambda t: t.startswith(("T", "COMM")), id3):
frame = id3[tag]
if isinstance(frame, mutagen.id3.TimeStampTextFrame):
# non-unicode fields
continue
try:
text = frame.text
except AttributeError:
continue
try:
text = [conv(x) for x in frame.text]
except (UnicodeError, LookupError):
continue
else:
frame.text = text
if not text or min(map(isascii, text)):
frame.encoding = 3
else:
frame.encoding = 1
if verbose == "debug":
print_(id3.pprint())
if not noupdate:
if remove_v1:
id3.save(filename, v1=False)
else:
id3.save(filename)
def test_ASFUnicodeAttribute_dunder(self):
attr = ASFUnicodeAttribute(u"foo")
self.assertEqual(bytes(attr), b"f\x00o\x00o\x00")
self.assertEqual(text_type(attr), u"foo")
if PY3:
self.assertEqual(repr(attr), "ASFUnicodeAttribute('foo')")
else:
self.assertEqual(repr(attr), "ASFUnicodeAttribute(u'foo')")
self.assertRaises(TypeError, int, attr)
def copy(src, dst, merge, write_v1=True, excluded_tags=None, verbose=False):
"""Returns 0 on success"""
if excluded_tags is None:
excluded_tags = []
try:
id3 = mutagen.id3.ID3(src, translate=False)
except mutagen.id3.ID3NoHeaderError:
print_(u"No ID3 header found in ", src, file=sys.stderr)
return 1
except Exception as err:
print_(str(err), file=sys.stderr)
return 1
if verbose:
print_(u"File", src, u"contains:", file=sys.stderr)
print_(id3.pprint(), file=sys.stderr)
for tag in excluded_tags:
id3.delall(tag)
if merge:
try:
target = mutagen.id3.ID3(dst, translate=False)
except mutagen.id3.ID3NoHeaderError:
# no need to merge
pass
except Exception as err:
print_(str(err), file=sys.stderr)
return 1
else:
for frame in id3.values():
target.add(frame)
id3 = target
# if the source is 2.3 save it as 2.3
if id3.version < (2, 4, 0):
id3.update_to_v23()
v2_version = 3
else:
id3.update_to_v24()
v2_version = 4
try:
id3.save(dst, v1=(2 if write_v1 else 0), v2_version=v2_version)
except Exception as err:
print_(u"Error saving", dst, u":\n%s" % text_type(err),
file=sys.stderr)
return 1
else:
if verbose:
print_(u"Successfully saved", dst, file=sys.stderr)
return 0
def test_basic(self):
d = os.path.dirname(self.filename)
p = os.path.join(d, fsn(u"%(stream)d.%(ext)s"))
res, out = self.call(fsn(u"--pattern"), p, self.filename)
self.failIf(res)
self.failIf(out)
for stream in [1002429366, 1806412655]:
stream_path = os.path.join(
d, fsn(text_type(stream)) + fsn(u".ogg"))
self.failUnless(os.path.exists(stream_path))
os.unlink(stream_path)
def test_conv_basic(self):
from mutagen.id3 import TALB
for codec in CODECS:
f = ID3(self.filename)
f.add(TALB(text=[AMBIGUOUS.decode("latin-1")], encoding=0))
f.save()
res, out = self.call(
fsn(u"-d"), fsn(u"-e"), fsn(text_type(codec)), self.filename)
f = ID3(self.filename)
self.failUnlessEqual(f["TALB"].encoding, 1)
self.failUnlessEqual(f["TALB"].text[0], AMBIGUOUS.decode(codec))
def test_basic(self):
d = os.path.dirname(self.filename)
p = os.path.join(d, fsn(u"%(stream)d.%(ext)s"))
res, out = self.call(fsn(u"--pattern"), p, self.filename)
self.failIf(res)
self.failIf(out)
for stream in [1002429366, 1806412655]:
stream_path = os.path.join(d,
fsn(text_type(stream)) + fsn(u".ogg"))
self.failUnless(os.path.exists(stream_path))
os.unlink(stream_path)
def test_conv_basic(self):
from mutagen.id3 import TALB
for codec in CODECS:
f = ID3(self.filename)
f.add(TALB(text=[AMBIGUOUS.decode("latin-1")], encoding=0))
f.save()
res, out = self.call(
fsn(u"-d"), fsn(u"-e"), fsn(text_type(codec)), self.filename)
f = ID3(self.filename)
self.failUnlessEqual(f["TALB"].encoding, 1)
self.failUnlessEqual(f["TALB"].text[0], AMBIGUOUS.decode(codec))
def test_ASFUnicodeAttribute(self):
if PY3:
self.assertRaises(TypeError, ASFUnicodeAttribute, b"\xff")
else:
self.assertRaises(ValueError, ASFUnicodeAttribute, b"\xff")
val = u'\xf6\xe4\xfc'
self.assertEqual(ASFUnicodeAttribute(val.encode("utf-8")), val)
self.assertRaises(ASFError, ASFUnicodeAttribute, data=b"\x00")
self.assertEqual(ASFUnicodeAttribute(u"foo").value, u"foo")
self.assertEqual(bytes(ASFUnicodeAttribute(u"foo")),
b"f\x00o\x00o\x00")
self.assertEqual(text_type(ASFUnicodeAttribute(u"foo")), u"foo")
def test_ASFUnicodeAttribute(self):
if PY3:
self.assertRaises(TypeError, ASFUnicodeAttribute, b"\xff")
else:
self.assertRaises(ValueError, ASFUnicodeAttribute, b"\xff")
val = u'\xf6\xe4\xfc'
self.assertEqual(ASFUnicodeAttribute(val.encode("utf-8")), val)
self.assertRaises(ASFError, ASFUnicodeAttribute, data=b"\x00")
self.assertEqual(ASFUnicodeAttribute(u"foo").value, u"foo")
self.assertEqual(
bytes(ASFUnicodeAttribute(u"foo")), b"f\x00o\x00o\x00")
self.assertEqual(
text_type(ASFUnicodeAttribute(u"foo")), u"foo")
def test_timestampspec(self):
from mutagen.id3 import TimeStampSpec, Frame, ID3TimeStamp
s = TimeStampSpec('name')
f = Frame(); f.encoding = 0
self.assertEquals((ID3TimeStamp('ab'), b'fg'), s.read(f, b'ab\x00fg'))
self.assertEquals((ID3TimeStamp('1234'), b''), s.read(f, b'1234\x00'))
self.assertEquals(b'1234\x00', s.write(f, ID3TimeStamp('1234')))
self.assertRaises(AttributeError, s.write, f, None)
if PY3:
self.assertRaises(TypeError, ID3TimeStamp, b"blah")
self.assertEquals(
text_type(ID3TimeStamp(u"2000-01-01")), u"2000-01-01")
self.assertEquals(
bytes(ID3TimeStamp(u"2000-01-01")), b"2000-01-01")
def test_timestampspec(self):
from mutagen.id3 import TimeStampSpec, Frame, ID3TimeStamp
s = TimeStampSpec('name')
f = Frame()
f.encoding = 0
self.assertEquals((ID3TimeStamp('ab'), b'fg'), s.read(f, b'ab\x00fg'))
self.assertEquals((ID3TimeStamp('1234'), b''), s.read(f, b'1234\x00'))
self.assertEquals(b'1234\x00', s.write(f, ID3TimeStamp('1234')))
self.assertRaises(AttributeError, s.write, f, None)
if PY3:
self.assertRaises(TypeError, ID3TimeStamp, b"blah")
self.assertEquals(text_type(ID3TimeStamp(u"2000-01-01")),
u"2000-01-01")
self.assertEquals(bytes(ID3TimeStamp(u"2000-01-01")), b"2000-01-01")
def description(self):
"""string or None if unknown"""
name = None
try:
name = self._TYPE_NAMES[self.audioObjectType]
except IndexError:
pass
if name is None:
return
if self.sbrPresentFlag == 1:
name += "+SBR"
if self.psPresentFlag == 1:
name += "+PS"
return text_type(name)
def test_comm(self):
from mutagen.id3 import COMM
for codec in CODECS:
f = ID3(self.filename)
frame = COMM(desc="", lang="eng", encoding=0,
text=[AMBIGUOUS.decode("latin-1")])
f.add(frame)
f.save()
res, out = self.call(
fsn(u"-d"), fsn(u"-e"), fsn(text_type(codec)), self.filename)
f = ID3(self.filename)
new_frame = f[frame.HashKey]
self.failUnlessEqual(new_frame.encoding, 1)
self.failUnlessEqual(new_frame.text[0], AMBIGUOUS.decode(codec))
def test_comm(self):
from mutagen.id3 import COMM
for codec in CODECS:
f = ID3(self.filename)
frame = COMM(desc="", lang="eng", encoding=0,
text=[AMBIGUOUS.decode("latin-1")])
f.add(frame)
f.save()
res, out = self.call(
fsn(u"-d"), fsn(u"-e"), fsn(text_type(codec)), self.filename)
f = ID3(self.filename)
new_frame = f[frame.HashKey]
self.failUnlessEqual(new_frame.encoding, 1)
self.failUnlessEqual(new_frame.text[0], AMBIGUOUS.decode(codec))
def description(self):
"""string or None if unknown"""
name = None
try:
name = self._TYPE_NAMES[self.audioObjectType]
except IndexError:
pass
if name is None:
return
if self.sbrPresentFlag == 1:
name += "+SBR"
if self.psPresentFlag == 1:
name += "+PS"
return text_type(name)
def main(argv):
from mutagen import File
parser = OptionParser()
parser.add_option("--no-flac", help="Compatibility; does nothing.")
parser.add_option("--no-mp3", help="Compatibility; does nothing.")
parser.add_option("--no-apev2", help="Compatibility; does nothing.")
(options, args) = parser.parse_args(argv[1:])
if not args:
raise SystemExit(parser.print_help() or 1)
for filename in args:
print_(u"--", filename)
try:
print_(u"-", File(filename).pprint())
except AttributeError:
print_(u"- Unknown file type")
except Exception as err:
print_(text_type(err))
print_(u"")
def main(argv):
from mutagen import File
parser = OptionParser()
parser.add_option("--no-flac", help="Compatibility; does nothing.")
parser.add_option("--no-mp3", help="Compatibility; does nothing.")
parser.add_option("--no-apev2", help="Compatibility; does nothing.")
(options, args) = parser.parse_args(argv[1:])
if not args:
raise SystemExit(parser.print_help() or 1)
for filename in args:
print_(u"--", filename)
try:
print_(u"-", File(filename).pprint())
except AttributeError:
print_(u"- Unknown file type")
except Exception as err:
print_(text_type(err))
print_(u"")
def test_compare_to_unicode(self):
frame = self.Frame(encoding=0, text=[u'1987', u'1988'])
self.failUnlessEqual(frame, text_type(frame))
def test_compare_to_unicode(self):
frame = self.Frame(encoding=0, text=[u'1987', u'1988'])
self.failUnlessEqual(frame, text_type(frame))
def __str__(self):
return text_type(self.value)
def __bytes__(self):
return text_type(self.value).encode('utf-8')
def __lt__(self, other):
return text_type(self) < other
def test_ASFBoolAttribute_dunder(self):
attr = ASFBoolAttribute(False)
self.assertEqual(bytes(attr), b"False")
self.assertEqual(text_type(attr), u"False")
self.assertEqual(repr(attr), "ASFBoolAttribute(False)")
self.assertRaises(TypeError, int, attr)
def render_text(name):
value = asf.to_content_description.get(name)
if value is not None:
return text_type(value).encode("utf-16-le") + b"\x00\x00"
else:
return b""
def test_construct(self):
self.assertEqual(text_type(self.TV(u"foo")), u"foo")
if not PY3:
self.assertEqual(text_type(self.TV(b"foo")), u"foo")
self.assertRaises(ValueError, self.TV, b"\xff")
def test_fileobj_name(self):
with tempfile.TemporaryFile(mode="rb") as h:
self.assertEqual(fileobj_name(h), text_type(h.name))
def write_files(edits, filenames, escape):
# unescape escape sequences and decode values
encoded_edits = []
for frame, value in edits:
if not value:
continue
try:
frame = frame_from_fsnative(frame)
except ValueError as err:
print_(text_type(err), file=sys.stderr)
assert isinstance(frame, str)
# strip "--"
frame = frame[2:]
try:
value = value_from_fsnative(value, escape)
except ValueError as err:
error(u"%s: %s" % (frame, text_type(err)))
assert isinstance(value, text_type)
encoded_edits.append((frame, value))
edits = encoded_edits
# preprocess:
# for all [frame,value] pairs in the edits list
# gather values for identical frames into a list
tmp = {}
for frame, value in edits:
if frame in tmp:
tmp[frame].append(value)
else:
tmp[frame] = [value]
# edits is now a dictionary of frame -> [list of values]
edits = tmp
# escape also enables escaping of the split separator
if escape:
string_split = split_escape
else:
string_split = lambda s, *args, **kwargs: s.split(*args, **kwargs)
for filename in filenames:
with _sig.block():
if verbose:
print_(u"Writing", filename, file=sys.stderr)
try:
id3 = mutagen.id3.ID3(filename)
except mutagen.id3.ID3NoHeaderError:
if verbose:
print_(u"No ID3 header found; creating a new tag",
file=sys.stderr)
id3 = mutagen.id3.ID3()
except Exception as err:
print_(str(err), file=sys.stderr)
continue
for (frame, vlist) in edits.items():
if frame == "POPM":
for value in vlist:
values = string_split(value, ":")
if len(values) == 1:
email, rating, count = values[0], 0, 0
elif len(values) == 2:
email, rating, count = values[0], values[1], 0
else:
email, rating, count = values
frame = mutagen.id3.POPM(email=email,
rating=int(rating),
count=int(count))
id3.add(frame)
elif frame == "APIC":
for value in vlist:
values = string_split(value, ":")
# FIXME: doesn't support filenames with an invalid
# encoding since we have already decoded at that point
fn = values[0]
if len(values) >= 2:
desc = values[1]
else:
desc = u"cover"
if len(values) >= 3:
try:
picture_type = int(values[2])
except ValueError:
error(u"Invalid picture type: %r" % values[1])
else:
picture_type = PictureType.COVER_FRONT
if len(values) >= 4:
mime = values[3]
else:
mime = mimetypes.guess_type(fn)[0] or "image/jpeg"
if len(values) >= 5:
error("APIC: Invalid format")
encoding = get_frame_encoding(frame, desc)
try:
with open(fn, "rb") as h:
data = h.read()
except IOError as e:
error(text_type(e))
frame = mutagen.id3.APIC(encoding=encoding,
mime=mime,
desc=desc,
type=picture_type,
data=data)
id3.add(frame)
elif frame == "COMM":
for value in vlist:
values = string_split(value, ":")
if len(values) == 1:
value, desc, lang = values[0], "", "eng"
elif len(values) == 2:
desc, value, lang = values[0], values[1], "eng"
else:
value = ":".join(values[1:-1])
desc, lang = values[0], values[-1]
frame = mutagen.id3.COMM(encoding=3,
text=value,
lang=lang,
desc=desc)
id3.add(frame)
elif frame == "UFID":
for value in vlist:
values = string_split(value, ":")
if len(values) != 2:
error(u"Invalid value: %r" % values)
owner = values[0]
data = values[1].encode("utf-8")
frame = mutagen.id3.UFID(owner=owner, data=data)
id3.add(frame)
elif frame == "TXXX":
for value in vlist:
values = string_split(value, ":", 1)
if len(values) == 1:
desc, value = "", values[0]
else:
desc, value = values[0], values[1]
frame = mutagen.id3.TXXX(encoding=3,
text=value,
desc=desc)
id3.add(frame)
elif issubclass(mutagen.id3.Frames[frame],
mutagen.id3.UrlFrame):
frame = mutagen.id3.Frames[frame](encoding=3, url=vlist)
id3.add(frame)
else:
frame = mutagen.id3.Frames[frame](encoding=3, text=vlist)
id3.add(frame)
id3.save(filename)
def write_files(edits, filenames, escape):
# unescape escape sequences and decode values
encoded_edits = []
for frame, value in edits:
if not value:
continue
try:
frame = frame_from_fsnative(frame)
except ValueError as err:
print_(text_type(err), file=sys.stderr)
assert isinstance(frame, str)
# strip "--"
frame = frame[2:]
try:
value = value_from_fsnative(value, escape)
except ValueError as err:
error(u"%s: %s" % (frame, text_type(err)))
assert isinstance(value, text_type)
encoded_edits.append((frame, value))
edits = encoded_edits
# preprocess:
# for all [frame,value] pairs in the edits list
# gather values for identical frames into a list
tmp = {}
for frame, value in edits:
if frame in tmp:
tmp[frame].append(value)
else:
tmp[frame] = [value]
# edits is now a dictionary of frame -> [list of values]
edits = tmp
# escape also enables escaping of the split separator
if escape:
string_split = split_escape
else:
string_split = lambda s, *args, **kwargs: s.split(*args, **kwargs)
for filename in filenames:
with _sig.block():
if verbose:
print_(u"Writing", filename, file=sys.stderr)
try:
id3 = mutagen.id3.ID3(filename)
except mutagen.id3.ID3NoHeaderError:
if verbose:
print_(u"No ID3 header found; creating a new tag",
file=sys.stderr)
id3 = mutagen.id3.ID3()
except Exception as err:
print_(str(err), file=sys.stderr)
continue
for (frame, vlist) in edits.items():
if frame == "POPM":
for value in vlist:
values = string_split(value, ":")
if len(values) == 1:
email, rating, count = values[0], 0, 0
elif len(values) == 2:
email, rating, count = values[0], values[1], 0
else:
email, rating, count = values
frame = mutagen.id3.POPM(
email=email, rating=int(rating), count=int(count))
id3.add(frame)
elif frame == "APIC":
for value in vlist:
values = string_split(value, ":")
# FIXME: doesn't support filenames with an invalid
# encoding since we have already decoded at that point
fn = values[0]
if len(values) >= 2:
desc = values[1]
else:
desc = u"cover"
if len(values) >= 3:
try:
picture_type = int(values[2])
except ValueError:
error(u"Invalid picture type: %r" % values[1])
else:
picture_type = PictureType.COVER_FRONT
if len(values) >= 4:
mime = values[3]
else:
mime = mimetypes.guess_type(fn)[0] or "image/jpeg"
if len(values) >= 5:
error("APIC: Invalid format")
encoding = get_frame_encoding(frame, desc)
try:
with open(fn, "rb") as h:
data = h.read()
except IOError as e:
error(text_type(e))
frame = mutagen.id3.APIC(encoding=encoding, mime=mime,
desc=desc, type=picture_type, data=data)
id3.add(frame)
elif frame == "COMM":
for value in vlist:
values = string_split(value, ":")
if len(values) == 1:
value, desc, lang = values[0], "", "eng"
elif len(values) == 2:
desc, value, lang = values[0], values[1], "eng"
else:
value = ":".join(values[1:-1])
desc, lang = values[0], values[-1]
frame = mutagen.id3.COMM(
encoding=3, text=value, lang=lang, desc=desc)
id3.add(frame)
elif frame == "USLT":
for value in vlist:
values = string_split(value, ":")
if len(values) == 1:
value, desc, lang = values[0], "", "eng"
elif len(values) == 2:
desc, value, lang = values[0], values[1], "eng"
else:
value = ":".join(values[1:-1])
desc, lang = values[0], values[-1]
frame = mutagen.id3.USLT(
encoding=3, text=value, lang=lang, desc=desc)
id3.add(frame)
elif frame == "UFID":
for value in vlist:
values = string_split(value, ":")
if len(values) != 2:
error(u"Invalid value: %r" % values)
owner = values[0]
data = values[1].encode("utf-8")
frame = mutagen.id3.UFID(owner=owner, data=data)
id3.add(frame)
elif frame == "TXXX":
for value in vlist:
values = string_split(value, ":", 1)
if len(values) == 1:
desc, value = "", values[0]
else:
desc, value = values[0], values[1]
frame = mutagen.id3.TXXX(
encoding=3, text=value, desc=desc)
id3.add(frame)
elif frame == "WXXX":
for value in vlist:
values = string_split(value, ":", 1)
if len(values) == 1:
desc, value = "", values[0]
else:
desc, value = values[0], values[1]
frame = mutagen.id3.WXXX(
encoding=3, url=value, desc=desc)
id3.add(frame)
elif issubclass(mutagen.id3.Frames[frame],
mutagen.id3.UrlFrame):
frame = mutagen.id3.Frames[frame](
encoding=3, url=vlist[-1])
id3.add(frame)
else:
frame = mutagen.id3.Frames[frame](encoding=3, text=vlist)
id3.add(frame)
id3.save(filename)
def test_str(self):
self.assertEqual(text_type(self.value), u"foo\x00bar\x00baz")
def render_text(name):
value = asf.to_content_description.get(name)
if value is not None:
return text_type(value).encode("utf-16-le") + b"\x00\x00"
else:
return b""
def __eq__(self, other):
return text_type(self) == other
def test_construct(self):
self.assertEqual(text_type(self.TV(u"foo")), u"foo")
if not PY3:
self.assertEqual(text_type(self.TV(b"foo")), u"foo")
self.assertRaises(ValueError, self.TV, b"\xff")
def test_ASFQWordAttribute_dunder(self):
attr = ASFQWordAttribute(data=b"\x00" * 8)
self.assertEqual(bytes(attr), b"0")
self.assertEqual(text_type(attr), u"0")
self.assertEqual(repr(attr).replace("0L", "0"), "ASFQWordAttribute(0)")
self.assertEqual(int(attr), 0)
def test_str(self):
self.assertEqual(text_type(self.value), u"foo\x00bar\x00baz")
def test_fileobj_name(self):
with tempfile.TemporaryFile(mode="rb") as h:
self.assertEqual(fileobj_name(h), text_type(h.name))
