DjangoBB LoFi version

Полная версия: related_name и "does not exist"

Начало » Django » related_name и "does not exist"

andreiru

Июль 27, 2013 18:15:27

Здравствуйте!

Есть модели User и Company

class User(AbstractBaseUser):
    ...
class Company(models.Model):
    user = models.OneToOneField(User, related_name='company', verbose_name=u'Пользователь')
    ...

Если я обращаюсь к Company из User и у пользователя нет компании, то вылетает ошибка “does not exist”

self.request.user.company => “does not exist”

Как с этим бороться, постоянно обертывать в try except или есть вариант удобней ?

FishHook

Июль 27, 2013 20:31:49

self.request.user.company.exists()

andreiru

Июль 28, 2013 17:27:25

FishHook

Выдает AttributeError: ‘Company’ object has no attribute ‘exists’

Это наверное потому что у меня OneToOneField !

andreiru

Июль 28, 2013 17:33:17

Может добавить в модель User:

    @property
    def company_or_none(self):
        if hasattr(self, 'company'):
            return self.company