1
0
mirror of https://github.com/krateng/maloja.git synced 2023-08-10 21:12:55 +03:00

Fixed merging of artists that already share tracks

This commit is contained in:
krateng 2022-04-22 22:59:02 +02:00
parent 7c9f6e9e2d
commit 1dfda0086e

View File

@ -446,15 +446,47 @@ def merge_tracks(target_id,source_ids,dbconn=None):
@connection_provider @connection_provider
def merge_artists(target_id,source_ids,dbconn=None): def merge_artists(target_id,source_ids,dbconn=None):
# todo: songs that have both artists! # some tracks could already have multiple of the to be merged artists
op = DB['trackartists'].update().where( # find literally all tracksartist entries that have any of the artists involved
DB['trackartists'].c.artist_id.in_(source_ids) op = DB['trackartists'].select().where(
).values( DB['trackartists'].c.artist_id.in_(source_ids + [target_id])
artist_id=target_id
) )
result = dbconn.execute(op) result = dbconn.execute(op)
track_ids = set(row.track_id for row in result)
# now just delete them all lmao
op = DB['trackartists'].delete().where(
#DB['trackartists'].c.track_id.in_(track_ids),
DB['trackartists'].c.artist_id.in_(source_ids + [target_id]),
)
result = dbconn.execute(op)
# now add back the real new artist
op = DB['trackartists'].insert().values([
{'track_id':track_id,'artist_id':target_id}
for track_id in track_ids
])
result = dbconn.execute(op)
# tracks_artists = {}
# for row in result:
# tracks_artists.setdefault(row.track_id,[]).append(row.artist_id)
#
# multiple = {k:v for k,v in tracks_artists.items() if len(v) > 1}
#
# print([(get_track(k),[get_artist(a) for a in v]) for k,v in multiple.items()])
#
# op = DB['trackartists'].update().where(
# DB['trackartists'].c.artist_id.in_(source_ids)
# ).values(
# artist_id=target_id
# )
# result = dbconn.execute(op)
# this could have created duplicate tracks # this could have created duplicate tracks
merge_duplicate_tracks(artist_id=target_id) merge_duplicate_tracks(artist_id=target_id)
clean_db() clean_db()
@ -901,7 +933,6 @@ def merge_duplicate_tracks(artist_id):
for track in track_identifiers: for track in track_identifiers:
if len(track_identifiers[track]) > 1: if len(track_identifiers[track]) > 1:
target,*src = track_identifiers[track] target,*src = track_identifiers[track]
log("Merging",src,"into",target)
merge_tracks(target,src) merge_tracks(target,src)