Создание расширенного профиля пользователя
-
19-09-2019 - |
Вопрос
У меня есть расширенная модель пользователя в Django:
class UserProfile(models.Model):
user = models.ForeignKey(User, unique=True)
#other things in that profile
И signals.py:
from registration.signals import user_registered
from models import UserProfile
from django.contrib.auth.models import User
def createUserProfile(sender, instance, **kwargs):
profile = users.models.UserProfile()
profile.setUser(sender)
profile.save()
user_registered.connect(createUserProfile, sender=User)
Я удостоверился, что сигнал зарегистрировался, имея это в моем __init__.py
:
import signals
Так что это должно создать мне новый пользовательский профиль для каждого пользователя, который регистрируется, верно? Но это не так. Я всегда получаю ошибки «Пользовательский сопостающий запрос не существует», когда я пытаюсь войти в систему, что означает, что запись базы данных нет.
Я должен сказать, что я использую Django-registration, которая предоставляет сигнал user_regystered.
Структура важных приложений для этого заключается в том, что у меня есть одно приложение под названием «Пользователи», там есть: models.py, signals.py, urls.py и views.py (и некоторые другие вещи, которые не должны иметь значения здесь ) Класс пользователя определяется в моделях.py.
Обновлять: Я изменил Signals.py на:
from django.db.models.signals import post_save
from models import UserProfile
from django.contrib.auth.models import User
def create_profile(sender, **kw):
user = kw["instance"]
if kw["created"]:
profile = UserProfile()
profile.user = user
profile.save()
post_save.connect(create_profile, sender=User)
Но теперь я получаю «целостность»:
"Column user_id не уникален"
РЕДАКТИРОВАТЬ 2:
Я нашел это. Похоже, каким -то образом я дважды регистрировал сигнал. Обходной путь для этого описан здесь: http://code.djangoproject.com/wiki/signals#helppost_saveseemstobeemsitiontwiceforeachsave
Я должен был добавить Dispatch_uid, теперь мои сигналы. Py выглядит так и работает:
from django.db.models.signals import post_save
from django.contrib.auth.models import User
from models import UserProfile
from django.db import models
def create_profile(sender, **kw):
user = kw["instance"]
if kw["created"]:
profile = UserProfile(user=user)
profile.save()
post_save.connect(create_profile, sender=User, dispatch_uid="users-profilecreation-signal")
Решение
Вы можете реализовать его с помощью post_save для пользователя:
from django.db.models.signals import post_save
from models import UserProfile
from django.contrib.auth.models import User
def create_profile(sender, **kwargs):
user = kwargs["instance"]
if kwargs["created"]:
profile = users.models.UserProfile()
profile.setUser(sender)
profile.save()
post_save.connect(create_profile, sender=User)
Редактировать:
Другое возможное решение, которое тестируется и работает (я использую его на своем сайте):
from django.db import models
from django.contrib.auth.models import User
from django.db.models.signals import post_save
def create_profile(sender, **kwargs):
user = kwargs["instance"]
if kwargs["created"]:
up = UserProfile(user=user, stuff=1, thing=2)
up.save()
post_save.connect(create_profile, sender=User)
Другие советы
Вы можете создать расширенный профиль, когда он впервые доступен для каждого пользователя:
from django.db import models
from django.contrib.auth.models import User
class UserProfile(models.Model):
user = models.ForeignKey(User, unique=True)
additional_info_field = models.CharField(max_length=50)
User.profile = property(lambda u: UserProfile.objects.get_or_create(user=u)[0])
затем используйте
from django.contrib.auth.models import User
user = User.objects.get(pk=1)
user.profile.additional_info_field
ref: http://www.codekoala.com/blog/2009/quick-django-tip-user-profiles/
Это помогло мне: primary_key = true
class UserProfile(models.Model):
user = models.OneToOneField(User, unique=True, primary_key=True, related_name="user")
phone = models.CharField(('phone'),max_length=30, blank=False, null=True)
user_building = models.ManyToManyField(Building, blank=True)
added_by = models.ForeignKey(User, blank=True, null=True, related_name="added")
Когда вы звоните profile.setUser()
, Я думаю, ты хочешь пройти instance
скорее, чем sender
как параметр.
От Документация сигнала user_redized, sender
относится к User
учебный класс; instance
это фактический объект пользователя, который был зарегистрирован.
Согласно моим последним исследованиям, создание отдельного файла, например, Singals.py, не работает.
Вам лучше подключить «create_profile» к «post_save» в ваших моделях.
Мой окончательный код для вашей ссылки:
# models.py
# Here goes the definition of class UserProfile.
class UserProfile(models.Model):
...
# Use signal to automatically create user profile on user creation.
# Another implementation:
# def create_user_profile(sender, **kwargs):
# user = kwargs["instance"]
# if kwargs["created"]:
# ...
def create_user_profile(sender, instance, created, **kwargs):
"""
:param sender: Class User.
:param instance: The user instance.
"""
if created:
# Seems the following also works:
# UserProfile.objects.create(user=instance)
# TODO: Which is correct or better?
profile = UserProfile(user=instance)
profile.save()
post_save.connect(create_user_profile,
sender=User,
dispatch_uid="users-profilecreation-signal")
Обновление на 2018 год:
Этот вопрос собрал много просмотров, возможно, пришло время для обновления.
Это последняя версия для последней Django.
from django.dispatch import receiver
from django.db.models.signals import post_save
from django.conf import settings
from models import UserProfile
@receiver(post_save, sender=settings.AUTH_USER_MODEL)
def create_user_profile(sender, instance=None, created=False, **kwargs):
if created:
UserProfile.objects.create(user=instance)