from rest_framework import serializers
from Course.models import Account
import hashlibclass RegisterSerializer(serializers.ModelSerializer):class Meta:model = Accountfields = "__all__"def create(self, validated_data):pwd = validated_data["pwd"]pwd_salt = "luffy_password" + pwd   # 加盐处理md5_str = hashlib.md5(pwd_salt.encode()).hexdigest()user_obj = Account.objects.create(username=validated_data["username"], pwd=md5_str)return user_obj

from rest_framework.views import APIView
from rest_framework.response import Response
from .serializers import RegisterSerializer
from utils.base_response import BaseResponse
from Course.models import Account
from utils.redis_pool import POOL
import redis
import uuid
from utils.my_auth import LoginAuth
from utils.geetest import GeetestLib
from django.http import HttpResponse
import json# Create your views here.class RegisterView(APIView):def post(self, request):res = BaseResponse()   # 实例化响应类# 用序列化器做校验ser_obj = RegisterSerializer(data=request.data)if ser_obj.is_valid():ser_obj.save()res.data = ser_obj.dataelse:res.code = 1020res.error = ser_obj.errorsreturn Response(res.dict)class LoginView(APIView):def post(self, request):res = BaseResponse()username = request.data.get("username", "")pwd = request.data.get("pwd", "")user_obj = Account.objects.filter(username=username, pwd=pwd).first()if not user_obj:res.code = 1030res.error = "用户名或密码错误"return Response(res.dict)# 用户登录成功生成一个token写入redis# 写入redis  token : user_idconn = redis.Redis(connection_pool=POOL) # 从连接池获取一个连接try:token = uuid.uuid4()conn.set(str(token), user_obj.id, ex=10000)# conn.set(str(token), user_obj.id)          # str(token) 用作keyres.data = tokenexcept Exception as e:print(e)res.code = 1031res.error = "创建令牌失败"return Response(res.dict)

class BaseResponse(object):"""封装响应的类"""def __init__(self):self.code = 1000self.data = Noneself.error = None@propertydef dict(self):return self.__dict__        # 返回字典

import redisfrom rest_framework.authentication import BaseAuthentication
from rest_framework.exceptions import AuthenticationFailedfrom .redis_pool import POOL
from Course.models import AccountCONN = redis.Redis(connection_pool=POOL)class LoginAuth(BaseAuthentication):def authenticate(self, request):# 从请求头中获取前端带过来的token。或者跟前端商量好token = request.META.get("HTTP_AUTHENTICATION", "")# print(request.META)# print(token)if not token:raise AuthenticationFailed("没有携带token")# 去redis比对user_id = CONN.get(str(token))  # get没有的值返回 Noneif user_id == None:raise AuthenticationFailed("token过期")  # 重新登录设置tokenuser_obj = Account.objects.filter(id=user_id).first()return user_obj, token

import redisPOOL = redis.ConnectionPool(host="127.0.0.1", port=6379, decode_responses=True, max_connections=10)