1. 程式人生 > 實用技巧 >07.微博繫結使用者介面

07.微博繫結使用者介面

目錄

微博繫結使用者介面

1.oauth/urls.py中新增路由

urlpatterns = [
    path('weibo/binduser/', views.OauthWeiboBindUser.as_view()),
]

2.oauth/views.py中新增檢視函式

class OauthWeiboBindUser(APIView):
    permission_classes = (AllowAny, )
    
    def post(self, request):
        # 繫結使用者, 兩種情況1.已經註冊的使用者,2.未註冊的使用者
        # 1.獲取使用者名稱,密碼,weibo_uid
        username = request.data.get('username')
        password = request.data.get('password')
        weibo_uid = request.data.get('weibo_uid')
        
        if not all([username, password, weibo_uid]):
            return Response({"code": 999, "msg": "引數不齊"})
        
        # 判斷是否存在該使用者
        try:
            user = User.objects.get(username=username)
        except Exception as e:
            user = None
        if user:
            # 如果存在表示是已註冊使用者,驗證密碼通過驗證就繫結,返回token繫結成功
            if user.check_password(password):
                ou = OauthUser(uid=weibo_uid, user=user, oauth_type="1")
                ou.save()
                payload = jwt_payload_handler(payload)  # 通過user物件獲取到jwt的payload資訊
                token = jwt_encode_handler(payload)  # 生成token
                data = jwt_response_payload_handler(token, user)
                data["type"] = "0"  #指定未登入成功
                return Response({"code": 0, "msg": "登入成功", "data": data})
        	else:
            	# 密碼錯誤就返回{"code": 999, "msg": "密碼錯誤"}
                return Response({"code": 999, "msg": "密碼錯誤"})
		else:
            # 如果不存在,表示為未註冊使用者
            # 生成新使用者,設定使用者名稱,密碼,然後繫結微博返回token
            user = User(username=username)
            user.set_password(password)
            user.save()
            ou = OauthUser(uid=weibo_uid, user=user, oauth_type="1")
            ou.save()
            payload = jwt_payload_handler(user)
            token = jwt_encode_handler(payload)
            data = jwt_response_payload_handler(token, user)
            data['type'] = "0"  # 指定為登入成功
            return Response({"code": 0, "msg": "登入成功", "data": data})