Question

我正在建立一个注册页面。我使用默认的User模型和一个名为UserInfo的新模型来获取其他用户详细信息。我需要在两个模型之间建立一个OneToOne关系。为了构建注册页面，我为每个模型创建了一个表单，并将它们放入相同的注册视图中。

我的问题是，如何将UserInfo中的用户（OneToOneField）设置为在创建UserInfo的同一视图中创建的用户？如果您查看下面的views.py，在我保存用户表单（signup_form）后立即尝试了一些类似的事情。

user2.user = User.objects.get（pk = pk） - 似乎无法正常工作
user2.user = request.user - 不起作用，因为在这种情况下request.user是匿名用户
user2.user = user - 无法正常工作

models.py：

class UserInfo(models.Model):

    TITLE = (
        ('Salesperson', 'Salesperson'),
        ('Sales Representative', 'Sales Representative'),
        ('Broker', 'Broker'),
        ('Broker of Record', 'Broker of Record'),
    )

    user = models.OneToOneField(User, on_delete=models.CASCADE)
    preferred_email = models.EmailField()
    office_phone_number = models.CharField(max_length=10)
    brokerage_of_agent = models.CharField(max_length=50)
    agent_title = models.CharField(max_length=20, choices=TITLE)

    def __str__(self):
        return self.preferred_email

forms.py：

class SignupForm(UserCreationForm):
    email = forms.EmailField(max_length=200, help_text='Required')

    class Meta:
        model = User
        fields = ('username', 'email', 'password1', 'password2')

class UserInfoForm(forms.ModelForm):

    class Meta:
        model = UserInfo
        fields = ['preferred_email', 'office_phone_number', 'brokerage_of_agent', 'agent_title']

views.py：

def signup(request):
    if request.user.is_authenticated:
        return HttpResponseRedirect('../dashboard/')
    if request.method == 'POST':
        signup_form = SignupForm(request.POST)
        basic_info_form = UserInfoForm(request.POST)
        while (True):  # The while loop is used to get access to 'break' for the @gmail Email check
            if signup_form.is_valid():
                if not signup_form.cleaned_data['email'].endswith('@gmail.com'):
                    print('You must register with your @gmail.com Email')
                    break
                user = signup_form.save(commit=False)
                user.is_active = False
                user.save()

                ### Look here!  I need to write something here I think, but I'm not sure what to write
                user2 = basic_info_form.save(commit=False)
                user2.user = User.objects.get(pk=pk) 
                user2 = user2.save()

                current_site = get_current_site(request)
                message = render_to_string('acc_active_email.html', {
                    'user':user, 
                    'domain':current_site.domain,
                    'uid': urlsafe_base64_encode(force_bytes(user.pk)),
                    'token': account_activation_token.make_token(user),
                })
                mail_subject = 'Activate your blog account.'
                to_email = signup_form.cleaned_data.get('email')
                email = EmailMessage(mail_subject, message, to=[to_email])
                email.send()
                return HttpResponse('Please confirm your email address to complete the registration')
    else:
        signup_form = SignupForm()
        basic_info_form = UserInfoForm()

    return render(request, 'signup.html', {'signup_form': signup_form, 'basic_info_form': basic_info_form})

HTML模板：

{% extends "base.html" %}
{% load static %}

{% block content %}
<h2>Sign up</h2>
<form method="post">
    {% csrf_token %}
    {% for field in signup_form %}
    <p>
        {{ field.label_tag }}<br>
        {{ field }}
        {% if field.help_text %}
        <small style="display: none">{{ field.help_text }}</small>
        {% endif %}
        {% for error in field.errors %}
    <p style="color: red">{{ error }}</p>
        {% endfor %}
    {% endfor %}

    {% for field in basic_info_form %}
    <p>
        {{ field.label_tag }}<br>
        {{ field }}
        {% if field.help_text %}
        <small style="display: none">{{ field.help_text }}</small>
        {% endif %}
        {% for error in field.errors %}
    <p style="color: red">{{ error }}</p>
        {% endfor %}
    {% endfor %}


    <button type="submit">Sign up</button>
</form>
{% endblock %}

Answer 1

user2.user = user实际上是正确的答案，但是一些笨拙的while循环搞砸了，所以我认为它不起作用。

在同一视图中创建2个模型（表单）之间的OneToOne关系

1 个答案: