def sign_up():
name = request.form['name']
password = generate_password_hash(request.form['password'])
sex = request.form['sex']
check = User.objects.filter(name=name)
if len(check) == 0:
new_user = User(name=name, password=password, sex=sex)
new_user.save()
return flask_error_template_fun({'msg': '注册成功'})
else:
return flask_error_template_fun({'msg': '用户名已存在'})
def login():
name = request.form['name']
password = request.form['password']
user = User.objects.filter(name=name)
if len(user) == 0:
return flask_error_template_fun({'msg': '没有此用户'})
else:
user_t = user.first()
if check_password_hash(user_t.password, password):
return flask_succ_template_fun({'msg': '登录成功'})
else:
return flask_error_template_fun({'msg': '密码错误'})
def upload_file():
if request.method == 'POST':
if 'file' not in request.files:
return flask_error_template_fun({'error': 'no file input'})
file = request.files['file']
if file.filename == '':
return flask_error_template_fun({'error': 'no file input'})
if file:
filename = request.form['filename']
file.save(os.path.join(UPLOAD_FOLDER_FILE, filename))
return flask_succ_template_fun({'msg': str(filename)})
else:
return flask_succ_template_fun({'msg': 'illegal file format'})
def download_file():
filename = request.args.getlist('filename')[0]
try:
return send_from_directory(UPLOAD_FOLDER_FILE,
filename,
as_attachment=True)
except:
return flask_error_template_fun({'msg': '无此文件'})
def get_id():
name = request.form['name']
user = User.objects.filter(name=name)
if len(user) != 0:
user_id = int(user.first().userid)
return flask_succ_template_fun({'id': user_id})
else:
return flask_error_template_fun({'msg': '查无此用户'})