From 8691b834ccf816f8fce485913354554445c49fa3 Mon Sep 17 00:00:00 2001 From: Thomas Scholtes Date: Mon, 10 Feb 2014 16:31:40 +0100 Subject: [PATCH] Image storage style have bytearray type --- beets/mediafile.py | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/beets/mediafile.py b/beets/mediafile.py index b0de8aebe..322964079 100644 --- a/beets/mediafile.py +++ b/beets/mediafile.py @@ -403,7 +403,7 @@ class StorageStyle(object): value = value.decode('utf8', 'ignore') else: value = unicode(value) - elif self.as_type in (bool, str, int): + else: value = self.as_type(value) if self.suffix: @@ -523,6 +523,7 @@ class MP4ImageStorageStyle(MP4ListStorageStyle): def __init__(self, **kwargs): super(MP4ImageStorageStyle, self).__init__(key='covr', **kwargs) + self.as_type = bytearray def store(self, mediafile, images): covers = [self._mp4_cover(image) for image in images] @@ -547,9 +548,6 @@ class MP4ImageStorageStyle(MP4ListStorageStyle): return mutagen.mp4.MP4Cover(data, kind) - def serialize(self, value): - return bytearray(value) - class MP3StorageStyle(StorageStyle): @@ -645,6 +643,7 @@ class MP3ImageStorageStyle(ListStorageStyle, MP3StorageStyle): def __init__(self): super(MP3ImageStorageStyle, self).__init__(key='APIC') + self.as_type = bytearray def fetch(self, mediafile): try: @@ -664,9 +663,6 @@ class MP3ImageStorageStyle(ListStorageStyle, MP3StorageStyle): ) mediafile.mgfile.tags.setall(self.key, [frame]) - def serialize(self, value): - return bytearray(value) - # The field itself. class MediaField(object):