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]