django与小程序实现登录验证功能的示例代码

下面就是详细讲解“django与小程序实现登录验证功能的示例代码”的完整攻略。

环境准备

首先，我们需要在本地搭建好Django和小程序开发环境。具体搭建方法可以参考官方文档。

实现过程

1. 创建用户模型

在Django中，我们通常使用Django自带的用户模型（User）来实现用户管理。我们需要先在models.py中定义一个用户模型，并利用Django自带的模型方法create_user()来创建用户。

from django.contrib.auth.models import AbstractBaseUser, BaseUserManager
from django.db import models

class UserManager(BaseUserManager):
    def create_user(self, email, password=None, **extra_fields):
        if not email:
            raise ValueError('The Email field must be set')
        email = self.normalize_email(email)
        user = self.model(email=email, **extra_fields)
        user.set_password(password)
        user.save(using=self._db)
        return user

class User(AbstractBaseUser):
    email = models.EmailField(unique=True)
    first_name = models.CharField(max_length=100, blank=True)
    last_name = models.CharField(max_length=100, blank=True)
    is_active = models.BooleanField(default=True)
    is_staff = models.BooleanField(default=False)

    objects = UserManager()

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = []

    def __str__(self):
        return f"{self.email}"

1. 视图函数实现登录验证

在Django中，我们可以使用装饰器@login_required来实现登录验证。首先，在views.py中导入这个装饰器，然后在需要进行登录验证的视图函数上添加此装饰器即可。

from django.contrib.auth.decorators import login_required
from django.shortcuts import render

@login_required
def index(request):
    return render(request, 'index.html')

1. 微信小程序实现登录验证

在小程序中，我们需要先在app.js中定义一个全局变量session_key，用来保存用户的会话密钥。然后在登录成功后，我们通过wx.request()方法向后端发送信息，请求后端返回用户信息。

App({
  onLaunch: function() {
    var that = this;
    wx.login({
      success: function(login) {
        wx.request({
          url: 'http://localhost:8000/api/login/',
          method: 'POST',
          data: {
            code: login.code
          },
          success: function(res) {
            that.globalData.session_key = res.data.session_key;
          }
        })
      }
    })
  }
})

在views.py中，我们可以使用Django REST framework提供的APIView来实现登录接口。具体代码如下所示：

from rest_framework.views import APIView
from rest_framework.response import Response
from django.contrib.auth import authenticate, login

class LoginView(APIView):
    def post(self, request):
        code = request.data.get('code')
        # 获取openid和session_key
        res = requests.get(
            f'https://api.weixin.qq.com/sns/jscode2session?'\
            f'appid={APPID}&'\
            f'secret={SECRET}&'\
            f'js_code={code}&'\
            'grant_type=authorization_code')
        openid = res.json().get('openid')
        session_key = res.json().get('session_key')

        # 验证openid是否存在，不存在则创建新用户并登录
        user = User.objects.filter(openid=openid).first()
        if not user:
            user = User.objects.create(email=openid+'@qq.com', openid=openid)
            login(request, user)

        # 将session_key保存到用户信息中
        user.session_key = session_key
        user.save()

        return Response({'session_key': session_key})

1. 小程序请求登录验证接口

在小程序中，我们定义一个按钮，用来触发登录流程。当用户点击按钮后，我们通过wx.login()方法获取登录凭证（code），然后将该凭证信息发送至后端的登录接口中进行验证。

<!-- index.wxml -->
<button bindtap="login">登录</button>

// index.js
const app = getApp();
Page({
  login: function() {
    wx.login({
      success: function (res) {
        wx.request({
          url: 'http://localhost:8000/api/login/',
          method: 'POST',
          data: {
            code: res.code
          },
          success: function (res) {
            // 保存session_key，用于后续接口调用
            app.globalData.session_key = res.data.session_key;
            wx.showToast({
              title: '登录成功',
              duration: 2000
            });
          }
        })
      }
    })
  }
})

示例说明

1. 示例1：在Django中实现登录验证

假设我们有一个需要登录才可访问的视图（例如“我的订单”），我们可以通过在视图函数上添加@login_required装饰器来实现登录验证。在用户未登录时，Django会自动跳转至登录页面。

@login_required
def my_orders(request):
    # TODO: 展示我的订单
    pass

1. 示例2：小程序向后端发送验证请求

当用户在小程序中点击登录按钮后，小程序会向后端的登录验证接口发送请求。后端验证通过后，将session_key返回给小程序。小程序保存该session_key，用于后续的用户验证以及接口调用。

wx.login({
  success: function (res) {
    wx.request({
      url: 'http://localhost:8000/api/login/',
      method: 'POST',
      data: {
        code: res.code
      },
      success: function (res) {
        app.globalData.session_key = res.data.session_key;
        wx.showToast({
          title: '登录成功',
          duration: 2000
        });
      }
    })
  }
})

以上就是Django与小程序实现登录验证功能的示例代码攻略。希望能对你有所帮助！