From 52a7a98970fa6b4a8af761a2cff99d4e496cc133 Mon Sep 17 00:00:00 2001 From: Aevann Date: Fri, 28 Jul 2023 03:27:01 +0300 Subject: [PATCH] fix profile songs --- files/routes/settings.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/files/routes/settings.py b/files/routes/settings.py index 835262108..fdffeea9a 100644 --- a/files/routes/settings.py +++ b/files/routes/settings.py @@ -810,13 +810,6 @@ def settings_song_change_mp3(v): def _change_song_youtube(vid, id): - db = db_session() - - v = db.query(User).filter_by(id=vid).options(load_only(User.song)).one() - - if v.song and path.isfile(f"/songs/{v.song}.mp3") and db.query(User).filter_by(song=v.song).count() == 1: - os.remove(f"/songs/{v.song}.mp3") - ydl_opts = { 'cookiefile': '/cookies', 'outtmpl': '/temp_songs/%(id)s.%(ext)s', @@ -838,6 +831,13 @@ def _change_song_youtube(vid, id): os.rename(f"/temp_songs/{id}.mp3", f"/songs/{id}.mp3") + db = db_session() + + v = db.query(User).filter_by(id=vid).options(load_only(User.song)).one() + + if v.song and path.isfile(f"/songs/{v.song}.mp3") and db.query(User).filter_by(song=v.song).count() == 1: + os.remove(f"/songs/{v.song}.mp3") + v.song = id db.add(v) db.commit()