Различные профили пользователей с django-профилями и django-регистрацией
Мой 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)
Я использую BaseProfile для сопоставления user_id с конкретным user_type. Я хотел использовать ProxyProfile в качестве прокси, который загружает профили в зависимости от user_type в ModelForm, как показано ниже
Содержимое моей формы.
class ProfileForm(ModelForm):
...
class Meta:
model = ProxyProfile
exclude = ('user','user_type')
...
ProfileForm предоставляется django-профилям с использованием следующего кода в urls.py:
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'
При регистрации пользователя все данные БД заполняются правильно (похоже, все в порядке). Я регистрируюсь, используя форму, переданную в django-регистрацию:
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 и использовавшиеся в качестве FormModel в ProfileForm, не отображаются (я не вижу, чтобы определенные для профиля поля не отображались на странице HTML) Возможно, есть другой способ (более похожий на способ Django:)) для этого (выберите и визуализировать модель профиля в зависимости от поля user_type, связанного с моделью пользователя).
Заранее спасибо:)
1 ответ
Хорошо, наконец-то у меня появилась идея, как я могу это сделать:)
В моих 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)
...
В пользовательском менеджере - BaseManager я возвращаю объект профиля, перезаписывая метод get(), используемый get_profile. Я должен использовать UserManager с именем просто "менеджер", чтобы предотвратить рекурсивный вызов пользовательского менеджера при назначении self.bt
Хорошо, это половина пути для достижения того, что я хочу, теперь я могу просматривать различные профили, прикрепленные к пользователям, с помощью приложения django-profile.
Далее я хочу использовать ModelForm для подготовки формы редактирования для профилей пользователей. Пользователи могут иметь разные профили, поэтому я применил магический трюк, представленный в этом фрагменте: http://djangosnippets.org/snippets/2081/
И теперь у меня в файле forms.py:
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):
...
В settings.py:
AUTH_PROFILE_MODULE = 'main.BaseProfile'
И это работает как очарование, но мне интересно, является ли способ Django добиться поддержки нескольких различных профилей с помощью django-профилей? Меня беспокоит, что я должен использовать get() еще несколько раз, прежде чем рендерить детали профиля или редактировать форму.
Но после 4 дней борьбы с Джанго, чтобы наконец-то это сделать, я могу хорошо выспаться сегодня вечером:)
ура