django操作redis 登錄驗證token

1.settings里配置

增加一個字典（里面配置redis信息）

 

2.安裝django_redis模塊，並導入 from django_redis import get_redis_connection

r = get_redis_connection()# 實例一個鏈接對象r 方法里面傳settings里配置的 redis名字，不傳默認是default

r.set()存入redis

3.思路

　　登錄時首先校驗用戶名密碼，然后產生token存入 redis並且設置失效時間，具體代碼如下

models.py （生成user表）

class User(BaseModel):
    username = models.CharField(verbose_name='用戶名',max_length=50,unique=True)
    password = models.CharField(verbose_name='密碼',max_length=50)

    def __str__(self):
        return self.username

    class Meta:
        db_table = 'user'

項目中的utils.py方法，生成token和驗證token的方法

import itsdangerous #使用itsdangerous生成臨時身份令牌
from django.conf import settings
settings.SECRET_KEY #settings下的secret_key
from .const import token_expire
class Token:
    tjss = itsdangerous.TimedJSONWebSignatureSerializer(settings.SECRET_KEY, expires_in=token_expire)

    @classmethod#定義為類方法，產生token
    def create_token(cls,data):
        token = cls.tjss.dumps(data).decode()
        return token

    @classmethod#驗證token
    def check_token(cls,token):
        try:
            data = cls.tjss.loads(token)
        except:
            return False
        return data

forms.py中 寫驗證用戶名密碼的邏輯

class UserForm(forms.Form,FormatErrMsg):
    username = forms.CharField(max_length=50,min_length=4)
    password = forms.CharField(max_length=50,min_length=6)

    def clean(self):
        '''登錄校驗 查詢數據庫存的用戶密碼並且將查詢到的添加到cleaned_data里'''
        username = self.cleaned_data.get('username')
        password = self.cleaned_data.get('password')
        u = models.User.objects.filter(username=username,password=password).first()
        if u:
            self.cleaned_data['u'] = u 
            print(self.cleaned_data)
        else:
            raise forms.ValidationError('賬號/密碼錯誤')

view.py中寫產生token的代碼，存入redis里，並設置超時時間

class LoginView(View):
    def get(self,request):
        form = UserForm(request.GET)
        if form.is_valid():
            user_id = form.cleaned_data.get('u').id
            user_name = form.cleaned_data.get('u').username
            data = {'id':user_id,'username':user_name}
            token = Token.create_token(data)
            r = get_redis_connection()
            key = session_pre + user_name
            r.set(key,token,token_expire)
            return custom_response.NbResponse(token=token)
        else:
            return custom_response.NbResponse(-1,form.error_msg)

其中const.py中定義了兩個常量

token_expire = 60*60*1 #設置tonke有效期為1小時

session_pre = 'session:'