mirror of
https://github.com/spl0k/supysonic.git
synced 2024-11-14 22:22:18 +00:00
c5890ab120
Changed CLI to use flask-script (much cleaner and smaller)
87 lines
2.2 KiB
Python
Executable File
87 lines
2.2 KiB
Python
Executable File
# 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 managers.user import UserManager
|
|
from managers.folder import FolderManager
|
|
|
|
import scanner
|
|
|
|
@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'))
|
|
|