Việc cần làm khi truy vấn Django trả về không có gì? Nó cung cấp cho tôi lỗi

Question

to_friend = User.objects.filter(username=friend_q)[0:1] 

Nếu 'friend_q' không nằm trong User.username ... nó sẽ gây ra lỗi. Chiến thuật được khuyến nghị là gì?

Cảm ơn bạn

Answer 1

Nếu friend_q không phải là người dùng có trong cơ sở dữ liệu, to_friend sẽ bằng danh sách trống.

>>> from django.contrib.auth.models import User 
>>> User.objects.filter(username='does-not-exist') 
[] 

Tuy nhiên, nó tốt hơn để sử dụng phương thức get() để tra cứu một mục cụ thể:

>>> User.objects.get(username='does-exist') 
<User: does-exist> 
>>> User.objects.get(username='does-not-exist') 
Traceback (most recent call last): 
    File "<console>", line 1, in <module> 
    File "/usr/lib/python2.5/django/db/models/manager.py", line 120, in get 
    File "/usr/lib/python2.5/django/db/models/query.py", line 305, in get 
DoesNotExist: User matching query does not exist. 

Bây giờ bạn có thể bắt các ngoại lệ DoesNotExist và có những hành động thích hợp.

try: 
    to_friend = User.objects.get(username=friend_q) 
except User.DoesNotExist: 
    # do something, raise error, ... 

Answer 2

Lý do bạn gặp phải lỗi là vì bạn đang cố gắng đánh giá đối tượng Truy vấn khi không có gì trong đó. [0: 1] ở cuối là cố gắng lấy mục đầu tiên ra khỏi danh sách trống. Nếu bạn chia nhỏ biểu thức một chút như dưới đây, bạn có thể kiểm tra danh sách trống trước khi lấy một phần tử ra khỏi nó.

to_friends = User.objects.filter(username=friend_q) 
if to_friends: 
    to_friend = to_friends[0] 

Answer 3

1. FWIW, Username là duy nhất. Vì vậy, U có thể sử dụng

   to_friend = User.objects.get(username=friend_q) 
   

2. Nếu bạn muốn nâng cao một 404 khi người dùng không tồn tại, bạn có thể là tốt,

   from django.shortcuts import get_object_or_404 
   to_friend = get_object_or_404(User,username=friend_q) 
   

3. Để ngăn chặn lỗi, bạn có thể chỉ cần đặt nó trong các thử trừ khối, rất pythonic.

   try: 
       to_friend = User.objects.get(username=friend_q) 
   except User.DoesNotExist: 
       to_friend = None 
   

4. Vì đây là một yêu cầu chung, bạn nên xem xét việc xác định get_user_or_none trên UserManager, và người quản lý khác mà sẽ yêu cầu None trở lại, tương tự như get_object_or_404. Điều này đã được xem xét cho lõi django, nhưng không được bao gồm, cho những gì bao giờ là lý do. Tuy nhiên, là một chức năng 4 dòng tiện dụng cho các mô hình cần None quay lại.