1. 程式人生 > >django寫使用者登入判定並跳轉制定頁面

django寫使用者登入判定並跳轉制定頁面

1.首先看要設定登陸的介面 book/view.py

@user_util.my_login    #相當於 select_all=my_login(select_all)
def select_all(request):
    # 查詢所有的書
    book_list = BookInfo.objects.all()
    # 返回
    return render(request, 'book/book_list.html', {'book_list': book_list})


@user_util.my_login    #相當於 select_by_id=my_login(select_by_id)
def select_by_id(request,book_id):
    bookinfo=BookInfo.objects.get(id=book_id)
    return render(request,'book/book_detail.html',{'bookinfo':bookinfo})

@user_util.my_login 是在utils/user_utils.py裡面寫的裝飾器
使用session判定是否登入:login_user_id = args[0].session.get(‘login_user_id’)

#登陸用的裝飾器

def my_login(func):
    def inner(*args,**kwargs):
        login_user_id = args[0].session.get('login_user_id')
        if login_user_id:
            return func(*args,**kwargs)
        else:
            return redirect(reverse('user:login'))
    return inner

2.設定登入模組的介面
設定session鍵值對進行儲存: request.session[‘login_user_id’]=user[0].id

def login(request):
    #獲取cookie
    remember_user_name=request.COOKIES.get('remember_user_name','')
    return render(request, "user/login.html",{'remember_user_name':remember_user_name})


def login_handler(request):
    # 判斷請求方式
    if request.method == "GET":
        request_info = request.GET
    else:
        request_info = request.POST

    # 獲取屬性
    user_name = request_info.get("user_name",'').strip()
    user_pwd = request_info.get("user_pwd",'').strip()
    remember = request_info.get("remember")
    user_pwd = my_md5(user_pwd)
    user=User.objects.filter(name=user_name, pwd=user_pwd)
    #print(user[0].id)

    # 查詢
    if len(user) != 0:
        #將登陸資訊儲存到session
        request.session['login_user_id']=user[0].id
        resp=HttpResponseRedirect(reverse("book:index"))
        #記住使用者名稱
        if remember=='1':
            resp.set_cookie('remember_user_name',user_name,3600*24*7)
        else:
            resp.set_cookie('remember_user_name',user_name,0)
        return resp


        # 轉發
        #return book_views.index(request)
        # 重定向
        #return HttpResponseRedirect(reverse("book:index"))

    else:

        return HttpResponseRedirect(reverse("user:login"))