50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
from allauth.account.models import EmailAddress
|
|
from allauth.socialaccount.app_settings import QUERY_EMAIL
|
|
from allauth.socialaccount.providers.base import ProviderAccount
|
|
from allauth.socialaccount.providers.oauth2.provider import OAuth2Provider
|
|
|
|
|
|
class AppleProvider(OAuth2Provider):
|
|
id = "apple"
|
|
name = "Apple"
|
|
account_class = ProviderAccount
|
|
|
|
def extract_uid(self, data):
|
|
return str(data["sub"])
|
|
|
|
def extract_common_fields(self, data):
|
|
fields = {"email": data.get("email")}
|
|
|
|
# If the name was provided
|
|
name = data.get("name")
|
|
if name:
|
|
fields["first_name"] = name.get("firstName", "")
|
|
fields["last_name"] = name.get("lastName", "")
|
|
|
|
return fields
|
|
|
|
def extract_email_addresses(self, data):
|
|
ret = []
|
|
email = data.get("email")
|
|
verified = data.get("email_verified")
|
|
if isinstance(verified, str):
|
|
verified = verified.lower() == "true"
|
|
if email:
|
|
ret.append(
|
|
EmailAddress(
|
|
email=email,
|
|
verified=verified,
|
|
primary=True,
|
|
)
|
|
)
|
|
return ret
|
|
|
|
def get_default_scope(self):
|
|
scopes = ["name"]
|
|
if QUERY_EMAIL:
|
|
scopes.append("email")
|
|
return scopes
|
|
|
|
|
|
provider_classes = [AppleProvider]
|