def _get_field_names(allow_primary_key=True, non_editable=False):
def not_in_seq(names):
return lambda name: name not in names
user_class = get_user_model()
fields = user_class._meta.get_fields()
default_field_names = [f.name for f in fields
if (getattr(f, 'serialize', False) or
getattr(f, 'primary_key', False))]
pk_field_names = [f.name for f in fields
if getattr(f, 'primary_key', False)]
hidden_field_names = set(get_user_setting('HIDDEN_FIELDS'))
hidden_field_names = hidden_field_names.union(['last_login', 'password'])
public_field_names = get_user_setting('PUBLIC_FIELDS')
editable_field_names = get_user_setting('EDITABLE_FIELDS')
field_names = (public_field_names if public_field_names is not None
else default_field_names)
if editable_field_names is None:
editable_field_names = field_names
editable_field_names = set(filter(not_in_seq(pk_field_names),
editable_field_names))
field_names = filter(not_in_seq(hidden_field_names), field_names)
if not allow_primary_key:
field_names = filter(not_in_seq(pk_field_names), field_names)
if non_editable:
field_names = filter(not_in_seq(editable_field_names), field_names)
field_names = tuple(field_names)
return field_names
def register(request):
'''
Register new user.
'''
serializer_class = registration_settings.REGISTER_SERIALIZER_CLASS
serializer = serializer_class(data=request.data)
serializer.is_valid(raise_exception=True)
kwargs = {}
if registration_settings.REGISTER_VERIFICATION_ENABLED:
verification_flag_field = get_user_setting('VERIFICATION_FLAG_FIELD')
kwargs[verification_flag_field] = False
email_field = get_user_setting('EMAIL_FIELD')
if (email_field not in serializer.validated_data
or not serializer.validated_data[email_field]):
raise BadRequest("User without email cannot be verified")
user = serializer.save(**kwargs)
output_serializer_class = registration_settings.REGISTER_OUTPUT_SERIALIZER_CLASS # noqa: E501
output_serializer = output_serializer_class(instance=user)
user_data = output_serializer.data
if registration_settings.REGISTER_VERIFICATION_ENABLED:
signer = RegisterSigner({
'user_id': user.pk,
}, request=request)
template_config = (
registration_settings.REGISTER_VERIFICATION_EMAIL_TEMPLATES)
send_verification(user, signer, template_config)
return Response(user_data, status=status.HTTP_201_CREATED)
def register_email(request):
'''
Register new email.
'''
user = request.user
serializer = RegisterEmailSerializer(data=request.data)
serializer.is_valid(raise_exception=True)
email = serializer.validated_data['email']
template_config = (
registration_settings.REGISTER_EMAIL_VERIFICATION_EMAIL_TEMPLATES)
if registration_settings.REGISTER_EMAIL_VERIFICATION_ENABLED:
signer = RegisterEmailSigner({
'user_id': user.pk,
'email': email,
},
request=request)
send_verification(user, signer, template_config, email=email)
else:
email_field = get_user_setting('EMAIL_FIELD')
setattr(user, email_field, email)
user.save()
return get_ok_response('Register email link email sent')
def send_verification(user, params_signer, template_config, email=None):
if email is None:
email_field = get_user_setting('EMAIL_FIELD')
email = getattr(user, email_field)
body_template = get_template(template_config['body'])
subject_template = get_template(template_config['subject'])
from_email = registration_settings.VERIFICATION_FROM_EMAIL
reply_to_email = (registration_settings.VERIFICATION_REPLY_TO_EMAIL
or from_email)
ctx = {
'user': user,
'email': email,
'verification_url': params_signer.get_url(),
'verification_data': params_signer.get_signed_data()
}
subject = subject_template.render(ctx).strip()
body = body_template.render(ctx)
email_msg = EmailMessage(
subject,
body,
from_email,
[email],
reply_to=[reply_to_email],
)
email_msg.send()
def register(request):
'''
Register new user.
'''
serializer_class = registration_settings.REGISTER_SERIALIZER_CLASS
serializer = serializer_class(data=request.data)
serializer.is_valid(raise_exception=True)
kwargs = {}
if registration_settings.REGISTER_VERIFICATION_ENABLED:
verification_flag_field = get_user_setting('VERIFICATION_FLAG_FIELD')
kwargs[verification_flag_field] = False
user = serializer.save(**kwargs)
profile_serializer_class = registration_settings.PROFILE_SERIALIZER_CLASS
profile_serializer = profile_serializer_class(instance=user)
user_data = profile_serializer.data
if registration_settings.REGISTER_VERIFICATION_ENABLED:
signer = RegisterSigner({
'user_id': user.pk,
}, request=request)
template_config = (
registration_settings.REGISTER_VERIFICATION_EMAIL_TEMPLATES)
send_verification(user, signer, template_config)
return Response(user_data, status=status.HTTP_201_CREATED)
def create_verification_notification(
user, params_signer, template_config_data, email=None):
if email is None:
email_field = get_user_setting('EMAIL_FIELD')
email = getattr(user, email_field)
from_email = registration_settings.VERIFICATION_FROM_EMAIL
reply_to_email = (registration_settings.VERIFICATION_REPLY_TO_EMAIL or
from_email)
context = {
'user': user,
'email': email,
'verification_url': params_signer.get_url(),
}
template_config = parse_template_config(template_config_data)
subject = render_to_string(
template_config.subject_template_name, context=context).strip()
text_body = template_config.text_body_processor(
render_to_string(
template_config.text_body_template_name, context=context))
email_msg = EmailMultiAlternatives(
subject, text_body, from_email, [email], reply_to=[reply_to_email])
if template_config.html_body_template_name:
html_body = render_to_string(
template_config.html_body_template_name, context=context)
email_msg.attach_alternative(html_body, 'text/html')
return email_msg
def process_verify_email_data(input_data):
if not registration_settings.REGISTER_EMAIL_VERIFICATION_ENABLED:
raise Http404()
serializer = VerifyEmailSerializer(data=input_data)
serializer.is_valid(raise_exception=True)
data = serializer.validated_data
signer = RegisterEmailSigner(data)
verify_signer_or_bad_request(signer)
email_field = get_user_setting('EMAIL_FIELD')
user = get_user_by_id(data['user_id'])
setattr(user, email_field, data['email'])
user.save()
def process_verify_registration_data(input_data):
if not registration_settings.REGISTER_VERIFICATION_ENABLED:
raise Http404()
serializer = VerifyRegistrationSerializer(data=input_data)
serializer.is_valid(raise_exception=True)
data = serializer.validated_data
signer = RegisterSigner(data)
verify_signer_or_bad_request(signer)
verification_flag_field = get_user_setting('VERIFICATION_FLAG_FIELD')
user = get_user_by_id(data['user_id'], require_verified=False)
setattr(user, verification_flag_field, True)
user.save()
def verify_email(request):
'''
Verify email via signature.
'''
if not registration_settings.REGISTER_EMAIL_VERIFICATION_ENABLED:
raise Http404()
serializer = VerifyEmailSerializer(data=request.data)
serializer.is_valid(raise_exception=True)
data = serializer.data
signer = RegisterEmailSigner(data, request=request)
verify_signer_or_bad_request(signer)
email_field = get_user_setting('EMAIL_FIELD')
user = get_user_by_id(data['user_id'])
setattr(user, email_field, data['email'])
user.save()
return get_ok_response('Email verified successfully')
def verify_registration(request):
'''
Verify registration via signature.
'''
if not registration_settings.REGISTER_VERIFICATION_ENABLED:
raise Http404()
serializer = VerifyRegistrationSerializer(data=request.data)
serializer.is_valid(raise_exception=True)
data = serializer.data
signer = RegisterSigner(data, request=request)
verify_signer_or_bad_request(signer)
verification_flag_field = get_user_setting('VERIFICATION_FLAG_FIELD')
user = get_user_by_id(data['user_id'], require_verified=False)
setattr(user, verification_flag_field, True)
user.save()
return get_ok_response('User verified successfully')
def get_login_fields():
user_class = get_user_model()
return get_user_setting('LOGIN_FIELDS') or [user_class.USERNAME_FIELD]
