yetangitu/funkwhale
1
0
Fork 0
mirror of https://code.eliotberriot.com/funkwhale/funkwhale.git synced 2025-10-04 05:39:16 +02:00
Code Issues Releases Wiki Activity

Update api/funkwhale_api/playlists/utils.py

Browse source
This commit is contained in:
petitminion 2021-06-09 13:48:59 +00:00
parent 706815115a
commit 53e4a2464c
1 changed files with 9 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

30
api/funkwhale_api/playlists/utils.py
View file

@ -18,7 +18,7 @@ logger = logging.getLogger(__name__)
def clean_namespace_xspf(xspf_file):
"""
This will delete any namaespace found in the xspf file. It will also delete any encoding info.
This way xspf file will be compatible with our get_playlist_metadata_from_xspf function.
This way xspf file will be compatible with our get_track_id_from_xspf function.
"""
file = open(xspf_file)
with file as f:
@ -44,36 +44,24 @@ def get_track_id_from_xspf(xspf_file):
xspf_file_clean = clean_namespace_xspf(xspf_file)
tree = etree.parse(xspf_file_clean)
tracks = tree.findall(".//track")
total_track_count = 0
added_file_count = 0
added_track_count = 0
for track in tracks:
track_id = ""
total_track_count = total_track_count + 1
total = str(total_track_count)
# Getting metadata of the xspf file
try:
artist = track.find(".//creator").text
except Exception as e:
logger.info("Error while parsing Xml file :%s" % e)
try:
title = track.find(".//title").text
except Exception as e:
logger.info("Error while parsing Xml file :%s" % e)
try:
album = track.find(".//album").text
except Exception as e:
logger.info("Error while parsing Xml file :%s" % e)
logger.info(f"Error while parsing Xml file : {e!r}")
# Finding track id in the db
try:
artist_id = Artist.objects.get(name=artist)
except Exception as e:
logger.info("Error while quering database : %s" % e)
try:
album_id = Album.objects.get(title=album)
except Exception as e:
logger.info("Error while quering database :%s" % e)
logger.info(f"Error while quering database : {e!r}")
try:
track_id = Track.objects.get(
title=title, artist=artist_id.id, album=album_id.id
@ -83,16 +71,16 @@ def get_track_id_from_xspf(xspf_file):
try:
track_id = Track.objects.get(title=title, artist=artist_id.id)
except Exception as e:
logger.info("Error while quering database :%s" % e)
logger.info(f"Error while quering database : {e!r}")
if track_id:
track_list.append(track_id.id)
added_file_count = added_file_count + 1
added_track_count = added_track_count + 1
logger.info(
str(total)
str(len(tracks))
+ " tracks where found in xspf file. "
+ str(added_file_count)
+ "are gonna be added to playlist."
+ str(added_track_count)
+ " are gonna be added to playlist."
)
return track_list