def home():
if not session.get('logged_in'):
return render_template('login.html')
else:
files = filehandler()
clnt = client()
resp = files.list_files(session['username'])
user = clnt.get_client(session['username'])
return render_template('layout.html', files=resp, user=user)
def delete():
my_file = filehandler()
if request.method == 'POST':
try:
del_file = request.form['filename']
my_file.del_file(session['username'], del_file)
except:
return home()
return home()
def download():
my_file = filehandler()
if request.method == 'GET':
base_url, infos = str(request.url).split("/download?")
username = ast.literal_eval(infos)['username']
filename = ast.literal_eval(infos)['filename']
attach = "attachment;filename=" + filename
return Response(my_file.get_object(username, filename),
mimetype='text/plain',
headers={"Content-Disposition": attach})
return home()
def edit():
my_file = filehandler()
base_url, infos = str(request.url).split("/edit?")
username = ast.literal_eval(infos)['username']
filename = ast.literal_eval(infos)['filename']
filedesc = ast.literal_eval(infos)['filedesc']
session['editname'] = username
session['editfile'] = filename
session['editdesc'] = filedesc
return render_template('edit.html',
username=username,
filename=filename,
filedesc=filedesc)
def upload():
my_file = filehandler()
if request.method == 'POST':
try:
body = request.files['upload_file']
except Exception as err:
return redirect(
url_for('error',
desc=str(err),
action_url=str.rstrip(request.url, 'upload')))
filename = secure_filename(body.filename)
filedesc = request.form['filedesc']
my_file.add_file(session['username'], filename, filedesc, body)
return home()
def save():
my_file = filehandler()
new_file = request.form['filename']
new_desc = request.form['filedesc']
if (new_file == ''):
new_file = session['editfile']
if (new_desc == ''):
new_desc = session['editdesc']
my_file.update_object(session['editname'], session['editfile'], new_file,
new_desc)
return home()
def user_delete():
my_clnt = client()
files = filehandler()
code = 0
message = ''
if request.method == 'POST':
try:
del_user = request.form['username']
if (del_user == session['username']):
return redirect(
url_for('error',
desc="Can't delete same Admin User",
action_url=str.rstrip(request.url, 'user_delete')))
files.del_userfiles(del_user)
code, message = my_clnt.del_client(del_user)
except:
return redirect(
url_for('error',
desc=message,
action_url=str.rstrip(request.url, 'user_delete')))
return home()
return admin()