Python装饰器优化API接口：DRY原则实践

Python装饰器是一种强大的编程工具，它允许我们在不改变原有函数定义的前提下，为函数添加额外的行为或修改其行为。在DRY（Don't Repeat Yourself，即“不要重复自己”）原则的指导下，装饰器能够帮助我们避免代码冗余，提高代码的可维护性和复用性。 在Web开发中，如Django框架，函数通常负责处理请求并返回响应。例如，`handle_request`函数接收请求并返回一个简单的字符串响应。当我们需要创建API方法，比如注册接口，可能会有多个类似的逻辑，比如检查请求类型、处理错误和返回JSON格式的数据。 原始的`register`函数示例中，为了满足返回JSON响应和处理GET请求错误，作者不得不在每个API方法中重复编写这部分代码，这就违反了DRY原则。通过使用装饰器，我们可以将这些通用逻辑提取出来，作为装饰器来应用到多个API方法上。 装饰器的本质是一个函数，它接受一个函数作为输入，并返回一个新的函数。当这个新函数被调用时，装饰器会在调用前后执行自定义的操作。例如，我们可以创建一个装饰器`json_response`，它会自动将函数的结果转化为JSON格式，以及检查请求方法是否为POST，并在非POST请求时返回错误码。 下面是一个简化的装饰器实现： ```python def json_response(func): def wrapper(request, *args, kwargs): if request.method != 'POST': return {"error": "this method only accepts posts!"}, 405 try: response_data = func(request, *args, kwargs) except Exception as e: response_data = {"error": str(e)} response = HttpResponse(json.dumps(response_data)) if 'error' in response_data: response.status_code = 500 return response return wrapper @register = json_response ``` 在这个例子中，`json_response`装饰器接收一个函数`func`作为输入，然后定义了一个内部函数`wrapper`。这个`wrapper`函数首先检查请求方法，然后调用`func`并处理可能的异常。最后，根据结果构建并返回一个JSON格式的HTTP响应，保持了DRY原则。 通过这种方式，我们可以在不改变原有`register`函数定义的情况下，轻松地将其转变为一个符合要求的API方法，同时避免了代码的冗余。这种设计模式使得代码更加模块化和易于维护，提升了整体代码质量。