在django中生成唯一的引用代码

时间:2014-04-05 05:31:57

标签: python django

我是django的新手,我希望为每个使用注册的用户生成唯一引荐代码。此推荐代码将通过电子邮件发送给用户。我是使用userena,我不知道如何使用 userena 。这些是关于该问题的一些链接,但不清楚如何使用 userena

来实现

https://github.com/byteweaver/django-referral

https://pypi.python.org/pypi/django-referral

http://paltman.com/2012/08/17/how-to-easily-add-referrals-to-a-website/

。如果有人对此问题有明确的想法,那么请与我分享。  干杯

1 个答案:

答案 0 :(得分:1)

您必须为保留引荐代码创建新模型。下面给出了模型语法。

import uuid
import base64
from django.db import models
from django.contrib.auth.models import User
from django.utils.translation import ugettext_lazy as _
from django.db.models.signals import post_save
from userena.models import UserenaLanguageBaseProfile
from userena.utils import user_model_label 


class Profile(UserenaLanguageBaseProfile):
    """ Default profile """

    user = models.OneToOneField(User,
                                unique=True,
                                verbose_name=_('user'),
                                related_name='profile')
    referral_code = models.CharField(max_length=300, blank=True, null=True)


    def get_absolute_url(self):
        return u'/profile/show/%d' % self.id
    def generate_verification_code(self):
        # Generate user's verification code
        # TODO: Move this to the model
        return base64.urlsafe_b64encode(uuid.uuid1().bytes.encode("base64").rstrip())[:25]

    def save(self, *args, **kwargs):
        """
        If this is a new user, generate code.
        Otherwise leave as is
        """
        if not self.pk:
            self.referral_code = self.generate_verification_code()
        elif not self.verification_code:
            self.referral_code = self.generate_verification_code()

        return super(Profile, self).save(*args, **kwargs)

在你的models.py

的最后一行添加这些行
def create_user_profile(sender, instance, created, **kwargs):
    if created:
        profile, created = Profile.objects.get_or_create(user=instance)

post_save.connect(create_user_profile, sender=User)

以上提及行将在创建用户时创建您的用户个人资料。

希望这对你有用!!!