supysonic/folder.py

# coding: utf-8

from flask import request, flash, render_template, redirect, url_for, session as fl_sess
import os.path
import uuid

from web import app
from db import session, Folder
from scanner import Scanner
from managers.user import UserManager
from managers.folder import FolderManager

@app.before_request
def check_admin():
	if not request.path.startswith('/folder'):
		return

	if not UserManager.get(fl_sess.get('userid'))[1].admin:
		return redirect(url_for('index'))

@app.route('/folder')
def folder_index():
	return render_template('folders.html', folders = Folder.query.filter(Folder.root == True).all())

@app.route('/folder/add', methods = [ 'GET', 'POST' ])
def add_folder():
	if request.method == 'GET':
		return render_template('addfolder.html')

	error = False
	(name, path) = map(request.form.get, [ 'name', 'path' ])
	if name in (None, ''):
		flash('The name is required.')
		error = True
	if path in (None, ''):
		flash('The path is required.')
		error = True
	if error:
		return render_template('addfolder.html')

	ret = FolderManager.add(name, path)
	if ret != FolderManager.SUCCESS:
		flash(FolderManager.error_str(ret))
		return render_template('addfolder.html')

	flash("Folder '%s' created. You should now run a scan" % name)

	return redirect(url_for('folder_index'))

@app.route('/folder/del/<id>')
def del_folder(id):
	try:
		idid = uuid.UUID(id)
	except ValueError:
		flash('Invalid folder id')
		return redirect(url_for('folder_index'))

	ret = FolderManager.delete(idid)
	if ret != FolderManager.SUCCESS:
		flash(FolderManager.error_str(ret))
	else:
		flash('Deleted folder')

	return redirect(url_for('folder_index'))

@app.route('/folder/scan')
@app.route('/folder/scan/<id>')
def scan_folder(id = None):
	s = Scanner(session)
	if id is None:
		for folder in Folder.query.filter(Folder.root == True):
			FolderManager.scan(folder.id, s)
	else:
		status = FolderManager.scan(id, s)
		if status != FolderManager.SUCCESS:
			flash(FolderManager.error_str(status))
			return redirect(url_for('folder_index'))

	added, deleted = s.stats()
	session.commit()

	flash('Added: %i artists, %i albums, %i tracks' % (added[0], added[1], added[2]))
	flash('Deleted: %i artists, %i albums, %i tracks' % (deleted[0], deleted[1], deleted[2]))
	return redirect(url_for('folder_index'))