def test_handle_picture_block(self): pic = Picture() pic.data = self.__get_jpeg() pic.type = 3 b64pic_cover = base64.b64encode(pic.write()) pic2 = Picture() pic2.data = self.__get_jpeg(size=6) pic2.type = 4 b64pic_other= base64.b64encode(pic2.write()) song = self.MutagenType(self.filename) song["metadata_block_picture"] = [b64pic_other, b64pic_cover] song.save() song = self.QLType(self.filename) self.failUnlessEqual(song("~picture"), "y") fn = song.find_cover() self.failUnlessEqual(pic.data, fn.read()) song.write() song = self.MutagenType(self.filename) self.failUnless(b64pic_other in song["metadata_block_picture"]) self.failUnless(b64pic_cover in song["metadata_block_picture"]) song["metadata_block_picture"] = [b64pic_other] song.save() song = self.QLType(self.filename) fn = song.find_cover() self.failUnlessEqual(pic2.data, fn.read())
def test_get_images(self): pic = Picture() pic.data = _get_jpeg() pic.type = APICType.COVER_FRONT b64pic_cover = base64.b64encode(pic.write()) # metadata_block_picture song = FLAC(self.filename) song["metadata_block_picture"] = [b64pic_cover] song.save() song = FLACFile(self.filename) self.assertEqual(len(song.get_images()), 1) self.assertEqual(song.get_images()[0].type, APICType.COVER_FRONT) # flac Picture song = FLAC(self.filename) pic = Picture() pic.data = _get_jpeg() pic.type = APICType.COVER_BACK song.add_picture(pic) song.save() song = FLACFile(self.filename) self.assertEqual(len(song.get_images()), 2) self.assertEqual(song.get_images()[-1].type, APICType.COVER_BACK)
def test_handle_picture_block(self): pic = Picture() pic.data = _get_jpeg() pic.type = APICType.COVER_FRONT b64pic_cover = base64.b64encode(pic.write()) pic2 = Picture() pic2.data = _get_jpeg(size=6) pic2.type = APICType.COVER_BACK b64pic_other = base64.b64encode(pic2.write()) song = self.MutagenType(self.filename) song["metadata_block_picture"] = [b64pic_other, b64pic_cover] song.save() song = self.QLType(self.filename) self.failUnlessEqual(song("~picture"), "y") fn = song.get_primary_image().file self.failUnlessEqual(pic.data, fn.read()) song.write() song = self.MutagenType(self.filename) self.failUnless(b64pic_other in song["metadata_block_picture"]) self.failUnless(b64pic_cover in song["metadata_block_picture"]) song["metadata_block_picture"] = [b64pic_other] song.save() song = self.QLType(self.filename) fn = song.get_primary_image().file self.failUnlessEqual(pic2.data, fn.read())
def download_and_fix_ogg(ogg, audio_metadata, cover_art_file): global DRY_RUN if DRY_RUN: print "This is a dry run. So pretending to download the ogg..." return "/tmp/ogg" print "Now downloading the ogg in order to set the metadata in it..." if not LIVE and len(sys.argv) >= 6 and os.path.exists(sys.argv[5]): ogg_local_fn = sys.argv[5] print "(using presupplied file %s)" % ogg_local_fn else: f, metadata = client.get_file_and_metadata(ogg) ogg_local_fn = fetch_file(f, metadata) print "Successfully downloaded (to %s): now editing metadata..." % ogg_local_fn audio = OggVorbis(ogg_local_fn) for k in audio_metadata.keys(): audio[k] = audio_metadata[k] # add cover art im=Image.open(cover_art_file) w,h=im.size p=Picture() imdata=open(cover_art_file,'rb').read() p.data=imdata p.type=3 p.desc='' p.mime='image/jpeg'; p.width=w; p.height=h p.depth=24 dt=p.write(); enc=base64.b64encode(dt).decode('ascii'); audio['metadata_block_picture']=[enc]; audio.save() print "Successfully updated metadata." return ogg_local_fn
def test_get_images(self): # coverart + coverartmime data = _get_jpeg() song = self.MutagenType(self.filename) song["coverart"] = base64.b64encode(data) song["coverartmime"] = "image/jpeg" song.save() song = self.QLType(self.filename) self.assertEqual(len(song.get_images()), 1) self.assertEqual(song.get_images()[0].mime_type, "image/jpeg") # metadata_block_picture pic = Picture() pic.data = _get_jpeg() pic.type = APICType.COVER_FRONT b64pic_cover = base64.b64encode(pic.write()) song = self.MutagenType(self.filename) song["metadata_block_picture"] = [b64pic_cover] song.save() song = self.QLType(self.filename) self.assertEqual(len(song.get_images()), 2) self.assertEqual(song.get_images()[0].type, APICType.COVER_FRONT)
def test_largest_valid(self): f = FLAC(self.filename) pic = Picture() pic.data = b"\x00" * (2 ** 24 - 1 - 32) self.assertEqual(len(pic.write()), 2 ** 24 - 1) f.add_picture(pic) f.save()
def set_image(self, image): """Replaces all embedded images by the passed image""" with translate_errors(): audio = self.MutagenType(self["~filename"]) try: data = image.read() except EnvironmentError as e: raise AudioFileError(e) pic = Picture() pic.data = data pic.type = APICType.COVER_FRONT pic.mime = image.mime_type pic.width = image.width pic.height = image.height pic.depth = image.color_depth audio.pop("coverart", None) audio.pop("coverartmime", None) audio["metadata_block_picture"] = base64.b64encode( pic.write()).decode("ascii") with translate_errors(): audio.save() self.has_images = True
def set_image(self, image): """Replaces all embedded images by the passed image""" with translate_errors(): tag = FLAC(self["~filename"]) try: data = image.read() except EnvironmentError as e: raise AudioFileError(e) pic = Picture() pic.data = data pic.type = APICType.COVER_FRONT pic.mime = image.mime_type pic.width = image.width pic.height = image.height pic.depth = image.color_depth tag.add_picture(pic) with translate_errors(): tag.save() # clear vcomment tags super(FLACFile, self).clear_images() self.has_images = True
def set_art(self, art_file): art_tag = mutagen.File(self.path, None, False) if type(art_tag) is mutagen.mp3.MP3: with open(art_file, 'rb') as f: image = f.read() art_tag.tags.delall('APIC') art_tag.tags.add(APIC( encoding=3, # 3 is for utf-8 mime='image/jpg', # image/jpeg or image/png type=3, # 3 is for the cover image desc=u'Cover', data=image )) art_tag.save() print 'Wrote embedded MP3 art.' if type(art_tag) is mutagen.flac.FLAC: image = Picture() image.type = 3 image.mime = 'image/jpg' image.desc = 'front cover' with open(art_file, 'rb') as f: image.data = f.read() art_tag.clear_pictures() art_tag.add_picture(image) art_tag.save() print 'Wrote embedded FLAC art'
def test_write_tags(tmpdir): """Test writing tags from a FLAC to mp3 file.""" # Prepare. flac = tmpdir.mkdir('flac').join('song.flac').ensure(file=True) mp3 = tmpdir.mkdir('mp3').join('song.mp3').ensure(file=True) with open(os.path.join(os.path.dirname(__file__), '1khz_sine.flac'), 'rb') as f: flac.write(f.read(), 'wb') with open(os.path.join(os.path.dirname(__file__), '1khz_sine.mp3'), 'rb') as f: mp3.write(f.read(), 'wb') flac, mp3 = str(flac.realpath()), str(mp3.realpath()) tags = FLAC(flac) tags.update(dict(artist='Artist2', date='2012', album='Album', tracknumber='01', title='Title', unsyncedlyrics='L')) image = Picture() image.type, image.mime = 3, 'image/jpeg' with open(os.path.join(os.path.dirname(__file__), '1_album_art.jpg'), 'rb') as f: image.data = f.read() tags.add_picture(image) tags.save() # Test. ConvertFiles.write_tags(flac, mp3) # Check. id3 = ID3(mp3) assert 'Artist2' == id3['TPE1'] assert '2012' == id3['TDRC'] assert 'Album' == id3['TALB'] assert '01' == id3['TRCK'] assert 'Title' == id3['TIT2'] assert 'L' == id3["USLT:Lyrics:'eng'"].text with open(os.path.join(os.path.dirname(__file__), '1_album_art.jpg'), 'rb') as f: assert f.read() == id3['APIC:'].data assert ({}, [], [], []) == find_files(str(tmpdir.join('flac')), str(tmpdir.join('mp3')))
def _add_picture(self, mime, type_, desc, data): picture = Picture() picture.data = data picture.mime = mime picture.type = type_ picture.desc = desc self._flac.add_picture(picture)
def export_digital(item): """ Export and/or split digital files for an item (parallelized) """ logging.warning('{0} -> {1}'.format(item.name, DIGITAL_FOLDER)) if 'std' in item.option: if JUST_ADD_TAGS: if item.digital_ext == 'mp3': mutagen_audio = EasyID3(item.digital_file_path) elif item.digital_ext == 'flac': mutagen_audio = FLAC(item.digital_file_path) else: raise Exception('Format {0} not recognized for item {1}.'.format(item.digital_ext, item.name)) mutagen_audio['title'] = item.album mutagen_audio['artist'] = item.artist mutagen_audio['albumartist'] = item.artist mutagen_audio['album'] = item.album mutagen_audio.save() else: # Export audio file digital_file = AudioSegment.from_wav(item.path) digital_file.export(out_f=item.digital_file_path, format=item.digital_ext, bitrate='192k', tags={ 'title':(item.album or item.artist), 'artist':item.artist, 'albumartist':item.artist, 'album':(item.album or item.artist)}) # Add cover art if item.thumb and (item.digital_ext == 'mp3'): mutagen_audio = MP3(item.digital_file_path, ID3=ID3) try: # Add ID3 tag if it doesn't exist mutagen_audio.add_tags() except error: pass mutagen_audio.tags.add( APIC( encoding=3, # 3 is for utf-8 mime='image/jpeg', # image/jpeg or image/png type=3, # 3 is for the cover image desc=u'Cover', data=open(item.thumb_path, 'rb').read() ) ) mutagen_audio.save() elif item.thumb and (item.digital_ext == 'flac'): mutagen_audio = File(item.digital_file_path) flac_image = Picture() flac_image.type = 3 mime = 'image/jpeg' flac_image.desc = 'Cover' with open(item.thumb_path, 'rb') as f: flac_image.data = f.read() mutagen_audio.add_picture(flac_image) mutagen_audio.save() else: logging.warning('No cover found for item {0}'.format(item.name)) if JUST_ADD_TAGS: os.rename(item.digital_file_path, item.digital_rename_path) if DROPBOX_COPY: # Copy finished digital file to Dropbox order folder shutil.copy2(item.digital_file_path, item.dropbox_order_folder) # Deluxe / 45 else: split.split_item(item, DIGITAL_FOLDER, DROPBOX_COPY)
def handle_audio_file(audio_file): print('Handling audio file', audio_file) extension = os.path.splitext(os.path.basename(audio_file))[1] if extension == '.mp3': mp3 = MP3(audio_file, ID3=ID3) print(list(dict(mp3).keys())) if not regex_list(dict(mp3).keys(), re.compile('[Aa][Pp][Ii][Cc].*')): artist = mp3['TPE1'][0] album = mp3['TALB'][0] cover_data = open(get_cover_art(artist, album, audio_file), mode='rb') apic = APIC() apic.encoding = 3 apic.mime = 'image/jpg' apic.type = PictureType.COVER_FRONT apic.desc = u'Cover' apic.data = cover_data.read() cover_data.close() print('Adding cover art', cover_data.name, '->', audio_file) mp3['APIC'] = apic mp3.save() else: print(audio_file, 'already has cover artwork.') elif extension == '.m4a' or extension is '.aac': m4a = MP4(audio_file) print(list(dict(m4a).keys())) if 'covr' not in m4a: artist = m4a['\xa9ART'][0] album = m4a['\xa9alb'][0] cover_data = open(get_cover_art(artist, album, audio_file), mode='rb') covr = MP4Cover() covr.imageformat = AtomDataType.JPEG covr.data = cover_data.read() cover_data.close() print('Adding cover art', cover_data.name, '->', audio_file) m4a['covr'] = [covr] m4a.save() else: print(audio_file, 'already has cover artwork.') elif extension == '.flac': flac = FLAC(audio_file) print(list(dict(flac).keys())) if not flac.pictures: artist = flac['artist'][0] album = flac['album'][0] cover_data = open(get_cover_art(artist, album, audio_file), mode='rb') picture = Picture() picture.type = 3 picture.mime = 'image/jpg' picture.desc = u'Cover' picture.data = cover_data.read() cover_data.close() print('Adding cover artwork', cover_data.name, '->', audio_file) flac.add_picture(picture) flac.save() else: print(audio_file, 'already has cover artwork.') move_or_overwrite(audio_file, dest_audio, os.path.join(dest_audio, os.path.basename(audio_file)))
def embed_cover_art(self, audio_file, cover_file): """Embeds cover art into an audio file. Arguments: audio_file (str): The path to the audio file to embed the artwork in. cover_file (str): The path to the artwork file to embed. """ if path.isfile(audio_file) and path.isfile(cover_file): mimetype = "image/png" if cover_file.endswith("png") else "image/jpeg" artwork = open(cover_file, "rb").read() desc = u"Cover Art" # Determine which filetype we're handling if audio_file.endswith("m4a"): audio = MP4(audio_file) covr = [] if cover_file.endswith("png"): covr.append(MP4Cover(artwork, MP4Cover.FORMAT_PNG)) else: covr.append(MP4Cover(artwork, MP4Cover.FORMAT_JPEG)) audio.tags["covr"] = covr elif audio_file.endswith("mp3"): audio = MP3(audio_file, ID3=ID3) # Add ID3 tags if they don't exist try: audio.add_tags() except error: pass audio.tags.add( APIC( encoding=3, # 3 is UTF-8 mime=mimetype, type=3, # 3 is for cover artwork desc=desc, data=artwork, ) ) elif audio_file.endswith("flac"): audio = FLAC(audio_file) image = Picture() image.type = 3 # 3 is for cover artwork image.mime = mimetype image.desc = desc image.data = artwork audio.clear_pictures() # Clear existing pictures audio.add_picture(image) # Save the audio file audio.save()
def test_picture(self): data = "abc" song = FLAC(self.filename) pic = Picture() pic.data = data song.add_picture(pic) song.save() song = FLACFile(self.filename) self.failUnless(song("~picture")) fn = song.find_cover() self.failUnlessEqual(fn.read(), pic.data)
def test_get_image(self): data = b"abc" song = FLAC(self.filename) pic = Picture() pic.data = data song.add_picture(pic) song.save() song = FLACFile(self.filename) self.failUnless(song("~picture")) fn = song.get_primary_image().file self.failUnlessEqual(fn.read(), pic.data)
def tag(album, track, file_path, cover_path): lyrics = get_lyrics(track['lyrics_tp'], track['track_id']) meta = utils.organize_meta(album=album, track=track, lyrics=lyrics) if str(file_path).endswith('.flac'): if cover_path: f_file = FLAC(file_path) f_image = Picture() f_image.type = 3 f_image.desc = 'Front Cover' with open(cover_path, 'rb') as f: f_image.data = f.read() f_file.add_picture(f_image) f_file.save() f = FLAC(file_path) logger_bugs.debug("Writing tags to {}".format( os.path.basename(file_path))) for k, v in meta.items(): f[k] = str(v) f.save() if str(file_path).endswith('.mp3'): legend = { "ALBUM": id3.TALB, "ALBUMARTIST": id3.TPE2, "ARTIST": id3.TPE1, "TRACKNUMBER": id3.TRCK, "DISCNUMBER": id3.TPOS, "COMMENT": id3.COMM, "COMPOSER": id3.TCOM, "COPYRIGHT": id3.TCOP, "DATE": id3.TDRC, "GENRE": id3.TCON, "ISRC": id3.TSRC, "LABEL": id3.TPUB, "PERFORMER": id3.TOPE, "TITLE": id3.TIT2, "LYRICS": id3.USLT } try: audio = id3.ID3(file_path) except ID3NoHeaderError: audio = id3.ID3() logger_bugs.debug("Writing tags to {}".format( os.path.basename(file_path))) for k, v in meta.items(): try: id3tag = legend[k] audio[id3tag.__name__] = id3tag(encoding=3, text=v) except KeyError: pass if cover_path: with open(cover_path, 'rb') as cov_obj: audio.add(id3.APIC(3, 'image/jpg', 3, '', cov_obj.read())) audio.save(file_path, 'v2_version=3')
def test_get_image(self): data = b"abc" song = FLAC(self.filename) pic = Picture() pic.data = data song.add_picture(pic) song.save() song = FLACFile(self.filename) self.failUnless(song("~picture")) fn = song.get_primary_image().file self.failUnlessEqual(fn.read(), pic.data)
def write_tags(f, meta, tag_cfg, cov, ext, embed_cov): if ext == ".flac": audio = FLAC(f) for k, v in meta.items(): if tag_cfg[k]: audio[k] = str(v) if embed_cov and cov: with open(cov, 'rb') as cov_obj: image = Picture() image.type = 3 image.mime = "image/jpeg" image.data = cov_obj.read() audio.add_picture(image) audio.save() elif ext == ".mp3": try: audio = id3.ID3(f) except ID3NoHeaderError: audio = id3.ID3() if tag_cfg['TRACKNUMBER'] and tag_cfg['TRACKTOTAL']: audio['TRCK'] = id3.TRCK(encoding=3, text=str(meta['TRACKNUMBER']) + "/" + str(meta['TRACKTOTAL'])) elif tag_cfg['TRACKNUMBER']: audio['TRCK'] = id3.TRCK(encoding=3, text=str(meta['TRACKNUMBER'])) if tag_cfg['DISCNUMBER']: audio['TPOS'] = id3.TPOS(encoding=3, text=str(meta['DISCNUMBER']) + "/" + str(meta['DISCTOTAL'])) legend={ "ALBUM": id3.TALB, "ALBUMARTIST": id3.TPE2, "ARTIST": id3.TPE1, "COMMENT": id3.COMM, "COMPOSER": id3.TCOM, "COPYRIGHT": id3.TCOP, "DATE": id3.TDAT, "GENRE": id3.TCON, "ISRC": id3.TSRC, "LABEL": id3.TPUB, "PERFORMER": id3.TOPE, "TITLE": id3.TIT2, # Not working. "URL": id3.WXXX, "YEAR": id3.TYER} for k, v in meta.items(): try: if tag_cfg[k]: id3tag = legend[k] audio[id3tag.__name__] = id3tag(encoding=3, text=v) except KeyError: pass if embed_cov and cov: with open(cov, 'rb') as cov_obj: audio.add(id3.APIC(3, 'image/jpeg', 3, '', cov_obj.read())) audio.save(f, 'v2_version=3')
def write_tags(pre_abs, meta, fmt, cov_abs): if fmt == "FLAC": audio = FLAC(pre_abs) for k, v in meta.items(): if v: audio[k] = str(v) if cov_abs: with open(cov_abs, "rb") as f: image = Picture() image.type = 3 image.mime = "image/jpeg" image.data = f.read() audio.add_picture(image) elif fmt == "MP3": try: audio = id3.ID3(pre_abs) except ID3NoHeaderError: audio = id3.ID3() audio['TRCK'] = id3.TRCK( encoding=3, text="{}/{}".format(meta['track'], meta['tracktotal']) ) legend={ "album": id3.TALB, "albumartist": id3.TPE2, "artist": id3.TPE1, #"comment": id3.COMM, "copyright": id3.TCOP, "isrc": id3.TSRC, "label": id3.TPUB, "title": id3.TIT2, "year": id3.TYER } for k, v in meta.items(): id3tag = legend.get(k) if v and id3tag: audio[id3tag.__name__] = id3tag(encoding=3, text=v) if cov_abs: with open(cov_abs, "rb") as cov_obj: audio.add(id3.APIC(3, "image/jpeg", 3, None, cov_obj.read())) else: audio = MP4(pre_abs) audio['\xa9nam'] = meta['title'] audio['\xa9alb'] = meta['album'] audio['aART'] = meta['albumartist'] audio['\xa9ART'] = meta['artist'] audio['trkn'] = [(meta['track'], meta['tracktotal'])] audio['\xa9day'] = meta['year'] audio['cprt'] = meta['copyright'] if cov_abs: with open(cov_abs, "rb") as f: audio['covr'] = [MP4Cover(f.read(), imageformat = MP4Cover.FORMAT_JPEG)] audio.save(pre_abs)
def add_flac_tags(path, tags, image, lyrics=None, image_mimetype='image/jpg'): tag = FLAC(path) pic = Picture() pic.data = image pic.type = 3 pic.mime = image_mimetype tag.add_picture(pic) for key, val in tags.items(): try: tag[key] = str(val) except Exception as e: print(e) tag.save()
def buttonAddPictureClicked(self): self.logger("buttonAddPictureClicked") filename = QtGui.QFileDialog.getOpenFileName(self, "Image File") reader = QtGui.QImageReader(filename) rformat = reader.format() image = reader.read() self.coverArtPixmap = QtGui.QPixmap.fromImage(image) self.logger( 'width = %d, height = %d' % (self.coverArtPixmap.width(), self.coverArtPixmap.height())) self.labelPicture.setPixmap( self.coverArtPixmap.scaled(self.labelPicture.width(), self.labelPicture.height(), QtCore.Qt.KeepAspectRatio)) pic = Picture() data = Qt.QByteArray() buf = Qt.QBuffer(data) pic.type = 3 # APICType.COVER_FRONT self.logger('format: %s' % rformat) if rformat == 'png': pic.mime = 'image/png' image.save(buf, 'PNG') elif rformat == 'jpg' or rformat == 'jpeg': pic.mime = 'image/jpeg' image.save(buf, 'JPG') else: pic.mime = 'image/unknown' self.logger("format: %s" % pic.mime) pic.data = data.data() pic.width = self.coverArtPixmap.width() pic.height = self.coverArtPixmap.height() pic.depth = self.coverArtPixmap.depth() pic.colors = image.colorCount() if len(self.audio.pictures) > 0: ret = QtGui.QMessageBox.warning( self, "FLAC Tagger", "This FLAC file already contains one or more pictures.\n" "Do you want to replace them?", QtGui.QMessageBox.Yes | QtGui.QMessageBox.Cancel, QtGui.QMessageBox.Cancel) if ret == QtGui.QMessageBox.Yes: self.audio.clear_pictures() self.audio.addPicture(pic) else: # make sure the old picture is displayed again self.displayPictures() else: self.audio.addPicture(pic)
def update_flac(self, path: str, track: beatport.Track): f = FLAC(path) if UpdatableTags.title in self.config.update_tags and self.config.overwrite: f['TITLE'] = track.title if UpdatableTags.artist in self.config.update_tags and self.config.overwrite: f['ARTIST'] = self.config.artist_separator.join( [a.name for a in track.artists]) if UpdatableTags.album in self.config.update_tags and ( self.config.overwrite or f.get('ALBUM') == None): f['ALBUM'] = track.album.name if UpdatableTags.label in self.config.update_tags and ( self.config.overwrite or f.get('LABEL') == None): f['LABEL'] = track.label.name if UpdatableTags.bpm in self.config.update_tags and ( self.config.overwrite or f.get('BPM') == None): f['BPM'] = str(track.bpm) if UpdatableTags.genre in self.config.update_tags and ( self.config.overwrite or f.get('GENRE') == None): f['GENRE'] = ', '.join([g.name for g in track.genres]) if UpdatableTags.date in self.config.update_tags and ( self.config.overwrite or f.get('DATE') == None): f['DATE'] = track.release_date.strftime('%Y-%m-%d') #Year - part of date if UpdatableTags.key in self.config.update_tags and ( self.config.overwrite or f.get('INITIALKEY') == None): f['INITIALKEY'] = track.id3key() if UpdatableTags.publishdate in self.config.update_tags and ( self.config.overwrite or f.get('ORIGINALDATE') == None): f['ORIGINALDATE'] = str(track.publish_date.year) #Other tags if UpdatableTags.other in self.config.update_tags: f['WWWAUDIOFILE'] = track.url() f['WWWPUBLISHER'] = track.label.url('label') #Redownlaod cover if self.config.replace_art: try: url = track.art(self.config.art_resolution) r = requests.get(url) image = Picture() image.type = 3 image.mime = 'image/jpeg' image.desc = 'Cover' image.data = r.content f.clear_pictures() f.add_picture(image) except Exception: logging.warning('Error downloading cover for file: ' + path) f.save()
def write_tags(song, data): try: tag = FLAC(song) tag.delete() images = Picture() images.type = 3 images.data = data['image'] tag.clear_pictures() tag.add_picture(images) tag['lyrics'] = data['lyric'] except FLACNoHeaderError: try: tag = File(song, easy=True) except: return tag['artist'] = data['artist'] tag['title'] = data['music'] tag['date'] = data['year'] tag['album'] = data['album'] tag['tracknumber'] = data['tracknum'] tag['discnumber'] = data['discnum'] tag['genre'] = data['genre'] tag['albumartist'] = data['ar_album'] tag['author'] = data['author'] tag['composer'] = data['composer'] tag['copyright'] = data['copyright'] tag['bpm'] = data['bpm'] tag['length'] = data['duration'] tag['organization'] = data['label'] tag['isrc'] = data['isrc'] tag['replaygain_*_gain'] = data['gain'] tag['lyricist'] = data['lyricist'] tag.save() try: audio = ID3(song) audio.add( APIC(encoding=3, mime="image/jpeg", type=3, desc=u"Cover", data=data['image'])) audio.add( USLT(encoding=3, lang=u"eng", desc=u"desc", text=data['lyric'])) audio.save() except _util.ID3NoHeaderError: pass
def set_image(self, image): pic = Picture() pic.data = image.read() pic.type = COVER_FRONT pic.mime = image.mime_type pic.width = image.width pic.height = image.height # pic.depth = image.color_depth self.audio.add_picture(pic) with translate_errors(): self.audio.save() # clear vcomment tags super().clear_images()
def test_clear_images(self): data = b"abc" song = FLAC(self.filename) pic = Picture() pic.data = data song.add_picture(pic) song.save() song = FLACFile(self.filename) self.assertTrue(song.get_primary_image()) song.clear_images() song.clear_images() song = FLACFile(self.filename) self.assertFalse(song.get_primary_image())
def test_clear_images(self): data = "abc" song = FLAC(self.filename) pic = Picture() pic.data = data song.add_picture(pic) song.save() song = FLACFile(self.filename) self.assertTrue(song.get_primary_image()) song.clear_images() song.clear_images() song = FLACFile(self.filename) self.assertFalse(song.get_primary_image())
def _set_tag(self, raw, tag, value): if tag == '__cover': raw.clear_pictures() for v in value: picture = Picture() picture.type = v.type picture.desc = v.desc picture.mime = v.mime picture.data = v.data raw.add_picture(picture) return # flac has text based attributes, so convert everything to unicode value = [xl.unicode.to_unicode(v) for v in value] CaseInsensitveBaseFormat._set_tag(self, raw, tag, value)
def set_image(self, image): """Replaces all embedded images by the passed image""" with image: pic = Picture() pic.data = image.read() pic.type = COVER_FRONT pic.mime = image.mime_type pic.width = image.width pic.height = image.height self.audio.pop("coverart", None) self.audio.pop("coverartmime", None) self.audio["metadata_block_picture"] = base64.b64encode( pic.write()).decode("ascii")
def add_flac_cover(filename, albumart): audio = File(filename) image = Picture() image.type = 3 if albumart.endswith('png'): mime = 'image/png' else: mime = 'image/jpeg' image.desc = 'front cover' with open(albumart, 'rb') as f: # better than open(albumart, 'rb').read() ? image.data = f.read() audio.add_picture(image) audio.save()
def _set_tag(self, raw, tag, value): if tag == '__cover': raw.clear_pictures() for v in value: picture = Picture() picture.type = v.type picture.desc = v.desc picture.mime = v.mime picture.data = v.data raw.add_picture(picture) return # flac has text based attributes, so convert everything to unicode value = [xl.unicode.to_unicode(v) for v in value] CaseInsensitveBaseFormat._set_tag(self, raw, tag, value)
def get_cover_picture(self, cover): """ Returns mutage Picture class for the cover image Usefull for OGG and FLAC format Picture type = cover image see http://flac.sourceforge.net/documentation_tools_flac.html#encoding_options """ f = file(cover) p = Picture() p.type = 3 p.data = f.read() p.mime = mimetypes.guess_type(cover)[0] f.close() return p
def __setattr__(self, attr, value): if attr in self.VALID_TAG_KEYS: if isinstance(value, Artwork): picture = Picture() picture.type = 3 picture.mime = value.mime picture.data = value.data self.audio.clear_pictures() self.audio.add_picture(picture) elif isinstance(value, str): self.audio[attr] = [value] else: raise TagValueError(value) else: object.__setattr__(self, attr, value)
def test_clear_tags_preserve_legacy_coverart(self): # FLAC does not use the cover art tags but has its separate image implementation pic = Picture() pic.data = load_coverart_file('mb.png') save_raw( self.filename, { 'coverart': PNG_BASE64, 'metadata_block_picture': base64.b64encode(pic.write()).decode('ascii') }) config.setting['clear_existing_tags'] = True config.setting['preserve_images'] = True metadata = save_and_load_metadata(self.filename, Metadata()) self.assertEqual(0, len(metadata.images))
def get_cover_picture(self, cover): """ Returns mutagen Picture class for the cover image Usefull for OGG and FLAC format Picture type = cover image see http://flac.sourceforge.net/documentation_tools_flac.html#encoding_options """ f = file(cover) p = Picture() p.type = 3 p.data = f.read() p.mime = mimetypes.guess_type(cover)[0] f.close() return p
def _set_tag(self, raw, tag, value): if tag == 'metadata_block_picture': new_value = [] for v in value: picture = Picture() picture.type = v.type picture.desc = v.desc picture.mime = v.mime picture.data = v.data new_value.append(base64.b64encode(picture.write())) value = new_value else: # vorbis has text based attributes, so convert everything to unicode value = [xl.unicode.to_unicode(v) for v in value] CaseInsensitveBaseFormat._set_tag(self, raw, tag, value)
def tag_put_picture(album_id): pic = Picture() album = get_album(album_id) with open(album['Path'] + "/folder.jpg", "rb") as f: pic.data = f.read() if not pic.data: return 'No folder.jpg found' pic.type = id3.PictureType.COVER_FRONT pic.mime = u"image/jpeg" pic.width = 500 pic.height = 500 pic.depth = 16 for p in tag_get_piece_paths(album_id): set_pic(p, pic) return 'success'
def write_tags(pre_path, meta, cov_path): del meta['track_padded'] audio = FLAC(pre_path) audio.delete() for k, v in meta.items(): if v: audio[k] = str(v) if cov_path: image = Picture() with open(cov_path, 'rb') as f: image.data = f.read() image.type = 3 image.mime = "image/jpeg" audio.add_picture(image) audio.save(pre_path)
def _set_tag(self, raw, tag, value): if tag == 'metadata_block_picture': new_value = [] for v in value: picture = Picture() picture.type = v.type picture.desc = v.desc picture.mime = v.mime picture.data = v.data new_value.append(base64.b64encode(picture.write())) value = new_value else: # vorbis has text based attributes, so convert everything to unicode value = [xl.unicode.to_unicode(v) for v in value] CaseInsensitveBaseFormat._set_tag(self, raw, tag, value)
def __setattr__(self, attr, value): if attr in self.VALID_TAG_KEYS: if isinstance(value, Artwork): picture = Picture() picture.type = 3 picture.mime = value.mime picture.data = value.data self.audio.clear_pictures() self.audio.add_picture(picture) elif isinstance(value, str): self.audio[attr] = [value] else: raise TagValueError(value) else: object.__setattr__(self, attr, value)
def AddImages(self): MadeUrls = [] MadeUrlsResults = [] QueriedImagesKeys = self.QueriedImages.keys() QueriedImagesKeys.sort() for flacPath in QueriedImagesKeys: index = -1 for Query in self.QueriedImages[flacPath]: try: index = MadeUrls.index(Query) except ValueError: #print ' filling --- cache %s:%s' % (Query,flacPath) #print MadeUrls try: #print "Query=%s,%s" % (Query,type(Query)) data = urllib2.urlopen(Query ) except urllib2.URLError: print "Could not open URL: %s for file : %s" % (Query,flacPath) continue MadeUrls.append(Query) MadeUrlsResults.append(data.read()) index = MadeUrls.index(Query) if index == -1: # we had succesfull urls for this file continue if len(MadeUrlsResults[index]) == 0: # we had no data continue #print ' have %s for %s' % (Query,flacPath) try: metadata = FLAC(flacPath) except FLACNoHeaderError as (strerror): print strerror continue if pict_test(metadata): print "Already has cover are:%s" % (flacPath) continue image = Picture() image.type = 3 image.desc = 'front cover' if Query.endswith('png'): mime = 'image/png' if Query.endswith('jpg'): mime = 'image/jpeg' image.data = MadeUrlsResults[index] metadata.add_picture(image) metadata.save()
def test_art_lyrics(tmpdir): """Test when everything is valid.""" flac_dir = tmpdir.mkdir('flac') flac = flac_dir.join('Artist2 - 2012 - Album - 01 - Title.flac').ensure(file=True) with open(os.path.join(os.path.dirname(__file__), '1khz_sine.flac'), 'rb') as f: flac.write(f.read(), 'wb') tags = FLAC(str(flac.realpath())) tags.update(dict(artist='Artist2', date='2012', album='Album', tracknumber='01', title='Title', unsyncedlyrics='L')) image = Picture() image.type, image.mime = 3, 'image/jpeg' with open(os.path.join(os.path.dirname(__file__), '1_album_art.jpg'), 'rb') as f: image.data = f.read() tags.add_picture(image) tags.save() a_messages = find_inconsistent_tags([str(flac.realpath())], False, False) assert {} == a_messages
def addInfoToFlac(filename, songInfo): filePath = os.path.join(songInfo["audioBasePath"], filename) audio = FLAC(filePath) pic = Picture() with open(songInfo["imgPath"], "rb") as f: pic.data = f.read() pic.type = PictureType.COVER_FRONT pic.mime = u"image/jpeg" audio.add_picture(pic) audio["title"] = songInfo["name"] audio["artist"] = songInfo["artist"] audio["album"] = songInfo["album"] audio.save()
def _image_flac(self): if self.image and cfg.embed_cover: audio = FLAC(self.path) img = Picture() img.type = 3 img.data = requests.get(self.image).content if cfg.overwrite_cover: audio.clear_pictures() audio.add_picture(img) self.cover_updated = True else: if self.cover_embedded is False: audio.clear_pictures() audio.add_picture(img) self.cover_updated = True audio.save()
def _add_ogg_image(audio: File, image: Image, image_data: bytes): picture = Picture() picture.data = image_data picture.type = 3 # Front cover picture.desc = u"Cover Art" picture.mime = _image_mimes[image.format] picture.width = image.width picture.height = image.height picture.depth = 24 picture_data = picture.write() encoded_data = base64.b64encode(picture_data) vcomment_value = encoded_data.decode("ascii") audio["metadata_block_picture"] = [vcomment_value] audio.save()
def __setattr__(self, attr, value): if attr in self.VALID_TAG_KEYS: if isinstance(value, Artwork): picture = Picture() picture.type = 3 picture.mime = value.mime picture.data = value.data self.audio['metadata_block_picture'] = [ b64encode(picture.write()).decode('utf-8') ] elif isinstance(value, str): self.audio[attr] = [value] else: raise TagValueError(value) else: object.__setattr__(self, attr, value)
def writeAlbumCov(file): if file.endswith('c'): audio = File(file) image = Picture() image.type = 3 mime = "image/jpeg" with open("cover.jpg", 'rb') as f: image.data = f.read() audio.clear_pictures() audio.add_picture(image) audio.save() else: audio = MP4(file) with open("cover.jpg", 'rb') as f: audio['covr'] = [MP4Cover(f.read(), imageformat = MP4Cover.FORMAT_JPEG)] audio.save(file)
def __setattr__(self, attr, value): if attr in self.VALID_TAG_KEYS: if isinstance(value, Artwork): picture = Picture() picture.type = 3 picture.mime = value.mime picture.data = value.data self.audio['metadata_block_picture'] = [ b64encode(picture.write()).decode('utf-8') ] elif isinstance(value, str): self.audio[attr] = [value] else: raise TagValueError(value) else: object.__setattr__(self, attr, value)
def test_invalid_overflow_recover_and_save_back(self): # save a picture which is too large for flac, but still write it # with a wrong block size f = FLAC(self.filename) f.clear_pictures() pic = Picture() pic.data = b"\x00" * (2 ** 24 - 32) pic._invalid_overflow_size = 42 f.add_picture(pic) f.save() # make sure we can read it and save it again f = FLAC(self.filename) self.assertTrue(f.pictures) self.assertEqual(len(f.pictures[0].data), 2 ** 24 - 32) f.save()
def test_invalid_overflow_recover_and_save_back(self): # save a picture which is too large for flac, but still write it # with a wrong block size f = FLAC(self.filename) f.clear_pictures() pic = Picture() pic.data = b"\x00" * (2**24 - 32) pic._invalid_overflow_size = 42 f.add_picture(pic) f.save() # make sure we can read it and save it again f = FLAC(self.filename) self.assertTrue(f.pictures) self.assertEqual(len(f.pictures[0].data), 2**24 - 32) f.save()
def download_flac(track): url = session.get_media_url(track_id=track.id) name_tag = translit(""+track.name, "ru", reversed=True).encode("UTF-8") name = translit(""+track.name, "ru", reversed=True).encode("UTF-8").replace("?", "").replace("(", "").replace(")", "").replace("<", "").replace(">", "").replace(":", "").replace("/", "").replace("|", "").replace('"', "").replace("*", "") artist_name = translit(u""+track.artist.name, "ru", reversed=True).encode("UTF-8") album_name = translit(u""+track.album.name, "ru", reversed=True).encode("UTF-8") releaseDate = str(track.album.release_date) print(name+' - '+url) album_artist = translit(u""+track.album.artist.name, "ru", reversed=True).encode("UTF-8") try: if ".m4a" not in url: download(url, "music/"+name+".flac") except Exception: flac = 0 try: if ".m4a" not in url: download(track.album.image, 'music/'+name+'.png') except Exception: flac = 0 if os.path.exists("music/"+name+".flac"): flac = 1 if not os.path.exists("music/"+name+".flac"): flac = 0 if ".flac" not in url: flac = 0 if ".flac" in url: flac = 1 if flac == 1: audio = FLAC("music/"+name+".flac") albumart = "music/"+name+".png" image = Picture() image.type = 3 mime = 'image/png' image.desc = 'front cover' with open(albumart, 'rb') as f: image.data = f.read() audio['artist'] = track.artist.name audio['title'] = track.name audio['album'] = track.album.name audio['date'] = releaseDate audio.add_picture(image) audio.save() os.remove("music/"+name+".png") elif flac == 0: print(name+" isn't in flac or can't be downloaded in this country.") else: print("Unknown error.")
def create_album_art(self, art_location: str): with open(art_location, 'rb') as image: image_data = image.read() image = Image.open(io.BytesIO(image_data)) picture = Picture() picture.data = image_data picture.type = 3 # COVER_FRONT picture.mime = file_type(art_location) picture.width, picture.height = image.size picture.depth = mode_to_depth(image.mode) picture.desc = "Front Cover" image.close() return picture
def add_albumart(working_dir, folder_name, title): audio = File(os.path.join(working_dir, folder_name, f'{title}.flac')) image = Picture() image.type = 3 mime = 'image/png' image.desc = 'front cover' # better than open(albumart, 'rb').read() ? with open( os.path.join(working_dir, folder_name, 'albumart', f'{title}.png'), 'rb') as f: image.data = f.read() audio.add_picture(image) audio.save()
def tag_flac(self, file_path, track_info, album_info, album_art_path=None): tagger = FLAC(file_path) self._meta_tag(tagger, track_info, album_info) if self.fmtopts['embed_album_art'] and album_art_path is not None: pic = Picture() with open(album_art_path, 'rb') as f: pic.data = f.read() pic.type = PictureType.COVER_FRONT pic.mime = u"image/jpeg" # TODO: detect this automatically? pic.width = 1280 pic.height = 1280 pic.depth = 24 tagger.add_picture(pic) tagger.save(file_path)
def _embed_cover(audio_file, song_obj, encoding): image = Picture() image.type = 3 image.desc = "Cover" image.mime = "image/jpeg" image.data = urlopen(song_obj.get_album_cover_url()).read() if encoding == "flac": audio_file.add_picture(image) elif encoding == "ogg" or encoding == "opus": # From the Mutagen docs (https://mutagen.readthedocs.io/en/latest/user/vcomment.html) image_data = image.write() encoded_data = base64.b64encode(image_data) vcomment_value = encoded_data.decode("ascii") audio_file["metadata_block_picture"] = [vcomment_value] return audio_file
def to_flac(self, flac_meta: FLAC, builtin_cuesheet=False) -> None: ''' :param builtin_cuesheet: Whether to use the cuesheet field builtin FLAC specs ''' def add_if_exist(obj, tag_key): if obj: flac_meta.tags[tag_key] = obj add_if_exist(self.title, 'ALBUM') add_if_exist(self.genre, 'GENRE') add_if_exist(self.date, 'DATE') if self.artists: flac_meta.tags['ALBUM ARTIST'] = list(self.artists) if self.discnumber is not None: flac_meta.tags['DISCNUMBER'] = str(self.discnumber) if self.cover: image = Image.open(BytesIO(self.cover)) pic = Picture() pic.type = PictureType.COVER_FRONT pic.data = self.cover pic.mime = Image.MIME[image.format] pic.width = image.width pic.height = image.height flac_meta.add_picture(pic) if self.cuesheet: if builtin_cuesheet: flac_meta.cuesheet = self.cuesheet.to_flac( flac_meta.info.sample_rate) flac_meta.cuesheet.tracks.append( CueSheetTrack(170, flac_meta.info.total_samples)) # lead-out # save track-wise tags in foobar2000 style tracks = next(iter(self.cuesheet.files.values())) for i, track in tracks.items(): add_if_exist(track.title, f'CUE_TRACK{i:02}_TITLE') add_if_exist(track.performer, f'CUE_TRACK{i:02}_PERFORMER') else: flac_meta.tags['CUESHEET'] = str(self.cuesheet) add_if_exist(self.cuesheet.catalog, 'Catalog') add_if_exist(self.cuesheet.rems.get('COMMENT', None), 'Comment') flac_meta.save()
def get_image_size(fname): '''Determine the image type of fhandle and return its size. from draco''' with open(fname, 'rb') as fhandle: head = fhandle.read(24) if len(head) != 24: return if imghdr.what(fname) == 'png': check = struct.unpack('>i', head[4:8])[0] if check != 0x0d0a1a0a: return width, height = struct.unpack('>ii', head[16:24]) elif imghdr.what(fname) == 'gif': width, height = struct.unpack('<HH', head[6:10]) elif imghdr.what(fname) == 'jpeg': try: fhandle.seek(0) # Read 0xff next size = 2 ftype = 0 while not 0xc0 <= ftype <= 0xcf: fhandle.seek(size, 1) byte = fhandle.read(1) while ord(byte) == 0xff: byte = fhandle.read(1) ftype = ord(byte) size = struct.unpack('>H', fhandle.read(2))[0] - 2 # We are at a SOFn block fhandle.seek(1, 1) # Skip `precision' byte. height, width = struct.unpack('>HH', fhandle.read(4)) except Exception: #IGNORE:W0703 printError() return else: return fhandle.seek(0) imgdata = fhandle.read() img = Picture() img.data = imgdata img.width = width img.height = height return img
def ogg_coverart(config): print("Adding " + config.coverart_mime + " cover art to " + config.ogg_file) coverart = config.coverart imgdata = open(coverart,'rb').read() im = Image.open(coverart) w,h = im.size p = Picture() p.data = imgdata p.type = 3 p.desc = 'Cover' p.mime = config.coverart_mime p.width = w p.height = h p.depth = 24 dt=p.write() enc=base64.b64encode(dt).decode('ascii') audio = OggVorbis(config.ogg_file) audio['metadata_block_picture']=[enc] audio.save()
def set_image(self, image): """Replaces all embedded images by the passed image""" try: audio = self.MutagenType(self["~filename"]) data = image.file.read() except EnvironmentError: return pic = Picture() pic.data = data pic.type = APICType.COVER_FRONT pic.mime = image.mime_type pic.width = image.width pic.height = image.height pic.depth = image.color_depth audio.pop("coverart", None) audio.pop("coverartmime", None) audio["metadata_block_picture"] = base64.b64encode(pic.write()) audio.save() self.has_images = True
def test_get_metadata(tmpdir): flac_dir = tmpdir.mkdir('flac') mp3_dir = '/tmp/mp3_dir' flac = flac_dir.join('Artist - 2001 - Album (Disc 1) - 01 - Title.flac').ensure(file=True) with open(os.path.join(os.path.dirname(__file__), '1khz_sine.flac'), 'rb') as f: flac.write(f.read(), 'wb') tags = FLAC(str(flac)) tags.update(dict(artist='Artist', date='2001', album='Album', discnumber='1', tracknumber='01', title='Title', unsyncedlyrics='L')) image = Picture() image.type, image.mime = 3, 'image/jpeg' with open(os.path.join(os.path.dirname(__file__), '1_album_art.jpg'), 'rb') as f: image.data = f.read() tags.add_picture(image) tags.save() song = Song(str(flac), str(flac_dir), mp3_dir) song.get_metadata() assert dict() == OPTIONS assert ['Artist', '2001', 'Album'] == [song.flac_artist, song.flac_date, song.flac_album] assert ['01', 'Title'] == [song.flac_track, song.flac_title] assert [True, True] == [song.flac_has_lyrics, song.flac_has_picture] assert song.metadata_ok