Improve adding tracks to end of playlist

This commit is contained in:
Keith Edmunds 2021-04-10 11:24:41 +01:00
parent f550c8944f
commit b3e3e503ca

View File

@ -168,7 +168,15 @@ class Playlists(Base):
return session.query(Playlists).filter(Playlists.id == plid).one()
def add_track(self, track, row):
def add_track(self, track, row=None):
"""
Add track to playlist at given row.
If row=None, add to end of playlist
"""
if not row:
row = PlaylistTracks.new_row(self.id)
glue = PlaylistTracks(row=row)
glue.track_id = track.id
self.tracks.append(glue)
@ -220,14 +228,6 @@ class PlaylistTracks(Base):
plt.row = new_row
session.commit()
@staticmethod
def remove_track(playlist_id, track_id):
DEBUG(f"remove_track({playlist_id}, {track_id})")
session.query(PlaylistTracks).filter(
PlaylistTracks.playlist_id == playlist_id,
PlaylistTracks.track_id == track_id
).delete()
class Settings(Base):
__tablename__ = 'settings'