swingmusic/server/app/api/playlist.py
geoffrey45 fe0d7cae1a [server] fix playlist methods
- deepcopy playlists in get_all_playlists
2022-03-27 18:13:51 +03:00

64 lines
1.6 KiB
Python

"""
Contains all the playlist routes.
"""
from copy import deepcopy
from typing import List
from flask import Blueprint, request
from app import instances, api
from app.lib import playlistlib
from app import models
from app import exceptions
playlist_bp = Blueprint("playlist", __name__, url_prefix="/")
PlaylistExists = exceptions.PlaylistExists
TrackExistsInPlaylist = exceptions.TrackExistsInPlaylist
@playlist_bp.route("/playlists", methods=["GET"])
def get_all_playlists():
ppp = deepcopy(api.PLAYLISTS)
playlists = []
for pl in ppp:
pl.tracks = []
playlists.append(pl)
return {"data": playlists}
@playlist_bp.route("/playlist/new", methods=["POST"])
def create_playlist():
data = request.get_json()
playlist = {"name": data["name"], "description": [], "tracks": []}
try:
p_in_db = instances.playlist_instance.get_playlist_by_name(playlist["name"])
if p_in_db:
raise PlaylistExists("Playlist already exists.")
except PlaylistExists as e:
return {"error": str(e)}, 409
upsert_id = instances.playlist_instance.insert_playlist(playlist)
p = instances.playlist_instance.get_playlist_by_id(upsert_id)
pp = models.Playlist(p)
api.PLAYLISTS.append(pp)
return {"playlist": pp}, 201
@playlist_bp.route("/playlist/<playlist_id>/add", methods=["POST"])
def add_track_to_playlist(playlist_id):
data = request.get_json()
trackid = data["track"]
try:
playlistlib.add_track(playlist_id, trackid)
except TrackExistsInPlaylist as e:
return {"error": str(e)}, 409
return {"msg": "I think It's done"}, 200