Различные профили пользователей с 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, как показано ниже
содержание моего forms.py:
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-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 и используются как FormModel в ProfileForm не отображаются (я не вижу, что определенные поля профиля не отображаются на HTML-странице) Может быть, есть какой-то другой способ (больше похожий на Django way :)) этот (выбор и отображение модели профиля в зависимости от поля 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 с именем просто "manager", чтобы предотвратить рекурсивный вызов пользовательского менеджера при назначении self.bt
хорошо, это половина пути к достижению того, что я хочу, теперь я могу просматривать различные профили, прикрепленные к пользователям с помощью django-profiles приложение.
затем я хочу использовать ModelForm для подготовки формы редактирования для профилей пользователей. Пользователи могут иметь разные профили, поэтому я применил волшебный трюк, представленный в этом фрагменте:http://djangosnippets.org/snippets/2081/
и теперь в моем forms.py у меня:
class FreeForm(forms.ModelForm):
class Meta:
model = FreeProfile
class PremiumForm(forms.ModelForm):
class Meta:
model = PremiumProfile
далее, простые модельные формы для каждого профиля собраны в форме профиля:
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):
...
In settings.py:
AUTH_PROFILE_MODULE = 'main.BaseProfile'
и это работает как шарм, но я интересно, если это способ Django для достижения поддержки нескольких различных профилей с помощью django-профилей? Меня беспокоит, что мне нужно использовать get() еще несколько раз, прежде чем я буду отображать детали профиля или редактировать форму.
но после 4 дней борьбы с Django, чтобы сделать это, наконец, я могу хорошо спать сегодня вечером:)
Ура