Python后端日常操作之在Django中「强行」使用MVVM设计模式

扫盲

首先带大家了解一下什么是MVVM模式：

什么是MVVM？MVVM是Model-View-ViewModel的缩写。

MVVM是MVC的增强版，实质上和MVC没有本质区别，只是代码的位置变动而已

从名字上看，MVVM比MVC架构中多了一个ViewModel，没错，就是这个ViewModel，他是MVVM相对于MVC改进的核心思想。在开发过程中，由于需求的变更或添加，项目的复杂度越来越高，代码量越来越大，此时我们会发现MVC维护起来有些吃力，首先被人吐槽的最多的就是MVC的简写变成了Massive-View-Controller（意为沉重的Controller）

由于Controller主要用来处理各种逻辑和数据转化，复杂业务逻辑界面的Controller非常庞大，维护困难，所以有人想到把Controller的数据和逻辑处理部分从中抽离出来，用一个专门的对象去管理，这个对象就是ViewModel，是Model和Controller之间的一座桥梁。当人们去尝试这种方式时，发现Controller中的代码变得非常少，变得易于测试和维护，只需要Controller和ViewModel做数据绑定即可，这也就催生了MVVM的热潮。

引言

大家都知道Django是MVT模式，Model就是View和Template/Interface之间的数据传递的「信使」，这种模式存在一个问题，就是当我们的业务不断扩大之后需要在接口返回出model里不包含的数据时该怎么办？例如一个商店，我们要动态计算它距离我们当前位置有多远，那么这个距离肯定是不包含在Model里面的，数据库也不可能实时存储这类数据。

那么这时候我们就需要在Model上，再加上一层ViewModel，顾名思义，视图模型，是用来在视图里传递和处理数据的模型。

简单实现

在App包下面创建一个view_models文件，内容如下：

from rest_framework.request import Request
from core.models import Store
from core.serializers import StoreSerializer

class StoreViewModel:
    def __init__(self, store: Store, distance=0.0, request: Request = None):
        self.store = store
        self.distance = distance
        self.request = request

    @property
    def serialize_data(self):
        return StoreSerializer(self.store, context={
            'distance': self.distance,
            'request': self.request,
        }).data

上面的代码定义了一个商店的视图模型，构造方法中除了我们的Model对象，还有Model中不包括的distance参数，还有一个request用来传递请求的context，这个在Drf中是很重要的，如果不处理好context的传递，会导致Drf在序列化一些文件或者链接类字段的时候丢失前半部分的域名。

接下来看看serialize_data这个属性，它做的工作很简单，就是把Model对象传给序列化器，然后在context中存入我们的额外参数distance和request。

再来看看序列化器要如何改造以适应ViewModel模型。

class StoreSerializer(serializers.ModelSerializer):
    distance = serializers.SerializerMethodField()

    class Meta:
        model = models.Store
        fields = '__all__'

    def get_distance(self, obj: models.Store):
        return self.context.get('distance', 0)

这里可以看到序列化器中，我是把额外的distance字段处理成SerializerMethodField，然后在get_distance方法中实现，通过self.context属性可以获取到我们在ViewModel中传入的context，这样就实现额外参数的序列化。

最后我们在看看在View，也就是控制器，看看如何将ViewModel和原本的分页，权限各类功能结合在一起。

class StoreViewSet(viewsets.ReadOnlyModelViewSet):
    """商家相关功能"""
    serializer_class = serializers.StoreSerializer
    queryset = models.Store.objects.all()

    @action(detail=False)
    def location(self, request):
        """根据地理位置筛选商家"""
        city = request.GET.get('city')
        town = request.GET.get('town')
        lat = request.GET.get('lat')
        lng = request.GET.get('lng')

        # 根据城市、区镇筛选商店
        queryset = models.Store.objects.filter(city=city, town=town)

        # 调用接口计算所有商店距离当前位置的距离，该接口返回ViewModel
        store_view_models = tencent_map.stores_distance(from_lat=lat, from_lng=lng, queryset=queryset, request=request)

        # 对ViewModelSet进行排序，按照距离
        store_view_models.sort(key=lambda store_view_model: store_view_model.distance)

        # 使用列表生成器，对每个ViewModel进行序列化
        stores_data = [store_vm.serialize_data for store_vm in store_view_models]

        # 对结果数据进行分页
        page = self.paginate_queryset(stores_data)
        return self.get_paginated_response(page)

上面的代码目前在开发环境运行良好，我已经写了详细的注释了，可以看到用ViewModel模式是可以和原本的ViewSet很好的结合在一起的，包括分页这些功能都可以正常使用。

小结

标题中我用了「强行」这个词，就是觉得我这样实现好像很不优雅，但又不至于hack，因为这个需求很简单，只要实现了就行，我也还没有去搜索其他的解决方案，在本文中提出了我的ViewModel与Django结合解决方案，如果大家有更好的解决方案可以留言一起探讨~

欢迎交流

我整理了一系列的技术文章和资料，在公众号「程序设计实验室」后台回复 linux、flutter、c#、netcore、android、java、python 等可获取相关技术文章和资料，同时有任何问题都可以在公众号后台留言~