Django: Lấy một đối tượng ở dạng DB hoặc 'Không có' nếu không có gì phù hợp

Có bất kỳ hàm Django nào cho phép tôi lấy một đối tượng trong cơ sở dữ liệu hay không nếu không có gì phù hợp?

Ngay bây giờ tôi đang sử dụng một cái gì đó như:

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

Nhưng điều đó không rõ ràng lắm, và thật lộn xộn khi có ở khắp mọi nơi.

Trong Django 1.6, bạn có thể sử dụng first()phương thức Queryset. Nó trả về đối tượng đầu tiên được đối sánh bởi bộ truy vấn hoặc Không có nếu không có đối tượng phù hợp.

Sử dụng:

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

Có hai cách để làm điều này;

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

Hoặc bạn có thể sử dụng một trình bao bọc:

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

Gọi nó như thế này

foo = get_or_none(Foo, baz=bar)

Để thêm một số mã mẫu vào câu trả lời của sorki (tôi muốn thêm điều này làm nhận xét, nhưng đây là bài đăng đầu tiên của tôi và tôi không có đủ danh tiếng để để lại nhận xét), tôi đã triển khai trình quản lý tùy chỉnh get_or_none như vậy:

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()

Và bây giờ tôi có thể làm điều này:

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

Bạn cũng có thể thử sử dụng django gây phiền nhiễu (nó có một chức năng hữu ích khác!)

cài đặt nó với:

pip install django-annoying

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

Cung cấp cho Foo trình quản lý tùy chỉnh của nó . Nó khá dễ dàng - chỉ cần đặt mã của bạn vào chức năng trong trình quản lý tùy chỉnh, đặt trình quản lý tùy chỉnh trong mô hình của bạn và gọi nó bằng Foo.objects.your_new_func(...).

Nếu bạn cần hàm chung (để sử dụng nó trên bất kỳ mô hình nào, không chỉ với trình quản lý tùy chỉnh), hãy viết hàm của riêng bạn và đặt nó ở đâu đó trên đường dẫn python của bạn và nhập, không lộn xộn nữa.

Cho dù thực hiện nó thông qua trình quản lý hay hàm chung, bạn cũng có thể muốn bắt 'MultipleObjectsReturned' trong câu lệnh TRY, vì hàm get () sẽ nâng cao điều này nếu kwargs của bạn truy xuất nhiều hơn một đối tượng.

Xây dựng trên chức năng chung:

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

và trong trình quản lý:

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

Đây là một biến thể của hàm trợ giúp cho phép bạn tùy chọn chuyển trong một QuerySetphiên bản, trong trường hợp bạn muốn lấy đối tượng duy nhất (nếu có) từ một bộ truy vấn khác với bộ truy vấn allđối tượng của mô hình (ví dụ: từ một tập hợp con các mục con thuộc một cá thể mẹ):

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

Điều này có thể được sử dụng theo hai cách, ví dụ:

1. obj = get_unique_or_none(Model, *args, **kwargs) như đã thảo luận trước
2. obj = get_unique_or_none(Model, parent.children, *args, **kwargs)

Tôi nghĩ rằng trong hầu hết các trường hợp, bạn chỉ có thể sử dụng:

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

Chỉ khi không quá quan trọng thì mục nhập mới sẽ được thêm vào bảng Foo (các cột khác sẽ có giá trị Không có / mặc định)