愛家（租房系統）---- 修改用戶信息

裝飾器

用戶模塊的很多操作都是需要在登錄狀態下完成的，所以需要使用裝飾器對登錄狀態進行檢驗。本項目中的裝飾器在工具集（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){
        }
    });
})