装饰器
用户模块的很多操作都是需要在登录状态下完成的,所以需要使用装饰器对登录状态进行检验。本项目中的装饰器在工具集(utils)的functions.py文件中定义:
'''
装饰器(验证登录状态)
'''
def is_login(view_fun):
@functools.wraps(view_fun)
def decorator():
try:
if 'user_id' in session:
return view_fun()
else:
return redirect('/user/login/')
except Exception as e:
return redirect('/user/login/')
return decorator
具体实现的功能:如果该用户出于登录状态,即允许用户继续自己的操作;如果该用户为出于登录状态则返回登录界面。
对退出及用户主页的业务实现:
'''
退出
'''
@user_blueprint.route('/logout/', methods=['DELETE'])
@is_login
def user_logout():
session.clear()
return jsonify(status_code.SUCCESS)
'''
登录之后进入用户首页
'''
@user_blueprint.route('/my/')
@is_login
def my():
return render_template('my.html')
退出业务对应的Ajax:
function logout() {
$.ajax({
url:'/user/logout/',
type:'DELETE',
success:function(data) {
if(data.code=='200') {
location.href = '/house/index/';
}
}
});
}
修改用户信息:
'''
修改用户信息
'''
@user_blueprint.route('/profile/', methods=['GET', 'PUT'])
@is_login
def profile():
if request.method == 'GET':
"""
跳转至修改用户信息界面
"""
return render_template('profile.html')
if request.method == 'PUT':
"""
处理用户提出的修改请求
"""
name = request.form.get('name')
avatar = request.files.get('avatar')
if avatar:
try:
# mime-type:国际规范,表示文件的类型,如text/html,text/xml,image/png,image/jpeg..
if not re.match('image/.*', avatar.mimetype):
return jsonify(status_code.USER_PROFILE_IMAGE_UPDATE_ERROR)
except:
return jsonify(code=status_code.PARAMS_ERROR)
# 保存到media中
con = Config()
url = os.path.join(con.UPLOAD_FOLDER, avatar.filename)
avatar.save(url)
# 保存用户的头像信息
try:
user = User.query.get(session['user_id'])
user.avatar = os.path.join('/static/upload', avatar.filename)
user.add_update()
except:
return jsonify(status_code.DATABASE_ERROR)
# 返回图片信息
return jsonify(code='200', url=os.path.join('/static/upload', avatar.filename))
elif name:
# 判断用户名是否存在
if User.query.filter_by(name=name).count():
return jsonify(status_code.USER_REGISTER_USER_IS_EXSITS)
else:
user = User.query.get(session['user_id'])
user.name = name
user.add_update()
return jsonify(status_code.SUCCESS)
else:
return jsonify(status_code.PARAMS_ERROR)
修改用户信息分为修改用户头像和用户昵称两个部分,分别对应两个响应Ajax局部刷新请求:
$(document).ready(function() {
$('#form-avatar').submit(function (e) {
e.preventDefault();
$(this).ajaxSubmit({
url: "/user/profile/",
type: "put",
dataType: "json",
success: function (data) {
if (data.code == '200') {
$('#user-avatar').attr('src','/static/upload/'+data.avatar);
} else {
$('#error_msg1').show();
}
}
});
return false;
});
$('#form-name').submit(function (e) {
e.preventDefault();
var name = $('#user-name').val();
$.ajax({
url:'/user/profile/',
type:'put',
data:{'name':name},
success:function (data) {
if(data.code== '200'){
showSuccessMsg();
}else{
$('.error_msg2').html('<i class="fa fa-exclamation-circle"></i>' +data.msg);
$('.error_msg2').show();
}
}
});
return false;
});
})
在用户主页中展示用户信息:
'''
返回用户数据
'''
@user_blueprint.route('/user/')
@is_login
def get_user_profile():
"""
返回用户数据
:return: 数据库中当前用户信息
"""
# 获取当前登录的用户
user_id = session['user_id']
# 查询当前用户的头像、用户名、手机号,并返回
user = User.query.get(user_id)
return jsonify(user=user.to_basic_dict())
接收后台返回的用户数据并渲染到前台的Ajax:
$(document).ready(function(){
$.ajax({
url:'/user/user/',
type:'GET',
dataType:'json',
success:function(data) {
$('#user-avatar').attr('src',data.user.avatar);
$('#user-name').html(data.user.name);
$('#user-mobile').text(data.user.phone);
},
error:function(data){
}
});
})
网友评论