supysonic/api/playlists.py

# coding: utf-8

# This file is part of Supysonic.
#
# Supysonic is a Python implementation of the Subsonic server API.
# Copyright (C) 2013  Alban 'spl0k' Féron
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from flask import request
from storm.expr import Or
import uuid
from web import app, store
from db import Playlist, User, Track
from . import get_entity

@app.route('/rest/getPlaylists.view', methods = [ 'GET', 'POST' ])
def list_playlists():
	query = store.find(Playlist, Or(Playlist.user_id == request.user.id, Playlist.public == True)).order_by(Playlist.name)

	username = request.args.get('username')
	if username:
		if not request.user.admin:
			return request.error_formatter(50, 'Restricted to admins')

		query = store.find(Playlist, Playlist.user_id == User.id, User.name == username).order_by(Playlist.name)

	return request.formatter({ 'playlists': { 'playlist': [ p.as_subsonic_playlist(request.user) for p in query ] } })

@app.route('/rest/getPlaylist.view', methods = [ 'GET', 'POST' ])
def show_playlist():
	status, res = get_entity(request, Playlist)
	if not status:
		return res

	info = res.as_subsonic_playlist(request.user)
	info['entry'] = [ t.as_subsonic_child(request.user) for t in res.tracks ]
	return request.formatter({ 'playlist': info })

@app.route('/rest/createPlaylist.view', methods = [ 'GET', 'POST' ])
def create_playlist():
	# Only(?) method where the android client uses form data rather than GET params
	playlist_id, name = map(lambda x: request.args.get(x) or request.form.get(x), [ 'playlistId', 'name' ])
	# songId actually doesn't seem to be required
	songs = request.args.getlist('songId') or request.form.getlist('songId')
	try:
		playlist_id = uuid.UUID(playlist_id) if playlist_id else None
		songs = set(map(uuid.UUID, songs))
	except:
		return request.error_formatter(0, 'Invalid parameter')

	if playlist_id:
		playlist = store.get(Playlist, playlist_id)
		if not playlist:
			return request.error_formatter(70, 'Unknwon playlist')

		if playlist.user_id != request.user.id and not request.user.admin:
			return request.error_formatter(50, "You're not allowed to modify a playlist that isn't yours")

		playlist.tracks.clear()
		if name:
			playlist.name = name
	elif name:
		playlist = Playlist()
		playlist.user_id = request.user.id
		playlist.name = name
		store.add(playlist)
	else:
		return request.error_formatter(10, 'Missing playlist id or name')

	for sid in songs:
		track = store.get(Track, sid)
		if not track:
			return request.error_formatter(70, 'Unknown song')

		playlist.tracks.add(track)

	store.commit()
	return request.formatter({})

@app.route('/rest/deletePlaylist.view', methods = [ 'GET', 'POST' ])
def delete_playlist():
	status, res = get_entity(request, Playlist)
	if not status:
		return res

	if res.user_id != request.user.id and not request.user.admin:
		return request.error_formatter(50, "You're not allowed to delete a playlist that isn't yours")

	store.remove(res)
	store.commit()
	return request.formatter({})

@app.route('/rest/updatePlaylist.view', methods = [ 'GET', 'POST' ])
def update_playlist():
	status, res = get_entity(request, Playlist, 'playlistId')
	if not status:
		return res

	if res.user_id != request.user.id and not request.user.admin:
		return request.error_formatter(50, "You're not allowed to delete a playlist that isn't yours")

	playlist = res
	name, comment, public = map(request.args.get, [ 'name', 'comment', 'public' ])
	to_add, to_remove = map(request.args.getlist, [ 'songIdToAdd', 'songIndexToRemove' ])
	try:
		to_add = set(map(uuid.UUID, to_add))
		to_remove = sorted(set(map(int, to_remove)))
	except:
		return request.error_formatter(0, 'Invalid parameter')

	if name:
		playlist.name = name
	if comment:
		playlist.comment = comment
	if public:
		playlist.public = public in (True, 'True', 'true', 1, '1')

	tracks = list(playlist.tracks)

	for sid in to_add:
		track = store.get(Track, sid)
		if not track:
			return request.error_formatter(70, 'Unknown song')
		if track not in playlist.tracks:
			playlist.tracks.add(track)

	for idx in to_remove:
		if idx < 0 or idx >= len(tracks):
			return request.error_formatter(0, 'Index out of range')
		playlist.tracks.remove(tracks[idx])

	store.commit()
	return request.formatter({})