1. 程式人生 > 實用技巧 >NO.166 禪道的自定義功能:匯出欄位和後臺的自定義

NO.166 禪道的自定義功能:匯出欄位和後臺的自定義

1.微博繫結使用者介面

1.1oauth/urls.py中新增路由

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

1.2oauth/views.py中新增試圖函式

class OauthWeiboBindUser(APIView):
permission_classes = (AllowAny,)
def post(self, request):
# 繫結使用者, 1. 已註冊使用者, 2. 未註冊使用者
# 1.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': '引數不全'})
# 0.判斷是否存在此使用者
try:
user = User.objects.get(username=username)
except Exception as e:
user = None
# 1. 已註冊使用者
if user:
# 1.2 , 如果存在就驗證 密碼, 驗證通過,就繫結, 返回token,登入成功
if user.check_password(password):
ou = OauthUser(uid=weibo_uid, user=user, oauth_type='1')
ou.save()
payload = jwt_payload_handler(user) # 通過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:
return Response({'code': 999, 'msg': '密碼錯誤'})
else:
# 2. 未註冊使用者
# 2.1 生成新使用者, 設定使用者名稱密碼, 儲存, 然後繫結, 返回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)
date['type'] = '0'  # 指定為登入成功
return Response({'code':0,'msg':'登入成功','date':date})