Django-重写登录认证,用户名和手机号都可以登录

1.自定义用户登录认证类

def get_user_by_account(account):
    """
    自定义 手机号和用户名都可以登录
    """
    try:
        if re.match('^1[3-9]\d{9}$', account):
            user = UserProfile.objects.get(phone=account)
        else:
            user = UserProfile.objects.get(username=account)
    except UserProfile.DoesNotExist:
        return None
    else:
        return user


class UsernameMobileAuthBackend(ModelBackend):
    """
    自定义用户名或手机号认证
    """
    def authenticate(self, request, username=None, password=None, **kwargs):
        user = get_user_by_account(username)
        if user is not None and user.check_password(password):
            return user

2.在settings.py中添加

AUTHENTICATION_BACKENDS = [
    'users.utils.UsernameMobileAuthBackend',
]

 

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
可以使用Django自带的User模型来实现用户注册和登录,同时结合Django-rest-framework提供的Authentication和Permission来保护API接口。 首先,定义一个UserSerializer用于序列化User模型: ```python from django.contrib.auth.models import User from rest_framework import serializers class UserSerializer(serializers.ModelSerializer): class Meta: model = User fields = ('id', 'username', 'password', 'email', 'first_name', 'last_name') extra_kwargs = {'password': {'write_only': True}} ``` 在用户注册时,需要对密码进行加密并保存到数据库中。这可以通过在UserSerializer中添加create()方法来实现: ```python class UserSerializer(serializers.ModelSerializer): class Meta: model = User fields = ('id', 'username', 'password', 'email', 'first_name', 'last_name') extra_kwargs = {'password': {'write_only': True}} def create(self, validated_data): user = User.objects.create_user( username=validated_data['username'], password=validated_data['password'], email=validated_data.get('email', ''), first_name=validated_data.get('first_name', ''), last_name=validated_data.get('last_name', ''), ) return user ``` 然后,在views.py中定义一个UserViewSet来处理用户注册和登录: ```python from django.contrib.auth import authenticate, login from django.contrib.auth.models import User from rest_framework import viewsets, status from rest_framework.decorators import action from rest_framework.response import Response from .serializers import UserSerializer class UserViewSet(viewsets.ModelViewSet): queryset = User.objects.all() serializer_class = UserSerializer @action(detail=False, methods=['post']) def register(self, request): serializer = UserSerializer(data=request.data) if serializer.is_valid(): user = serializer.save() return Response({'id': user.id, 'username': user.username}, status=status.HTTP_201_CREATED) else: return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) @action(detail=False, methods=['post']) def login(self, request): username = request.data.get('username') password = request.data.get('password') user = authenticate(request, username=username, password=password) if user is not None: login(request, user) return Response({'id': user.id, 'username': user.username}, status=status.HTTP_200_OK) else: return Response({'error': 'Invalid username or password'}, status=status.HTTP_401_UNAUTHORIZED) ``` 在上面的代码中,register()方法用于处理用户注册请求,login()方法用于处理用户登录请求。在登录时,使用Django自带的authenticate()函数进行认证,如果认证通过,则调用Django自带的login()函数进行登录。 最后,在urls.py中添加UserViewSet的路由: ```python from django.urls import path, include from rest_framework import routers from .views import UserViewSet router = routers.DefaultRouter() router.register(r'users', UserViewSet, basename='user') urlpatterns = [ path('', include(router.urls)), ] ``` 现在,就可以使用以下API接口进行用户注册和登录: - POST /users/register/ - 用户注册 - POST /users/login/ - 用户登录 在注册时需要传递的参数为:用户名、密码、电子邮件、名字和姓氏。在登录时需要传递的参数为:用户名和密码。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值