django rest_framework 實現用戶登錄認證

 

1、安裝

pip install djangorestframework

 

 

2、創建項目及應用

 創建過程略

目錄結構如圖

 

3、設置settings.py

設置數據庫連接

# MySQL  增加mysql 連接
DATABASES = {
    'default':{
     'ENGINE':'django.db.backends.mysql',
     'HOST':'127.0.0.1',
     'PORT':'3306',
     'NAME':'dbname',  # 數據庫名
     'USER':'username',
     'PASSWORD':'password',
     'OPTIONS':{
         'sql_mode': 'traditional'
     },
    }
}

 

添加應用

# Application definition

INSTALLED_APPS = [
     ...
    'rest_framework',
    'api',
]

 

 4、創建用戶信息models,在models.py

from django.db import models

# Create your models here.

class User(models.Model):
    username = models.CharField(max_length=32,unique=True)
    password = models.CharField(max_length=32)

    class Meta:
        db_table = 'user'
        verbose_name = verbose_name_plural = '用戶信息表'

class userToken(models.Model):
    username = models.OneToOneField(to='User',on_delete=models.DO_NOTHING)
    token = models.CharField(max_length=60)

    class Meta:
        db_table =  'user_token'
        verbose_name = verbose_name_plural = '用戶token表'

 

在項目下執行命令生成數據表

python manage.py makemigrations 

python manage.py migrate 

 

5、在views.py 文件里,編寫登錄驗證程序

from django.shortcuts import render

# Create your views here.

import time
from api import models
from django.http import JsonResponse
from rest_framework.views import APIView

class AuthView(APIView):

    def post(self,request,*args,**kwargs):

        ret = {'code':1000,'msg':None}
        try:
            # 參數是datadict 形式
            usr = request.data.get('username')
            pas = request.data.get('password')

            # usr = request._request.POST.get('username')
            # pas = request._request.POST.get('password')

            # usr = request.POST.get('username')
            # pas = request.POST.get('password')

            print(usr)
            # obj = models.User.objects.filter(username='yang', password='123456').first()
            obj = models.User.objects.filter(username=usr,password=pas).first()
            print(obj)
            print(type(obj))
            print(obj.username)
            print(obj.password)
            if not obj:
                ret['code'] = '1001'
                ret['msg'] = '用戶名或者密碼錯誤'
                return JsonResponse(ret)
                # 里為了簡單，應該是進行加密，再加上其他參數
            token = str(time.time()) + usr
            print(token)
            models.userToken.objects.update_or_create(username=obj, defaults={'token': token})
            ret['msg'] = '登錄成功'
            #ret['token'] = token
        except Exception as e:
            ret['code'] = 1002
            ret['msg'] = '請求異常'
        return JsonResponse(ret)

 

6、在urls.py設置訪問路徑

from django.contrib import admin
from django.urls import path
from django.conf.urls import url

from api.views import AuthView

urlpatterns = [
    path('admin/', admin.site.urls),
    url(r'^api/v1/auth/$', AuthView.as_view()),
]

 

7、通過postman驗證登錄接口

 

 

 

 如圖顯示登錄成功。

 

 

 遇到的坑：

views獲取Post參數的方式：

# 如下有三種方式獲取參數，有些文章介紹，必須使用第二種，
# 其實不然，具體使用哪種要看參數的傳入形式而定，建議使用第一種，
#因為第一種支持key-value 和 json 格式 ，后面兩種僅支持 key-value.

usr = request.data.get('username')
pas = request.data.get('password')

# usr = request._request.POST.get('username')
# pas = request._request.POST.get('password')

# usr = request.POST.get('username')
# pas = request.POST.get('password')

 

 postman 配置

 1、如圖，如果沒有選擇設置content-type，則參數要選擇 form-data,上面獲取參數的方式 三種皆可。

 

 2、如果選擇了content-type為application/json，則body 要選擇raw ，如圖

 

 

 而此時獲取參數的方式只能選擇第一種。