9

我正在使用 django-registration，我正在尝试连接到它的信号以自动创建一个 UserProfile。

信号定义：

from django.dispatch import Signal

# A new user has registered.
user_registered = Signal(providing_args=["user", "request"])

通过 django-registration 发送信号：

    def register(self, request, **kwargs):
    """
    Create and immediately log in a new user.

    """
    username, email, password = kwargs['username'], kwargs['email'], kwargs['password1']
    User.objects.create_user(username, email, password)

    # authenticate() always has to be called before login(), and
    # will return the user we just created.
    new_user = authenticate(username=username, password=password)
    login(request, new_user)
    signals.user_registered.send(sender=self.__class__,
                                 user=new_user,
                                 request=request)
    return new_user

我的信号连接：

from registration.signals import *
from core.models import UserProfile
from django.contrib.auth.models import User

def createUserProfile(sender, instance, **kwargs):
    UserProfile.objects.get_or_create(user=instance)

user_registered.connect(createUserProfile, sender=User)

不用说没有创建 UserProfile。我在这里想念什么？

非常感谢！

编辑：我将我的 connect() 及其相应的方法移到了 model.py 中，但仍然没有运气。

新代码：

from django.db import models

from django.contrib import auth
from django.contrib.auth import login
from core.forms import AuthForm
from registration.signals import *
from django.contrib.auth.models import User


# Create your models here.

class UserProfile(models.Model) :
    user = models.ForeignKey(User, unique=True)

    def __unicode__(self):
        return self.user.username


def createUserProfile(sender, instance, **kwargs):
    print "creating profile"
    UserProfile.objects.get_or_create(user=instance)

user_registered.connect(createUserProfile, sender=User)

我正在使用 Pycharm 进行调试，一开始我在 user_registered.connect() 上的断点被命中。所以我假设 connect() 正在正确注册。但是，我仍然没有看到 createUserProfile 正在运行。还有什么我想念的吗？

谢谢！

回答：呵呵。我的连接和接收器代码错误。正确代码：

def createUserProfile(sender, user, request, **kwargs):
UserProfile.objects.get_or_create(user=user)

user_registered.connect(createUserProfile)

我在 django-registration 中阅读了 signals.py 后才意识到

4

2 回答 2

7

您需要在服务器启动时导入的模块中注册（连接）您的信号。您所在的文件user_registered.connect(createUserProfile, sender=User)很可能不会在启动时导入。来自 django 文档：

您可以将信号处理和注册代码放在您喜欢的任何地方。但是，您需要确保尽早导入它所在的模块，以便在需要发送任何信号之前注册信号处理。这使您的应用程序的 models.py 成为注册信号处理程序的好地方。

http://docs.djangoproject.com/en/dev/topics/signals/#connecting-receiver-functions

因此，您的自定义应用程序的 models.py 将是一个好地方（或任何其他在服务器启动时肯定导入的模块）。

于 2011-04-14T05:11:08.183 回答
3

Torsten 是对的：另一种方法是使用文档中所述的装饰器：

    from registration.signals import user_registered
    # ...
    @receiver(user_registered)
    def your_function_name_here(sender, user, request, **kwargs):
            # your code here
            pass

我喜欢这种方式，因为它紧凑且可读。

于 2012-10-30T10:23:27.523 回答