编写注册接口

目录

• 1.编写注册接口
  • 1.1user/urls.py中添加路由
  • 1.2user/views.py中写注册试图函数
• 2.重写django认证
  • 2.1syl/settings.py中指定自定义后端认证函数位置
  • 2.2 user/utils.py中重写认证函数
• 3.注册用户&测试登录
  • 3.1注册接口测试
  • 3.2登录接口测试
  • 3.3获取用户列表接口测试

1.编写注册接口

1.1user/urls.py中添加路由

urlpatterns = [
    path('register/', view.RegisterView.as_view()),  # 注册视图
]

1.2user/views.py中写注册试图函数

class RegisterView(APIView):
    """
    用户注册，权限是：匿名可以访问
    """
    # 自定义权限类
    permission_classes = (AllowAny)
    
    def post(self, request):
        """
        接受邮箱和密码，前端校验两遍一致性，注册成功后返回成功，然后用户自行登录获取token
        1.随机用户名
        2.生成用户
        3.设置用户密码
        4.保存用户
        :params request:
        :return: {'code': 0, 'msg': '注册成功'}
        """
        email = request.data.get("email")
        password = request.data.get("password")
        
        if all([email, password]):
            pass
        else:
            return Response({"code": 9999, "msg": "参数不齐"})
        
        rand_name = self.randomUsername()
        user = User(username=rand_name, email=email)
        user.set_password(password)
        user.save()
        
        return Response({"code": 0, "msg": "注册成功"})
        
        def randomUsername(self):
            """
            生成随机用户名：格式：SYL + 年月日时分 + 5位随机数
            ：return:
            """
            d = datetime.datetime.now()
            base = 'SYL'
            time_str = "%04d%02d%02d%02d%02d" % (d.year, d.month, d.day, d.hour, d.minute)
            rand_num = str(random.randint(1000, 99999))
            
            return base + time_str + rand_num

2.重写django认证

2.1syl/settings.py中指定自定义后端认证函数位置

# 自定义验证后端
AUTHENTICATION_BACKENDS = ["user.utils.EmailAuthBackend"]

2.2 user/utils.py中重写认证函数

# 以前使用username进行用户验证，现在修改成email进行验证
class EmailAuthBackend:
    def authenticate(self, request, username=None, password=None):
        try:
            user = User.objects.get(username=username)
        except Exception as e:
            user = None
        if not user:
            try:
                user = User.objects.get(email=username)
            except Exception as e:
                user = None
        if use and user.check_password(password):
            return user
        else:
            return None
        
    def get_user(self, user_id):
        try:
            return User.objects.get(pk=user_id)
        except User.DoesNotExist:
            return None

3.注册用户&测试登录

3.1注册接口测试

http://192.168.56.100:8888/user/register/

3.2登录接口测试

注：认证时只能识别username, 所以必须要在请求携带username字段

http://192.168.56.100:8888/user/login

3.3获取用户列表接口测试

• 访问接口

http://192.168.56.100:8888/user/user/5/

• 测试自定义权限

# 自定义权限类
permission_classes = (MyPermission, )
# 自定义认证类，自定义会覆盖全局配置
authentication_classes = (JSONWebTokenAuthentication, )