使用Django-Profiles和Django-Registration不同的用户配置文件
-
04-10-2019 - |
题
我的models.py:
USER_TYPES = (
('D', 'Demo' ),
('F', 'Free' ),
('P', 'Premium'),
)
class BaseProfile(models.Model):
user = models.OneToOneField(User, primary_key=True)
user_type = models.CharField(max_length=1, blank=True, choices=USER_TYPES)
class DemoProfile(models.Model):
user = models.OneToOneField(User, primary_key=True)
demo = models.CharField(max_length=10, blank=True)
...
class FreeProfile(models.Model):
user = models.OneToOneField(User, primary_key=True)
free = models.CharField(max_length=10, blank=True)
...
class PremiumProfile(models.Model):
user = models.OneToOneField(User, primary_key=True)
premium = models.CharField(max_length=10, blank=True)
...
class ProxyProfile(BaseProfile):
class Meta:
proxy = True
def get_profile(self):
if self.user_type == 'D':
return DemoProfile._default_manager.get(user__id__exact=self.user_id)
elif self.user_type == 'F':
return FreeProfile._default_manager.get(user__id__exact=self.user_id)
else:
return PremiumProfile._default_manager.get(user__id__exact=self.user_id)
我使用基本折叠式将USER_ID映射到特定的user_type。我想将ProxyProfile用作代理,该代理按照模型的配置文件加载user_type,如下所示
我的形式的内容。
class ProfileForm(ModelForm):
...
class Meta:
model = ProxyProfile
exclude = ('user','user_type')
...
使用urls.py中的以下代码向django-profiles提供了profiolform:
urlpatterns += patterns('',
url(r'^profiles/edit/', edit_profile,
{'form_class': ProfileForm},
name='profiles_edit_profile'),
(r'^profiles/',include('profiles.urls')),
)
我还设置在settings.py:
AUTH_PROFILE_MODULE = 'main.ProxyProfile'
在用户注册期间,所有DB数据都正确填充(看起来一切都可以)。我使用传递给Django-Registration的表格注册:
urlpatterns += patterns('',
url(r'^register/$', register,
{'form_class': UserRegistrationForm},
name='registration.views.register'),
(r'', include('registration.urls')),
)
从forms.py:
class UserRegistrationForm(RegistrationFormUniqueEmail, RegistrationFormTermsOfService):
utype = forms.ChoiceField(choices=USER_CHOICES)
def save(self, profile_callback=None):
new_user = RegistrationProfile.objects.create_inactive_user(username=self.cleaned_data['username'],
password.self.cleaned_data['password1'],
email=self.cleaned_data['email'],
)
new_base_profile = BaseProfile(user=new_user, user_type=self.cleaned_data['utype'])
if self.cleaned_data['utype'] == "D":
new_profile = DemoProfile(user=new_user)
if self.cleaned_data['utype'] == "F":
new_profile = FreeProfile(user=new_user)
if self.cleaned_data['utype'] == "P":
new_profile = PremiumProfile(user=new_user)
new_profile.save()
new_base_profile.save()
return new_user
注册阶段正常运行。
我对个人资料编辑/详细信息页面有问题。我的配置文件在ProxyProfile模型中过滤并用作profileForm中的formmodel(我看不到profile特定字段没有渲染到html页面),也许还有其他方法(更像是django方式::)来做到这一点(SELECT SELECT并根据与用户模型相关的User_type字段的渲染配置文件模型。
提前致谢 :)
解决方案
好的,终于我知道了我如何做到这一点:)
在我的Models.py中:
class BaseManager(models.Manager):
def get(self, **kwargs):
self.u = kwargs['user__id__exact']
self.bt = BaseProfile.manager.get(user__id__exact=self.u)
if self.bt.user_type == 'F':
return FreeProfile.objects.get(pk=self.u)
elif self.bt.user_type == 'I':
return PremiumProfile.objects.get(pk=self.u)
else:
return None
class BaseProfile(models.Model):
objects = BaseManager()
manager = UserManager()
user = models.OneToOneField(User, primary_key=True)
user_type = models.CharField(max_length=1, blank=True, choices=USER_TYPES)
class FreeProfile(models.Model):
user = models.OneToOneField(User, primary_key=True)
free = models.CharField(max_length=10, blank=True)
...
class PremiumProfile(models.Model):
user = models.OneToOneField(User, primary_key=True)
premium = models.CharField(max_length=10, blank=True)
...
在Custom Manager -BaseManager中,我通过overtriting get()方法返回配置文件对象,get_profile使用。我必须使用名为“经理”的usermanager来防止分配self.bt时自定义经理的递归呼叫
好的,这是实现我想要的东西的一半,现在我可以使用Django-Profiles应用程序查看与用户相关的不同配置文件。
接下来,我想使用Modelform为用户配置文件准备编辑表单。用户可以拥有不同的配置文件,因此我已经应用了此片段中介绍的魔术: http://djangosnippets.org/snippets/2081/
现在以我的形式。我有:
class FreeForm(forms.ModelForm):
class Meta:
model = FreeProfile
class PremiumForm(forms.ModelForm):
class Meta:
model = PremiumProfile
接下来,每个配置文件的简单模型表单都在ProfileForm中组装:
class ProfileForm(ModelForm):
def __init__(self, *args, **kwargs):
self.user = kwargs['instance'].user
profile_kwargs = kwargs.copy()
profile_kwargs['instance'] = self.user
self.bt = BaseProfile.manager.get(user__id__exact=self.user.id)
if self.bt.user_type == 'F':
self.profile_fields = FreeForm(*args, **profile_kwargs)
elif self.bt.user_type == 'P':
self.profile_fields = PremiumForm(*args, **profile_kwargs)
super(ProfileForm, self).__init__(*args, **kwargs)
self.fields.update(self.profile_fields.fields)
self.initial.update(self.profile_fields.initial)
class Meta:
model = BaseProfile
def save(self):
...
在steratings.py中:
AUTH_PROFILE_MODULE = 'main.BaseProfile'
它的工作就像魅力一样,但我想知道这是否是使用Django-Profiles获得多个不同配置文件的Django的方式?我担心我必须再使用get()在渲染个人资料详细信息或编辑表单之前再使用几次。
但是,经过4天与Django挣扎以完成此操作,终于我今晚可以睡得很好:)
干杯