在 Django Rest Framework 中缺少密钥时防止错误答案

【问题标题】：Prevent error when missing key in Django Rest Framework在 Django Rest Framework 中缺少密钥时防止错误
【发布时间】：2022-12-05 08:22:16
【问题描述】：

我正在构建一个 API 来处理我的其他网站的数据库和电子邮件营销数据。

我有一个端点用于从数据库和 ActiveCampaign 中删除用户，以防止错误地发出任何删除请求，我正在检查请求正文中是否有 key:value 对，如果有 'delete': true，则继续，如果没有，我想返回一条带有状态代码的错误消息，让我（或我将来可能包含在项目中的其他人）知道什么是错误的。

我的是：在检查是否有一个名为'delete' 的密钥时，我得到一个错误并且我的程序停止工作。

我想知道是否有一种方法可以在经过一些检查后只“做事”，但不会破坏我的程序，如果发生了意想不到的事情，它会向请求来源发回错误。

这是我正在尝试工作的类/函数：

class Leads(APIView):
    @staticmethod
    def delete(request):
        if request.data["delete"]:
            delete_from_db = Lead.objects.filter(email=request.data["email"])
            lead = LeadHelper(email=request.data["email"] if request.data["email"] else "")
            lead.delete_from_activecampaign()
            return Response([delete_from_db], status=status.HTTP_200_OK)
        else:
            payload = {
                "message": "Denied because 'delete': 'true' was not found in request, did you sent this by error?"
            }
            return Response(payload, status=status.HTTP_401_UNAUTHORIZED)

我的主要问题是，如果没有 'delete' 密钥，它甚至无法运行 else 语句

这是错误：

Internal Server Error: /leads/create/
Traceback (most recent call last):
  File "/Users/matheus/dev/clients/guardian_api/venv/lib/python3.11/site-packages/django/core/handlers/exception.py", line 55, in inner
    response = get_response(request)
               ^^^^^^^^^^^^^^^^^^^^^
  File "/Users/matheus/dev/clients/guardian_api/venv/lib/python3.11/site-packages/django/core/handlers/base.py", line 191, in _get_response
    response = wrapped_callback(request, *callback_args, **callback_kwargs)
               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/matheus/dev/clients/guardian_api/venv/lib/python3.11/site-packages/django/views/decorators/csrf.py", line 54, in wrapped_view
    return view_func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/matheus/dev/clients/guardian_api/venv/lib/python3.11/site-packages/django/views/generic/base.py", line 103, in view
    return self.dispatch(request, *args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/matheus/dev/clients/guardian_api/venv/lib/python3.11/site-packages/rest_framework/views.py", line 509, in dispatch
    response = self.handle_exception(exc)
               ^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/matheus/dev/clients/guardian_api/venv/lib/python3.11/site-packages/rest_framework/views.py", line 469, in handle_exception
    self.raise_uncaught_exception(exc)
  File "/Users/matheus/dev/clients/guardian_api/venv/lib/python3.11/site-packages/rest_framework/views.py", line 480, in raise_uncaught_exception
    raise exc
  File "/Users/matheus/dev/clients/guardian_api/venv/lib/python3.11/site-packages/rest_framework/views.py", line 506, in dispatch
    response = handler(request, *args, **kwargs)
               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/matheus/dev/clients/guardian_api/leads/views.py", line 47, in delete
    if request.data["delete"]:
       ~~~~~~~~~~~~^^^^^^^^^^
KeyError: 'delete'
[04/Dec/2022 20:59:47] "DELETE /leads/create/ HTTP/1.1" 500 87663

【问题讨论】：

data 只是一本字典 iirc。您可以像检查任何其他字典一样检查键是否存在，或使用 get。

标签： python django if-statement django-rest-framework

【解决方案1】：

.get()函数在key丢失的情况下不会抛出异常；它会返回None。

if request.data.get("delete"):

或者您可以明确检查密钥是否存在：

if 'delete' in request.data:

【讨论】：