def file_rename(filename):
user_dir = user_directory()
originfile = user_dir.get() + filename
originfilepath = os.path.split(originfile)[0] + '/'
newfile = originfilepath + request.form['newname']
os.rename(originfile, newfile)
return redirect(url_for('file.file_mgmt'))
def file_upload():
if request.method == 'GET':
fileName = None
else:
fileObj = request.files['uploadFileName']
fileName = fileObj.filename
user_dir = user_directory()
fileDir = user_dir.get()
if fileName:
fileObj.save(fileDir + str(fileName))
return render_template('file/file_upload.html', filename=fileName)
def user_login():
form = LoginForm()
if request.method == 'POST' and form.validate():
table_instance = table(g.db_ip, g.db_port, g.db_user, go.db_pwd,
g.db_name, g.db_charset, g.db_table)
if table_instance.query_user(form.username.data) and \
table_instance.query_password(form.username.data) == form.password.data:
session['username'] = form.username.data
user_dir = user_directory()
user_dir.set(form.username.data)
flash(u'登陆成功', 'success')
else:
flash(u'用户名或者密码错误', 'danger')
table_instance.close_table()
return render_template('user/user_LogReg.html', form=form)
def user_registe():
form = RegisteForm()
if request.method == 'POST' and form.validate():
table_instance = table(g.db_ip, g.db_port, g.db_user, go.db_pwd,
g.db_name, g.db_charset, g.db_table)
if table_instance.query_user(form.username.data):
flash(u'用户名已存在', 'danger')
else:
table_instance.add_user(form.username.data, form.password.data)
user_dir = user_directory()
user_dir.create(form.username.data)
user_dir.set(form.username.data)
flash(u'注册成功', 'success')
table_instance.close_table()
return render_template('user/user_LogReg.html', form=form)
def file_mgmt():
path_lists = []
user_dir = user_directory()
for lst in os.walk(user_dir.get()):
path_dicts = {'root': '', 'dirs': '', 'files': ''}
path_dicts['root'] = lst[0].replace(user_dir.get(), '/')
path_dicts['dirs'] = lst[1]
path_dicts['files'] = lst[2]
path_lists.append(path_dicts)
filename_lists = []
for path in path_lists:
for file in path['files']:
filename_lists.append(os.path.join(path['root'], file))
file_lists = []
for filename in filename_lists:
file_dicts = {'filename': '', 'filefullname': ''}
file_dicts['filename'] = os.path.split(filename)[1]
file_dicts['filefullname'] = filename
file_lists.append(file_dicts)
return render_template('file/file_mgmt.html', file_lists=file_lists)
def file_delete(filename):
user_dir = user_directory()
originfile = user_dir.get() + filename
os.remove(originfile)
return redirect(url_for('file.file_mgmt'))
def file_download(filename):
user_dir = user_directory()
return send_from_directory(user_dir.get(), filename, as_attachment=True)