我可以在 ModelSerializer 的创建和更新中调用一个函数吗

Question

我可以在创建和更新中调用相同的函数而不覆盖它吗？

您好，我在这方面还很陌生，正在尝试使用 DRY 原理 来实现我的代码。 所以我在 ModelSerializer's 中为 create_update_password 调用一个函数，同时为此创建或更改密码，我必须 覆盖创建和更新 在 ModelSerializer 和 create_update_password 中。除了调用这个函数之外，我没有在其中任何一个中做任何额外的功能。那么有什么方法可以让我在创建和更新中调用这个函数而不覆盖它。 如果有人可以提出任何建议，我真的很喜欢它。下面是代码示例 谢谢你

    def create_update_password(self, instance, validated_data):
        password = validated_data["password"]
        instance.set_password(password)
        instance.save()

class UserModelSerializer(ModelSerializer):

    def create(self, validated_data):
        instance = super(UserModelSerializer, self).create(validated_data)
        self.create_update_password(instance, validated_data)
        return instance

    def update(self, instance, validated_data):
        instance = super(UserModelSerializer, self).update(instance, validated_data)
        self.create_update_password(instance, validated_data)
        return instance

Answer 1

您可以使用装饰器将相同的逻辑应用于两种方法，因此：

def create_update_password(f):
    @wraps(f)
    def g(self, *args, **kwargs):
        validated_data = kwargs.get('validated_data') or args[-1]
        result = f(self, *args, **kwargs)
        result.set_password(validated_data['password'])
        return result

然后将其与@method_decorator 一起使用：

from django.utils.decorators import method_decorator

@method_decorator(create_update_password, 'create')
@method_decorator(create_update_password, 'update')
class UserModelSerializer(ModelSerializer):
    # …