web-dev-qa-db-ja.com

DjangoでOneToOneFieldがNoneかどうかを確認します

このような2つのモデルがあります。

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

ただし、type1またはtype2のプロファイルを持たないユーザーの場合、そのようなコードを実行すると次のエラーが生成されます。

Type1Profile matching query does not exist.

ユーザーが持っているプロファイルのタイプを確認するにはどうすればよいですか？

ありがとう

pythondjango-modelsone-to-one
77
John Bright

（OneToOne）リレーションが存在するかどうかを確認するには、hasattr関数を使用できます。

if hasattr(request.user, 'type1profile'):
    # do something
Elif hasattr(request.user, 'type2profile'):
    # do something else
else:
    # do something else
82
joctee

特定のモデルでNonenessのモデルの対応するフィールドをテストするだけで、null可能な1対1の関係がnullであるかどうかを確認できますが、only1対1の関係が発生するモデルでテストする場合。たとえば、これら2つのクラスが与えられた場合…

_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's restaurant here.
_

hasattrを使用するというjocteeの提案は実際には機能しますが、hasattrがall例外を抑制するため、実際には偶然にしか機能しません（ DoesNotExist）を含める必要がありますが、AttributeErrorsだけではありません。 Piet Delportが指摘したように、この動作は次のチケットごとにPython 3.2で実際に修正されました： http://bugs.python.org/issue9666 。さらに—そして意見を述べるリスクがある-上記のtry/except構造は、hasattrを使用しているときのDjango= FUDを作成し、悪い習慣を広める可能性がある初心者向けの問題を曇らせることができます。

42
Joshua Pokotilow

joctee's answer が好きです。これはとても簡単だからです。

if hasattr(request.user, 'type1profile'):
    # do something
Elif hasattr(request.user, 'type2profile'):
    # do something else
else:
    # do something else

他のコメント者は、特定のバージョンのPythonまたはDjangoでは動作しない可能性があるという懸念を提起しましたが、 Django documentation オプションの1つとして：

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.

私は Joshua に同意します。例外をキャッチすることで何が起きているかが明確になりますが、それは私にとっては厄介に思えます。おそらくこれは合理的な妥協案ですか？

>>> 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()
14
Don Kirkby

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!

これをジェネリック関数として使用して、元のクラス（ここではプロファイルクラス）と関連インスタンス（ここではrequest.user）を指定すると、逆OneToOneインスタンスを取得できると思います。

9
Geradeausanwalt

使用する select_related！

>>> user = User.objects.select_related('type1profile').get(pk=111)
>>> user.type1profile
None
3
ivan133

Has_attrの組み合わせを使用していますが、なしです。

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
0
FreeWorlder