装饰器

用户模块的很多操作都是需要在登录状态下完成的，所以需要使用装饰器对登录状态进行检验。本项目中的装饰器在工具集（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){
        }
    });
})