Tivix/django-rest-auth
1
1
Fork 0
mirror of https://github.com/Tivix/django-rest-auth.git synced 2025-07-15 18:32:18 +03:00
Code Issues Packages Projects Releases Wiki Activity

Merge c47b641c62 into cdd04aa9be

Browse Source
This commit is contained in:
Lee H 2022-04-03 16:06:27 +09:00 committed by GitHub
commit 4e609cdf17
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
rest_auth/registration/serializers.py
View File

@ -35,6 +35,7 @@ class SocialAccountSerializer(serializers.ModelSerializer):
class SocialLoginSerializer(serializers.Serializer): class SocialLoginSerializer(serializers.Serializer):
access_token = serializers.CharField(required=False, allow_blank=True) access_token = serializers.CharField(required=False, allow_blank=True)
refresh_token = serializers.CharField(required=False, allow_blank=True)
code = serializers.CharField(required=False, allow_blank=True) code = serializers.CharField(required=False, allow_blank=True)
def _get_request(self): def _get_request(self):
@ -80,6 +81,8 @@ class SocialLoginSerializer(serializers.Serializer):
# Case 1: We received the access_token # Case 1: We received the access_token
if attrs.get('access_token'): if attrs.get('access_token'):
access_token = attrs.get('access_token') access_token = attrs.get('access_token')
# Also optional refresh token
refresh_token = attrs.get('refresh_token')
# Case 2: We received the authorization code # Case 2: We received the authorization code
elif attrs.get('code'): elif attrs.get('code'):
@ -110,12 +113,17 @@ class SocialLoginSerializer(serializers.Serializer):
) )
token = client.get_access_token(code) token = client.get_access_token(code)
access_token = token['access_token'] access_token = token['access_token']
refresh_token = token.get('refresh_token')
else: else:
raise serializers.ValidationError( raise serializers.ValidationError(
_("Incorrect input. access_token or code is required.")) _("Incorrect input. access_token or code is required."))
social_token = adapter.parse_token({'access_token': access_token}) token_dict = {'access_token': access_token}
if refresh_token:
# Also pass the refresh_token if there is one
token_dict.update({'refresh_token': refresh_token})
social_token = adapter.parse_token(token_dict)
social_token.app = app social_token.app = app
try: try: