Django无效uuid查询抛出500错误,如何简化实现自动返回404响应?
方案1:自定义模型管理器(全局生效,最省心)
直接给Document模型自定义查询管理器,重写get方法自动处理UUID格式校验,所有调用Document.objects.get(uuid=xxx)的位置都会自动生效,无需修改原有查询代码:
import uuid from django.db import models from django.http import Http404 class DocumentManager(models.Manager): def get(self, *args, **kwargs): # 当查询参数包含uuid时先校验格式 if 'uuid' in kwargs: try: uuid.UUID(kwargs['uuid']) except ValueError: raise Http404("Invalid uuid") return super().get(*args, **kwargs) # 在你的Document模型中替换默认管理器 class Document(models.Model): uuid = models.UUIDField(unique=True, editable=False) # 其他原有字段... objects = DocumentManager()
修改后你原来的查询代码直接写Document.objects.get(uuid=object_uuid)就可以,格式错误自动返回404。
方案2:仅Admin场景优化(无需修改模型)
你当前重写change_view的代码存在冗余查询,Admin自带的get_object方法已经会自动处理对象不存在的逻辑,你可以直接重写该方法添加UUID校验即可:
import uuid from django.http import Http404 from django.contrib import admin class DocumentAdmin(admin.ModelAdmin): def get_object(self, request, object_id, from_field=None): # 先校验UUID格式 try: uuid.UUID(object_id) except ValueError: raise Http404("Invalid uuid") return super().get_object(request, object_id, from_field=from_field) def change_view(self, request, object_id, form_url="", extra_context=None): extra_context = extra_context or {} # 直接复用get_object返回的对象,无需重复查询 obj = self.get_object(request, object_id) extra_context["my_field"] = obj.my_field return super().change_view(request, object_id, form_url, extra_context=extra_context)
这种方案不会影响模型在其他场景的查询逻辑,仅对Admin后台生效。
方案3:轻量封装工具函数
如果你不想修改模型和Admin的基础代码,也可以封装通用查询函数统一处理校验逻辑:
import uuid from django.http import Http404 def get_by_uuid_or_404(model, uuid_val): try: uuid.UUID(uuid_val) return model.objects.get(uuid=uuid_val) except (ValueError, model.DoesNotExist): raise Http404
使用时直接调用即可:
x = get_by_uuid_or_404(Document, object_uuid)
内容的提问来源于stack exchange,提问作者Hippolyte BRINGER
相关产品推荐
相关产品推荐

