def example_2(self):
file = File("track05.mp3")
file.state = File.NORMAL
file.metadata['album'] = 'Explosive Doowops, Volume 4'
file.metadata['title'] = 'Why? Oh Why?'
file.metadata['artist'] = 'The Fantasys'
file.metadata['artistsort'] = 'Fantasys, The'
file.metadata['albumartist'] = self.config.setting['va_name']
file.metadata['albumartistsort'] = self.config.setting['va_name']
file.metadata['tracknumber'] = '5'
file.metadata['totaltracks'] = '26'
file.metadata['discnumber'] = '2'
file.metadata['totaldiscs'] = '2'
file.metadata['date'] = '1999-02-03'
file.metadata['releasetype'] = 'compilation'
file.metadata['releasestatus'] = 'official'
file.metadata['releasecountry'] = 'US'
file.metadata['compilation'] = '1'
file.metadata['~extension'] = 'mp3'
file.metadata[
'musicbrainz_albumid'] = 'bcc97e8a-2055-400b-a6ed-83288285c6fc'
file.metadata[
'musicbrainz_albumartistid'] = '89ad4ac3-39f7-470e-963a-56509c546377'
file.metadata[
'musicbrainz_artistid'] = '06704773-aafe-4aca-8833-b449e0a6467f'
file.metadata[
'musicbrainz_trackid'] = 'd92837ee-b1e4-4649-935f-e433c3e5e429'
return file
def example_1(self):
file = File("ticket_to_ride.mp3")
file.state = File.NORMAL
file.metadata['album'] = 'Help!'
file.metadata['title'] = 'Ticket to Ride'
file.metadata['artist'] = 'The Beatles'
file.metadata['artistsort'] = 'Beatles, The'
file.metadata['albumartist'] = 'The Beatles'
file.metadata['albumartistsort'] = 'Beatles, The'
file.metadata['tracknumber'] = '7'
file.metadata['totaltracks'] = '14'
file.metadata['discnumber'] = '1'
file.metadata['totaldiscs'] = '1'
file.metadata['date'] = '1965-08-06'
file.metadata['releasetype'] = 'album'
file.metadata['releasestatus'] = 'official'
file.metadata['releasecountry'] = 'US'
file.metadata['~extension'] = 'mp3'
file.metadata[
'musicbrainz_albumid'] = '2c053984-4645-4699-9474-d2c35c227028'
file.metadata[
'musicbrainz_albumartistid'] = 'b10bbbfc-cf9e-42e0-be17-e2c3e1d2600d'
file.metadata[
'musicbrainz_artistid'] = 'b10bbbfc-cf9e-42e0-be17-e2c3e1d2600d'
file.metadata[
'musicbrainz_trackid'] = '898a2916-f64d-48d3-ab1a-3446fb450448'
return file
def setUp(self):
super().setUp()
self.tagger.acoustidmanager = MagicMock()
self.file = File('somepath/somefile.mp3')
self.set_config_values({
'save_acoustid_fingerprints': True,
})
def example_2(self):
file = File("track05.mp3")
file.state = File.NORMAL
# The data for this example does not match the release on MusicBrainz,
# but still works well enough as an example.
file.metadata['album'] = 'Explosive Doowops, Volume 4'
file.metadata['title'] = 'Why? Oh Why?'
file.metadata['artist'] = 'The Fantasys'
file.metadata['artistsort'] = 'Fantasys, The'
file.metadata['albumartist'] = config.setting['va_name']
file.metadata['albumartistsort'] = config.setting['va_name']
file.metadata['tracknumber'] = '5'
file.metadata['totaltracks'] = '26'
file.metadata['discnumber'] = '2'
file.metadata['totaldiscs'] = '2'
file.metadata['date'] = '1999-02-03'
file.metadata['releasetype'] = ['album', 'compilation']
file.metadata['~primaryreleasetype'] = ['album']
file.metadata['~secondaryreleasetype'] = ['compilation']
file.metadata['releasestatus'] = 'official'
file.metadata['releasecountry'] = 'US'
file.metadata['compilation'] = '1'
file.metadata['~extension'] = 'mp3'
file.metadata[
'musicbrainz_albumid'] = 'bcc97e8a-2055-400b-a6ed-83288285c6fc'
file.metadata[
'musicbrainz_albumartistid'] = '89ad4ac3-39f7-470e-963a-56509c546377'
file.metadata[
'musicbrainz_artistid'] = '06704773-aafe-4aca-8833-b449e0a6467f'
file.metadata[
'musicbrainz_recordingid'] = 'd92837ee-b1e4-4649-935f-e433c3e5e429'
file.metadata[
'musicbrainz_releasetrackid'] = 'eac99807-93d4-3668-9714-fa0c1b487ccf'
return file
def example_2(self):
file = File("track05.mp3")
file.state = File.NORMAL
file.metadata['album'] = "Coup d'État, Volume 1: Ku De Ta / Prologue"
file.metadata['title'] = "I've Got to Learn the Mambo"
file.metadata['artist'] = "Snowboy feat. James Hunter"
file.metadata['artistsort'] = "Snowboy feat. Hunter, James"
file.metadata['albumartist'] = config.setting['va_name']
file.metadata['albumartistsort'] = config.setting['va_name']
file.metadata['tracknumber'] = '5'
file.metadata['totaltracks'] = '13'
file.metadata['discnumber'] = '2'
file.metadata['totaldiscs'] = '2'
file.metadata['discsubtitle'] = "Beat Up"
file.metadata['date'] = '2005-07-04'
file.metadata['releasetype'] = ['album', 'compilation']
file.metadata['~primaryreleasetype'] = 'album'
file.metadata['~secondaryreleasetype'] = 'compilation'
file.metadata['releasestatus'] = 'official'
file.metadata['releasecountry'] = 'AU'
file.metadata['compilation'] = '1'
file.metadata['~multiartist'] = '1'
file.metadata['~extension'] = 'mp3'
file.metadata['musicbrainz_albumid'] = '4b50c71e-0a07-46ac-82e4-cb85dc0c9bdd'
file.metadata['musicbrainz_recordingid'] = 'b3c487cb-0e55-477d-8df3-01ec6590f099'
file.metadata['musicbrainz_releasetrackid'] = 'f8649a05-da39-39ba-957c-7abf8f9012be'
file.metadata['musicbrainz_albumartistid'] = '89ad4ac3-39f7-470e-963a-56509c546377'
file.metadata['musicbrainz_artistid'] = ['7b593455-d207-482c-8c6f-19ce22c94679',
'9e082466-2390-40d1-891e-4803531f43fd']
return file
def callback(self, objs):
if len(objs) != 1 or not isinstance(objs[0], Cluster):
return
cluster = objs[0]
# Temporary list for sorting.
files = []
# Create a placeholder file to keep the cluster from being deleted
# when all the original files are removed.
tmpFile = File("tmpFile")
# Make a copy of all the file data in the cluster
for file in cluster.files:
files.append(file)
# Add the paceholder.
cluster.add_file(tmpFile)
# Delete all the files from the cluster. Use the temp list as the
# file reference to avoid iterator related problems.
for file in files:
cluster.remove_file(file)
# Sort the list.
files.sort(
key=lambda file: (file.metadata["artist"], file.metadata["title"]))
# Copy all the data back to the cluster
for file in files:
cluster.add_file(file)
# Remove the placeholder
cluster.remove_file(tmpFile)
def example_1(self):
file = File("ticket_to_ride.mp3")
file.state = File.NORMAL
file.metadata['album'] = 'Help!'
file.metadata['title'] = 'Ticket to Ride'
file.metadata['artist'] = 'The Beatles'
file.metadata['artistsort'] = 'Beatles, The'
file.metadata['albumartist'] = 'The Beatles'
file.metadata['albumartistsort'] = 'Beatles, The'
file.metadata['tracknumber'] = '7'
file.metadata['totaltracks'] = '14'
file.metadata['discnumber'] = '1'
file.metadata['totaldiscs'] = '1'
file.metadata['date'] = '1965-08-06'
file.metadata['releasetype'] = ['album', 'soundtrack']
file.metadata['~primaryreleasetype'] = ['album']
file.metadata['~secondaryreleasetype'] = ['soundtrack']
file.metadata['releasestatus'] = 'official'
file.metadata['releasecountry'] = 'US'
file.metadata['~extension'] = 'mp3'
file.metadata['musicbrainz_albumid'] = '2c053984-4645-4699-9474-d2c35c227028'
file.metadata['musicbrainz_albumartistid'] = 'b10bbbfc-cf9e-42e0-be17-e2c3e1d2600d'
file.metadata['musicbrainz_artistid'] = 'b10bbbfc-cf9e-42e0-be17-e2c3e1d2600d'
file.metadata['musicbrainz_recordingid'] = 'ed052ae1-c950-47f2-8d2b-46e1b58ab76c'
file.metadata['musicbrainz_releasetrackid'] = '7668a62a-2fac-3151-a744-5707ac8c883c'
return file
def example_1(self):
file = File("ticket_to_ride.mp3")
file.state = File.NORMAL
file.metadata['album'] = 'Help!'
file.metadata['title'] = 'Ticket to Ride'
file.metadata['~releasecomment'] = '2014 mono remaster'
file.metadata['artist'] = 'The Beatles'
file.metadata['artistsort'] = 'Beatles, The'
file.metadata['albumartist'] = 'The Beatles'
file.metadata['albumartistsort'] = 'Beatles, The'
file.metadata['tracknumber'] = '7'
file.metadata['totaltracks'] = '14'
file.metadata['discnumber'] = '1'
file.metadata['totaldiscs'] = '1'
file.metadata['originaldate'] = '1965-08-06'
file.metadata['originalyear'] = '1965'
file.metadata['date'] = '2014-09-08'
file.metadata['releasetype'] = ['album', 'soundtrack']
file.metadata['~primaryreleasetype'] = ['album']
file.metadata['~secondaryreleasetype'] = ['soundtrack']
file.metadata['releasestatus'] = 'official'
file.metadata['releasecountry'] = 'US'
file.metadata['~extension'] = 'mp3'
file.metadata[
'musicbrainz_albumid'] = 'd7fbbb0a-1348-40ad-8eef-cd438d4cd203'
file.metadata[
'musicbrainz_albumartistid'] = 'b10bbbfc-cf9e-42e0-be17-e2c3e1d2600d'
file.metadata[
'musicbrainz_artistid'] = 'b10bbbfc-cf9e-42e0-be17-e2c3e1d2600d'
file.metadata[
'musicbrainz_recordingid'] = 'ed052ae1-c950-47f2-8d2b-46e1b58ab76c'
file.metadata[
'musicbrainz_releasetrackid'] = '392639f5-5629-477e-b04b-93bffa703405'
return file
def test_remove_from_empty_track(self):
track = Track('00000000-0000-0000-0000-000000000000')
track.linked_files.append(File('test1.flac'))
update_metadata_images(track)
remove_metadata_images(track, [track.linked_files[0]])
self.assertEqual(set(), set(track.orig_metadata.images))
self.assertTrue(track.orig_metadata.has_common_images)
def test_remove_from_empty_cluster(self):
cluster = Cluster('Test')
cluster.files.append(File('test1.flac'))
update_metadata_images(cluster)
remove_metadata_images(cluster, [cluster.files[0]])
self.assertEqual(set(), set(cluster.metadata.images))
self.assertTrue(cluster.metadata.has_common_images)
def create_test_files():
test_images = [
CoverArtImage(url='file://file1', data=create_fake_png(b'a')),
CoverArtImage(url='file://file2', data=create_fake_png(b'b')),
]
test_files = [
File('test1.flac'),
File('test2.flac'),
File('test2.flac')
]
test_files[0].metadata.images.append(test_images[0])
test_files[1].metadata.images.append(test_images[1])
test_files[2].metadata.images.append(test_images[1])
test_files[0].orig_metadata.images.append(test_images[0])
test_files[1].orig_metadata.images.append(test_images[1])
test_files[2].orig_metadata.images.append(test_images[1])
return (test_images, test_files)
def test_multi(self):
files = [
File('foo1'),
File('foo2'),
File('foo3'),
File('foo4'),
File('foo5'),
]
fc = FileCluster()
fc.add('album 1', 'artist1', files[0])
fc.add('Album 1', 'artist 2', files[1])
fc.add('album\t1', 'Artist 1', files[2])
fc.add('Album 1', 'Artist 2', files[3])
fc.add('album 2', 'Artist 1', files[4])
self.assertEqual('Album 1', fc.title)
self.assertEqual('Artist 1', fc.artist)
self.assertEqual(files, list(fc.files))
def test_guess_title_tracknumber(self):
f = File('/somepath/01 somefile.mp3')
metadata = Metadata({
'album': 'somealbum',
})
f._guess_tracknumber_and_title(metadata)
self.assertEqual(metadata['tracknumber'], '1')
self.assertEqual(metadata['title'], 'somefile')
def test_remove_from_empty_album(self):
album = Album('00000000-0000-0000-0000-000000000000')
album.unmatched_files.files.append(File('test1.flac'))
update_metadata_images(album)
remove_metadata_images(album, [album.unmatched_files.files[0]])
self.assertEqual(set(), set(album.metadata.images))
self.assertEqual(set(), set(album.orig_metadata.images))
self.assertTrue(album.metadata.has_common_images)
self.assertTrue(album.orig_metadata.has_common_images)
def test_add_and_remove(self):
file = File('foo.flac')
file.acoustid_fingerprint = 'foo'
file.acoustid_length = 120
self.acoustidmanager.add(file, '00000000-0000-0000-0000-000000000001')
self.tagger.window.enable_submit.assert_called_with(False)
self.acoustidmanager.update(file,
'00000000-0000-0000-0000-000000000002')
self.tagger.window.enable_submit.assert_called_with(True)
self.acoustidmanager.remove(file)
self.tagger.window.enable_submit.assert_called_with(False)
def test_is_submitted(self):
file = File('foo.flac')
file.acoustid_fingerprint = 'foo'
file.acoustid_length = 120
self.assertTrue(self.acoustidmanager.is_submitted(file))
self.acoustidmanager.add(file, '00000000-0000-0000-0000-000000000001')
self.assertTrue(self.acoustidmanager.is_submitted(file))
self.acoustidmanager.update(file,
'00000000-0000-0000-0000-000000000002')
self.assertFalse(self.acoustidmanager.is_submitted(file))
self.acoustidmanager.update(file, '')
self.assertTrue(self.acoustidmanager.is_submitted(file))
def _add_unsubmitted_files(self, count):
files = []
for i in range(0, count):
file = File('foo%d.flac' % i)
files.append(file)
file.acoustid_fingerprint = 'foo'
file.acoustid_length = 120
self.acoustidmanager.add(file, None)
self.acoustidmanager.update(file,
'00000000-0000-0000-0000-%012d' % i)
self.assertFalse(self.acoustidmanager.is_submitted(file))
return files
def setUp(self):
super().setUp()
self.file = File('/somepath/somefile.mp3')
self.INVALIDSIMVAL = 666
self.file.similarity = self.INVALIDSIMVAL # to check if changed or not
self.file.supports_tag = lambda x: False if x.startswith('unsupported'
) else True
self.set_config_values({
'clear_existing_tags': False,
'compare_ignore_tags': [],
'enabled_plugins': [],
})
def setUp(self):
super().setUp()
self.file = File('/somepath/somefile.mp3')
self.set_config_values({
'ascii_filenames': False,
'clear_existing_tags': False,
'enabled_plugins': [],
'file_naming_format': '%album%/%title%',
'move_files_to': '/media/music',
'move_files': False,
'rename_files': False,
'windows_compatibility': True,
})
self.metadata = Metadata({
'album': 'somealbum',
'title': 'sometitle',
})
def setUp(self):
super().setUp()
AcousticBrainzFeatureExtractionTest.singleton = self
# Setup mock extractor
settings['acousticbrainz_extractor'] = mock_extractor
self.set_config_values(settings)
self.tagger.ab_extractor = ABExtractor()
self.assertTrue(self.tagger.ab_extractor.available())
# Load an irrelevant test file
self.file = File("./test/data/test.mp3")
# Load the AB features sample and copy the recording ID
with open(self.ab_features_file, "r", encoding="utf-8") as f:
ab_features = json.load(f)
# Copy the MB recordingID to the file, as we only work with already matched files
self.file.metadata['musicbrainz_recordingid'] = ab_features['metadata']['tags']['musicbrainz_trackid']
self.tagger.webservice = MagicMock()
def default_examples():
"""Generator for default example files.
Yields:
File: the next example File object
"""
# example 1
efile = File("ticket_to_ride.mp3")
efile.state = File.NORMAL
efile.metadata.update({
'album':
'Help!',
'title':
'Ticket to Ride',
'~releasecomment':
'2014 mono remaster',
'artist':
'The Beatles',
'artistsort':
'Beatles, The',
'albumartist':
'The Beatles',
'albumartistsort':
'Beatles, The',
'tracknumber':
'7',
'totaltracks':
'14',
'discnumber':
'1',
'totaldiscs':
'1',
'originaldate':
'1965-08-06',
'originalyear':
'1965',
'date':
'2014-09-08',
'releasetype': ['album', 'soundtrack'],
'~primaryreleasetype': ['album'],
'~secondaryreleasetype': ['soundtrack'],
'releasestatus':
'official',
'releasecountry':
'US',
'~extension':
'mp3',
'musicbrainz_albumid':
'd7fbbb0a-1348-40ad-8eef-cd438d4cd203',
'musicbrainz_albumartistid':
'b10bbbfc-cf9e-42e0-be17-e2c3e1d2600d',
'musicbrainz_artistid':
'b10bbbfc-cf9e-42e0-be17-e2c3e1d2600d',
'musicbrainz_recordingid':
'ed052ae1-c950-47f2-8d2b-46e1b58ab76c',
'musicbrainz_releasetrackid':
'392639f5-5629-477e-b04b-93bffa703405',
})
yield efile
# example 2
config = get_config()
efile = File("track05.mp3")
efile.state = File.NORMAL
efile.metadata.update({
'album':
"Coup d'État, Volume 1: Ku De Ta / Prologue",
'title':
"I've Got to Learn the Mambo",
'artist':
"Snowboy feat. James Hunter",
'artistsort':
"Snowboy feat. Hunter, James",
'albumartist':
config.setting['va_name'],
'albumartistsort':
config.setting['va_name'],
'tracknumber':
'5',
'totaltracks':
'13',
'discnumber':
'2',
'totaldiscs':
'2',
'discsubtitle':
"Beat Up",
'originaldate':
'2005-07-04',
'originalyear':
'2005',
'date':
'2005-07-04',
'releasetype': ['album', 'compilation'],
'~primaryreleasetype':
'album',
'~secondaryreleasetype':
'compilation',
'releasestatus':
'official',
'releasecountry':
'AU',
'compilation':
'1',
'~multiartist':
'1',
'~extension':
'mp3',
'musicbrainz_albumid':
'4b50c71e-0a07-46ac-82e4-cb85dc0c9bdd',
'musicbrainz_recordingid':
'b3c487cb-0e55-477d-8df3-01ec6590f099',
'musicbrainz_releasetrackid':
'f8649a05-da39-39ba-957c-7abf8f9012be',
'musicbrainz_albumartistid':
'89ad4ac3-39f7-470e-963a-56509c546377',
'musicbrainz_artistid': [
'7b593455-d207-482c-8c6f-19ce22c94679',
'9e082466-2390-40d1-891e-4803531f43fd'
],
})
yield efile
def _create_file(self, album, artist, filename="foo.mp3"):
file = File(filename)
file.metadata['album'] = album
file.metadata['artist'] = artist
return file
def test_file_metadata(self):
metadata = Metadata()
file = File('somepath/somefile.mp3')
self.assertEqual('', script_to_filename('$has_file()', metadata))
self.assertEqual(
'1', script_to_filename('$has_file()', metadata, file=file))
def test_add_invalid(self):
file = File('foo.flac')
self.acoustidmanager.add(file, '00000000-0000-0000-0000-000000000001')
self.tagger.window.enable_submit.assert_not_called()
def setUp(self):
super().setUp()
self.file = File('somepath/somefile.mp3')
def dummy_file(i):
file = File('foo%d.flac' % i)
file.acoustid_fingerprint = 'Z' * FINGERPRINT_SIZE
file.acoustid_length = 120
return file
def setUp(self):
super().setUp()
self.tmp_directory = self.mktmpdir()
filepath = os.path.join(self.tmp_directory, 'a.mp3')
self.file = File(filepath)
def setUp(self):
super().setUp()
self.tagger.acoustidmanager = MagicMock()
self.file = File('somepath/somefile.mp3')
