開發註冊介面
阿新 • • 發佈:2020-10-07
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): """ 接收郵箱和密碼, 前端校驗兩遍一致性, 註冊成功後返回成功, 然後使用者自行登入獲取token1. 隨機使用者名稱 2. 生成使用者 3. 設定使用者密碼 4. 儲存使用者 :param request: :return: {'code':0,'msg':'註冊成功'} """ email = request.data.get('email') passwrod = request.data.get('password') if all([email, passwrod]): pass else: return Response({'code':9999,'msg':'引數不全'}) rand_name= self.randomUsername() user = User(username=rand_name, email=email) user.set_password(passwrod) 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(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 中重寫認證函式
# 以前使用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 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 註冊介面測試