編寫註冊介面

是SY呀發表於2020-11-02

1.編寫註冊介面

1.1 user/urls.py 中新增路由

 

urlpatterns = [
     path('register/', views.RegisterView.as_view()), 
    # 註冊檢視, /user/register/
 ]

1.2 user/views.py 中寫註冊檢視函式

class RegisterView(APIView):
    permission_classes = (AllowAny,)
    def post(self,request):
        """
        接收郵箱和密碼, 前端校驗兩遍一致性, 註冊成功後返回成功, 然後使用者自行登入獲取token
        1. 隨機使用者名稱
        2. 生成使用者
        3. 設定使用者密碼
        4. 儲存使用者
        :param 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):
        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(10000, 99999))
        return base + time_str + rand_num

2.重寫django認證

 

2.1 syl/settings.py 中指定自定義後端認證函式位置

# 自定義驗證後端 
AUTHENTICATION_BACKENDS = ['user.utils.EmailAuthBackend']

2.2 user/utils.py 中重寫認證函式

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 user 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,)

 

 

相關文章