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')     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 註冊介面測試