У меня есть две модели:
class Type1Profile(models.Model):
user = models.OneToOneField(User, unique=True)
...
class Type2Profile(models.Model):
user = models.OneToOneField(User, unique=True)
...
Мне нужно что-то сделать, если у пользователя есть профиль Type1 или Type2:
if request.user.type1profile != None:
# do something
elif request.user.type2profile != None:
# do something else
else:
# do something else
Но для пользователей, у которых нет профилей типа 1 или типа2, выполнение такого кода вызывает следующую ошибку:
Type1Profile matching query does not exist.
Как проверить тип профиля, который пользователь имеет?
Спасибо
Чтобы проверить, существует ли отношение (OneToOne) или нет, вы можете использовать функцию hasattr:
if hasattr(request.user, 'type1profile'):
# do something
elif hasattr(request.user, 'type2profile'):
# do something else
else:
# do something else
Можно проверить, является ли нулевое отношение "один-к-одному" нулевым для конкретной модели просто путем тестирования соответствующего поля в модели для None ness, но только если вы протестируете на модели, где один-на-один возникает одно отношение. Например, учитывая эти два класса...
class Place(models.Model):
name = models.CharField(max_length=50)
address = models.CharField(max_length=80)
class Restaurant(models.Model): # The class where the one-to-one originates
place = models.OneToOneField(Place, blank=True, null=True)
serves_hot_dogs = models.BooleanField()
serves_pizza = models.BooleanField()
... чтобы убедиться, что Restaurant имеет Place, мы можем использовать следующий код:
>>> r = Restaurant(serves_hot_dogs=True, serves_pizza=False)
>>> r.save()
>>> if r.place is None:
>>> print "Restaurant has no place!"
Restaurant has no place!
Чтобы убедиться, что Place имеет Restaurant, важно понять, что ссылка на свойство Restaurant на экземпляр Place вызывает исключение Restaurant.DoesNotExist, если нет соответствующего ресторана. Это происходит из-за того, что Django выполняет внутренний поиск с помощью QuerySet.get(). Например:
>>> p2 = Place(name='Ace Hardware', address='1013 N. Ashland')
>>> p2.save()
>>> p2.restaurant
Traceback (most recent call last):
...
DoesNotExist: Restaurant matching query does not exist.
В этом случае преобладает бритва Occam, и лучший способ сделать определение о том, имеет ли Place Restautrant, стандартную конструкцию try/except, как описано .
>>> try:
>>> restaurant = p2.restaurant
>>> except Restaurant.DoesNotExist:
>>> print "Place has no restaurant!"
>>> else:
>>> # Do something with p2 restaurant here.
Хотя использование joctee для использования hasattr работает на практике, оно действительно работает только случайно, поскольку hasattr подавляет все исключения (включая DoesNotExist), а не только AttributeError s, как и должно было. Как отметил Piet Delport, это поведение было фактически исправлено в Python 3.2 за следующий билет: http://bugs.python.org/issue9666. Более того - и с риском озвучивания - я считаю, что приведенная выше конструкция try/except более характерна для работы Django, а использование hasattr может помешать проблеме для новичков, что может создать FUD и распространять вредные привычки.
Мне нравится Joctee ответ, потому что это так просто.
if hasattr(request.user, 'type1profile'):
# do something
elif hasattr(request.user, 'type2profile'):
# do something else
else:
# do something else
Другие комментаторы высказали опасения, что он может не работать с определенными версиями Python или Django, но в документации Django этот метод показан как один из вариантов:
Вы также можете использовать hasattr, чтобы избежать необходимости перехвата исключений:
>>> hasattr(p2, 'restaurant')
False
Конечно, документация также показывает технику ловли исключений:
У p2 нет ассоциированного ресторана:
>>> from django.core.exceptions import ObjectDoesNotExist
>>> try:
>>> p2.restaurant
>>> except ObjectDoesNotExist:
>>> print("There is no restaurant here.")
There is no restaurant here.
Я согласен с Джошуа в том, что, поймав это исключение, проясняю, что происходит, но мне это кажется беспорядочным. Возможно, это разумный компромисс?
>>> print(Restaurant.objects.filter(place=p2).first())
None
Это просто запрос объектов Restaurant по месту. Возвращает None если в этом месте нет ресторана.
Здесь исполняемый фрагмент для вас, чтобы играть с параметрами. Если у вас установлены Python, Django и SQLite3, он должен просто запуститься. Я протестировал его с Python 2.7, Python 3.4, Django 1.9.2 и SQLite3 3.8.2.
# Tested with Django 1.9.2
import sys
import django
from django.apps import apps
from django.apps.config import AppConfig
from django.conf import settings
from django.core.exceptions import ObjectDoesNotExist
from django.db import connections, models, DEFAULT_DB_ALIAS
from django.db.models.base import ModelBase
NAME = 'udjango'
def main():
setup()
class Place(models.Model):
name = models.CharField(max_length=50)
address = models.CharField(max_length=80)
def __str__(self): # __unicode__ on Python 2
return "%s the place" % self.name
class Restaurant(models.Model):
place = models.OneToOneField(Place, primary_key=True)
serves_hot_dogs = models.BooleanField(default=False)
serves_pizza = models.BooleanField(default=False)
def __str__(self): # __unicode__ on Python 2
return "%s the restaurant" % self.place.name
class Waiter(models.Model):
restaurant = models.ForeignKey(Restaurant)
name = models.CharField(max_length=50)
def __str__(self): # __unicode__ on Python 2
return "%s the waiter at %s" % (self.name, self.restaurant)
syncdb(Place)
syncdb(Restaurant)
syncdb(Waiter)
p1 = Place(name='Demon Dogs', address='944 W. Fullerton')
p1.save()
p2 = Place(name='Ace Hardware', address='1013 N. Ashland')
p2.save()
r = Restaurant(place=p1, serves_hot_dogs=True, serves_pizza=False)
r.save()
print(r.place)
print(p1.restaurant)
# Option 1: try/except
try:
print(p2.restaurant)
except ObjectDoesNotExist:
print("There is no restaurant here.")
# Option 2: getattr and hasattr
print(getattr(p2, 'restaurant', 'There is no restaurant attribute.'))
if hasattr(p2, 'restaurant'):
print('Restaurant found by hasattr().')
else:
print('Restaurant not found by hasattr().')
# Option 3: a query
print(Restaurant.objects.filter(place=p2).first())
def setup():
DB_FILE = NAME + '.db'
with open(DB_FILE, 'w'):
pass # wipe the database
settings.configure(
DEBUG=True,
DATABASES={
DEFAULT_DB_ALIAS: {
'ENGINE': 'django.db.backends.sqlite3',
'NAME': DB_FILE}},
LOGGING={'version': 1,
'disable_existing_loggers': False,
'formatters': {
'debug': {
'format': '%(asctime)s[%(levelname)s]'
'%(name)s.%(funcName)s(): %(message)s',
'datefmt': '%Y-%m-%d %H:%M:%S'}},
'handlers': {
'console': {
'level': 'DEBUG',
'class': 'logging.StreamHandler',
'formatter': 'debug'}},
'root': {
'handlers': ['console'],
'level': 'WARN'},
'loggers': {
"django.db": {"level": "WARN"}}})
app_config = AppConfig(NAME, sys.modules['__main__'])
apps.populate([app_config])
django.setup()
original_new_func = ModelBase.__new__
@staticmethod
def patched_new(cls, name, bases, attrs):
if 'Meta' not in attrs:
class Meta:
app_label = NAME
attrs['Meta'] = Meta
return original_new_func(cls, name, bases, attrs)
ModelBase.__new__ = patched_new
def syncdb(model):
""" Standard syncdb expects models to be in reliable locations.
Based on https://github.com/django/django/blob/1.9.3
/django/core/management/commands/migrate.py#L285
"""
connection = connections[DEFAULT_DB_ALIAS]
with connection.schema_editor() as editor:
editor.create_model(model)
main()
Как насчет использования try/except блоков?
def get_profile_or_none(user, profile_cls):
try:
profile = getattr(user, profile_cls.__name__.lower())
except profile_cls.DoesNotExist:
profile = None
return profile
Тогда используйте вот так!
u = request.user
if get_profile_or_none(u, Type1Profile) is not None:
# do something
elif get_profile_or_none(u, Type2Profile) is not None:
# do something else
else:
# d'oh!
Я полагаю, вы могли бы использовать это как универсальную функцию для получения любого обратного экземпляра OneToOne, учитывая исходный класс (здесь: ваши классы профиля) и связанный с ним экземпляр (здесь: request.user).
Используйте select_related!
>>> user = User.objects.select_related('type1profile').get(pk=111)
>>> user.type1profile
None
Я использую комбинацию has_attr и None:
class DriverLocation(models.Model):
driver = models.OneToOneField(Driver, related_name='location', on_delete=models.CASCADE)
class Driver(models.Model):
pass
@property
def has_location(self):
return not hasattr(self, "location") or self.location is None