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