supysonic/api/media.py

# coding: utf-8

from flask import request, send_file
import os.path, uuid
import Image
from time import time as now

import config
from web import app
from db import Track, Folder, User
from lastfm import LastFm

@app.route('/rest/stream.view', methods = [ 'GET', 'POST' ])
def stream_media():
	id, maxBitRate, format, timeOffset, size, estimateContentLength = map(request.args.get, [ 'id', 'maxBitRate', 'format', 'timeOffset', 'size', 'estimateContentLength' ])
	if not id:
		return request.error_formatter(10, 'Missing media id')

	try:
		tid = uuid.UUID(id)
	except:
		return request.error_formatter(0, 'Invalid media id')

	track = Track.query.get(tid)
	if not track:
		return request.error_formatter(70, 'Media not found'), 404

	if maxBitRate:
		try:
			maxBitRate = int(maxBitRate)
		except:
			return request.error_formatter(0, 'Invalid bitrate value')

		if track.bitrate > maxBitRate:
			# TODO transcode
			pass

	if format != 'mp3':
		# TODO transcode
		pass

	if estimateContentLength == 'true':
		return send_file(track.path), 200, { 'Content-Length': os.path.getsize(track.path) }

	return send_file(track.path)

@app.route('/rest/getCoverArt.view', methods = [ 'GET', 'POST' ])
def cover_art():
	id = request.args.get('id')
	if not id:
		return request.error_formatter(10, 'Missing cover art id')
	try:
		fid = uuid.UUID(id)
	except:
		return request.error_formatter(0, 'Invalid cover art id')
	folder = Folder.query.get(fid)
	if not folder or not folder.has_cover_art or not os.path.isfile(os.path.join(folder.path, 'cover.jpg')):
		return request.error_formatter(70, 'Cover art not found')

	size = request.args.get('size')
	if size:
		try:
			size = int(size)
		except:
			return request.error_formatter(0, 'Invalid size value')
	else:
		return send_file(os.path.join(folder.path, 'cover.jpg'))

	im = Image.open(os.path.join(folder.path, 'cover.jpg'))
	if size > im.size[0] and size > im.size[1]:
		return send_file(os.path.join(folder.path, 'cover.jpg'))

	size_path = os.path.join(config.get('CACHE_DIR'), str(size))
	path = os.path.join(size_path, id)
	if os.path.exists(path):
		return send_file(path)
	if not os.path.exists(size_path):
		os.makedirs(size_path)

	im.thumbnail([size, size], Image.ANTIALIAS)
	im.save(path, 'JPEG')
	return send_file(path)

@app.route('/rest/scrobble.view', methods = [ 'GET', 'POST' ])
def scrobble():
	tid, time, submission, u = map(request.args.get, [ 'id', 'time', 'submission', 'u' ])
	if not tid:
		return request.error_formatter(10, 'Missing file id')
	try:
		tid = uuid.UUID(tid)
	except:
		return request.error_formatter(0, 'Invalid file id')
	track = Track.query.get(tid)
	if not track:
		return request.error_formatter(70, 'File not found')

	if time:
		try:
			time = int(time) / 1000
		except:
			return request.error_formatter(0, 'Invalid time value')
	else:
		time = int(now())

	user = User.query.filter(User.name == u).one()
	lfm = LastFm(user, app.logger)

	if submission in (None, '', True, 'true', 'True', 1, '1'):
		lfm.scrobble(track, time)
	else:
		lfm.now_playing(track)

	return request.formatter({})