22 lines
698 B
Python
22 lines
698 B
Python
|
"""Utils for tests in the People core application"""
|
||
|
|
from rest_framework_simplejwt.tokens import AccessToken
|
||
|
|
|
||
|
|
|
||
|
|
class OIDCToken(AccessToken):
|
||
|
|
"""Set payload on token from user/contact/email"""
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def for_user(cls, user):
|
||
|
|
token = super().for_user(user)
|
||
|
|
identity = user.identities.filter(is_main=True).first()
|
||
|
|
token["first_name"] = (
|
||
|
|
user.profile_contact.short_name if user.profile_contact else "David"
|
||
|
|
)
|
||
|
|
token["last_name"] = (
|
||
|
|
" ".join(user.profile_contact.full_name.split()[1:])
|
||
|
|
if user.profile_contact
|
||
|
|
else "Bowman"
|
||
|
|
)
|
||
|
|
token["email"] = identity.email
|
||
|
|
return token
|