Доступ к объекту со ссылкой на себя в свойстве/методе Django
У меня есть модель пользователя
class Account(AbstractBaseUser):
first_name = models.CharField(max_length = 255, unique = False)
last_name = models.CharField(max_length = 255, unique = False)
email = models.EmailField(verbose_name = "email", max_length = 60, unique = True)
is_sponsor = models.BooleanField(blank=True, default=True, help_text='Check if the user is a sponsor', verbose_name='Is this user a sponsor?')
sponsored_by = models.ForeignKey('self',blank=True,null=True, on_delete=models.SET_NULL, default=None,limit_choices_to={'is_sponsor': True}),
is_active = models.BooleanField(default=True)
is_staff = models.BooleanField(default=False)
is_superuser = models.BooleanField(default=False)
USERNAME_FIELD = 'email'
REQUIRED_FIELDS = [
'first_name',
'last_name',
]
Я создал свойство, для которого мне нужно использовать 'sponsored_by'
@property
def sponsored_by_test(self):
if self.sponsored_by is not None:
if self.sponsored_by.first_name=="test":
return True
return False
Когда я тестирую его,
x = Account.objects.get(first_name="test")
print(x.sponsored_by)
>> (<django.db.models.fields.related.ForeignKey>,)
Следовательно, я не могу получить доступ к переменной 'sponsored_by' в свойстве (или где-либо еще).
Я что-то упустил?
Вам нужно вызвать свойство x.sponsored_by_test
, а не x.sponsored_by
.
Так просто:
In [11]: b = Account.objects.create(first_name='test2', last_name='2', email='123@qq.com2')
In [12]: x.sponsored_by=b
In [13]: x.save()
In [14]: print(x.sponsored_by)
123@qq.com2
для свойства
@property
def sponsored_by_test(self):
return getattr(self.sponsored_by, 'first_name', None) == "test"