django - 获取 Django 模型中下一个对象的 id

Question

我有一个函数，它在我的类 Image 中传递一个对象的 id 。我需要模型中下一个对象的id。目前，我正在以效率最低的方式执行此操作，因为我需要获取所有对象来执行此操作。我当前的实现是:

def get_next_id(curr_id):  
  result = []
  Image_list = Image.objects.all()
  total = Image.objects.all().count()
  for i in range(len(Image_list)):
    result.append(Image_list[i].id)
  index_curr = result.index(curr_id)
  if index_curr == total:
    new_index = 0
  else:
    new_index = index_curr + 1
  return Image_list[new_index]

如果有人可以提供更好的方法，或者使这个方法更有效率，我将不胜感激。谢谢。

Answer 1

我建议这样:

def get_next_id(curr_id):
    try:
        ret = Image.objects.filter(id__gt=curr_id).order_by("id")[0:1].get().id
    except Image.DoesNotExist:
        ret = Image.objects.aggregate(Min("id"))['id__min']
    return ret

这不会处理表为空的特殊情况，但是如果表为空，您首先就不应该有有效的 curr_id 。它也不能防止将无意义的值作为 curr_id 传递。

它的作用是获取第一个大于当前 ID 的 id。 [0:1] 切片将从数据库返回的数据限制为第一条记录:实际上是数据库而不是 Python 执行切片。如果没有比当前id大的id，则取最小的id。

回应您关于如何反向执行此操作的评论:

def get_prev_id(curr_id):
    try:
        ret = Image.objects.filter(id__lt=curr_id).order_by("-id")[0:1].get().id
    except Image.DoesNotExist:
        ret = Image.objects.aggregate(Max("id"))['id__max']
    return ret

变化是:

1. 使用id__lt，并按-id排序。

2. 使用 Max 而不是 Min 进行聚合，并使用 id__max 键而不是 id__min > 获取值。