def edit_file(file_id):
if request.method == 'GET':
f = files.get_file_by_id(file_id)
folders = files.get_folders(f.folderid) if f.folderid else []
return render_template('file.html', file=f, folders=folders)
if request.method == 'POST':
errors = files.edit_file(file_id, request.form)
if errors:
f = files.get_file_by_id(file_id)
folders = files.get_folders(f.folderid) if f.folderid else []
return render_template('file.html', file=f, folders=folders)
referrer = request.args.get('referrer', url_for('list_file'))
return redirect(referrer)
def search_folders():
if 'user' not in session:
abort(404)
query = request.args.get('q', None)
if query is None:
return redirect(request.referrer)
user = session['user']
folders = files.get_folders(user.id if user.id != 1 else None)
folders = [ f for f in folders if f.name.find(query) >= 0 ]
return render_template('folders.html', folders=folders, search_view=True, query=query)
def add_file_to_folder(file_id, folder_id):
user = session['user']
# superuser can not add file to folder
if user.id == 1:
abort(404)
# add file to folder
if folder_id:
files.add_file_to_folder(file_id, folder_id, user.id)
return redirect(url_for('list_file'))
folders = files.get_folders(user.id if user.id != 1 else None)
return render_template('folders.html', folders=folders, file_id=file_id, type='addto')
def list_folder():
user = session['user']
folders = files.get_folders(user.id if user.id != 1 else None)
return render_template('folders.html', folders=folders)
