1. 程式人生 > 實用技巧 >編寫註冊介面

編寫註冊介面

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

3.2 登入介面測試

注: 認證時只能識別username,所以必須要在請求中攜帶username欄位

3.3 獲取使用者列表介面測試

  • 訪問介面

  • 測試自定義許可權
# 自定義許可權類
permission_classes = (MyPermission,)
# 自定義認證類, 自定義會覆蓋全域性配置
authentication_classes = (JSONWebTokenAuthentication,)