如何从扩展用户模型中检索数据

时间:2017-07-15 12:53:32

标签: django django-models django-forms django-templates django-views

我正在尝试通过django创建一个简单的注册/登录页面。 我使用了UserCreationForm并使用了一个模型UserProfile来扩展用户模型。 我希望在用户登录后检索从我的主页上的表单,即部门发布的数据。 我是django的新手,所以简短的解释将不胜感激。 提前致谢

forms.py

from django import forms
from django.contrib.auth.forms import UserCreationForm 
from django.contrib.auth.models import User
from mysite.core.models import UserProfile
from django.db import models

class SignUpForm(UserCreationForm):
    first_name = forms.CharField(max_length=30, required=False, help_text='Optional.')
    department = forms.CharField(max_length=30)
    last_name = forms.CharField(max_length=30, required=False, help_text='Optional.')
    email = forms.EmailField(max_length=254, help_text='Required. Inform a valid email address.')

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

    def save(self, commit=True):
        # Save the provided password in hashed format
        user = super(SignUpForm, self).save(commit=False)
        user_profile = UserProfile(user=user, department=self.cleaned_data['department'])
        user.save()
        user_profile.save()
        return user, user_profile

views.py

from django.contrib.auth.decorators import login_required
from django.contrib.auth import login, authenticate
from django.shortcuts import render, redirect
from mysite.core.forms import SignUpForm

@login_required
def home(request):
    return render(request, 'home.html')

def signup(request):
    if request.method == 'POST':
        form = SignUpForm(request.POST)
        if form.is_valid():
            user,user_profile = form.save(commit=False)
            username = user.cleaned_data.get('username')
            raw_password = user.cleaned_data.get('password1')
            user = authenticate(username=username, password=raw_password)
            login(request, user)
            return redirect('home')
    else:
        form = SignUpForm()
    return render(request, 'signup.html', {'form': form})

models.py

from django.db import models
from django.contrib.auth.models import User
class UserProfile(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE,unique=True)
    department = models.CharField(max_length=500, blank=True)

home.html in templates:

{% extends 'base.html' %}
{% block content %}
<h2>Welcome, <small>{{ user.username }}</small>!</h2>
<p>Your email address: {{ user.email }}</p>
<p>Department: {{ user_profile.department }}</p>
{% endblock %}

我可以打印用户名和电子邮件,但部门空白。

1 个答案:

答案 0 :(得分:3)

首先,你的关系错了。 User和UserProfile之间存在一对一的关系;用户只能拥有一个配置文件,配置文件只能属于一个用户。现在你的方式,用户可以拥有许多配置文件,这是没有意义的。

您应该使用OneToOneField替换ForeignKey。

完成后,您将可以通过以下关系访问个人资料数据:user.userprofile.department,依此类推。