Django: รับวัตถุจากฐานข้อมูลหรือ 'ไม่มี' หากไม่มีสิ่งใดที่ตรงกัน

มีฟังก์ชั่น Django ที่จะให้ฉันได้รับวัตถุจากฐานข้อมูลหรือไม่ถ้าไม่มีอะไรที่ตรงกัน

ตอนนี้ฉันกำลังใช้สิ่งที่ชอบ:

foo = Foo.objects.filter(bar=baz)
foo = len(foo) > 0 and foo.get() or None

แต่มันไม่ชัดเจนนักและมันก็ยุ่งเหยิงที่จะมีทุกที่

ใน Django 1.6คุณสามารถใช้first()วิธี Queryset ส่งคืนอ็อบเจ็กต์แรกที่จับคู่โดย queryset หรือ None ถ้าไม่มีอ็อบเจ็กต์ที่ตรงกัน

การใช้งาน:

p = Article.objects.order_by('title', 'pub_date').first()

มีสองวิธีในการดำเนินการนี้

try:
    foo = Foo.objects.get(bar=baz)
except model.DoesNotExist:
    foo = None

หรือคุณสามารถใช้กระดาษห่อหุ้ม:

def get_or_none(model, *args, **kwargs):
    try:
        return model.objects.get(*args, **kwargs)
    except model.DoesNotExist:
        return None

เรียกแบบนี้

foo = get_or_none(Foo, baz=bar)

หากต้องการเพิ่มโค้ดตัวอย่างในคำตอบของ sorki (ฉันจะเพิ่มสิ่งนี้เป็นความคิดเห็น แต่นี่เป็นโพสต์แรกของฉันและฉันไม่มีชื่อเสียงมากพอที่จะแสดงความคิดเห็น) ฉันติดตั้งตัวจัดการแบบกำหนดเอง get_or_none ดังนี้:

from django.db import models

class GetOrNoneManager(models.Manager):
    """Adds get_or_none method to objects
    """
    def get_or_none(self, **kwargs):
        try:
            return self.get(**kwargs)
        except self.model.DoesNotExist:
            return None

class Person(models.Model):
    name = models.CharField(max_length=255)
    objects = GetOrNoneManager()

และตอนนี้ฉันสามารถทำได้:

bob_or_none = Person.objects.get_or_none(name='Bob')

คุณยังสามารถลองใช้ django ที่น่ารำคาญ (มันมีฟังก์ชั่นที่มีประโยชน์อีกอย่าง!)

ติดตั้งด้วย:

pip install django-annoying

from annoying.functions import get_object_or_None
get_object_or_None(Foo, bar=baz)

ให้ฟูผู้จัดการที่กำหนดเอง มันสวยง่าย - เพียงแค่ใส่รหัสของคุณลงในฟังก์ชั่นในการจัดการที่กำหนดเอง, Foo.objects.your_new_func(...)ผู้จัดการฝ่ายการกำหนดเองตั้งอยู่ในรูปแบบของคุณและเรียกมันด้วย

หากคุณต้องการฟังก์ชั่นทั่วไป (เพื่อใช้กับรุ่นใดก็ได้ไม่ใช่แค่นั้นกับผู้จัดการที่กำหนดเอง) ให้เขียนของคุณเองและวางไว้ที่ใดที่หนึ่งบนเส้นทาง python ของคุณและนำเข้าอย่าให้ยุ่งอีกต่อไป

ไม่ว่าจะทำผ่านผู้จัดการหรือฟังก์ชันทั่วไปคุณอาจต้องการจับ 'MultipleObjectsReturned' ในคำสั่ง TRY เนื่องจากฟังก์ชัน get () จะเพิ่มค่านี้หาก kwargs ของคุณดึงข้อมูลมากกว่าหนึ่งวัตถุ

สร้างจากฟังก์ชันทั่วไป:

def get_unique_or_none(model, *args, **kwargs):
    try:
        return model.objects.get(*args, **kwargs)
    except (model.DoesNotExist, model.MultipleObjectsReturned), err:
        return None

และในผู้จัดการ:

class GetUniqueOrNoneManager(models.Manager):
    """Adds get_unique_or_none method to objects
    """
    def get_unique_or_none(self, *args, **kwargs):
        try:
            return self.get(*args, **kwargs)
        except (self.model.DoesNotExist, self.model.MultipleObjectsReturned), err:
            return None

นี่คือรูปแบบของฟังก์ชันตัวช่วยที่ช่วยให้คุณสามารถเลือกส่งผ่านQuerySetอินสแตนซ์ได้ในกรณีที่คุณต้องการรับอ็อบเจ็กต์ที่ไม่ซ้ำกัน (ถ้ามี) จากชุดแบบสอบถามอื่นที่ไม่ใช่ชุดข้อมูลของโมเดลall(เช่นจากชุดย่อยของรายการย่อยที่เป็นของ a อินสแตนซ์หลัก):

def get_unique_or_none(model, queryset=None, *args, **kwargs):
    """
        Performs the query on the specified `queryset`
        (defaulting to the `all` queryset of the `model`'s default manager)
        and returns the unique object matching the given
        keyword arguments.  Returns `None` if no match is found.
        Throws a `model.MultipleObjectsReturned` exception
        if more than one match is found.
    """
    if queryset is None:
        queryset = model.objects.all()
    try:
        return queryset.get(*args, **kwargs)
    except model.DoesNotExist:
        return None

สามารถใช้ได้สองวิธีเช่น:

1. obj = get_unique_or_none(Model, *args, **kwargs) ตามที่กล่าวไว้ทั่วไป
2. obj = get_unique_or_none(Model, parent.children, *args, **kwargs)

ฉันคิดว่าในกรณีส่วนใหญ่คุณสามารถใช้:

foo, created = Foo.objects.get_or_create(bar=baz)

เฉพาะในกรณีที่ไม่สำคัญที่จะเพิ่มรายการใหม่ในตาราง Foo (คอลัมน์อื่น ๆ จะมีค่า None / default)