如何在Django QuerySet中执行Python代码？实现字节单位转换

首先明确：Django QuerySet的annotate操作是在数据库层面执行的，无法直接传入自定义Python函数——因为QuerySet最终会被编译成SQL语句，数据库无法识别Python代码。不过你可以通过两种方式实现字节到可读单位的转换：

方案一：先聚合求和，再在Python层面转换单位 #

这是最常用、最易维护的方式：先让数据库完成字节数的求和聚合，再用Python代码遍历结果转换单位。

修改你的代码如下：

def get_context_data(self, *args, **kwargs):
    context = super().get_context_data(*args, **kwargs)
 
    # 第一步：数据库层面聚合字节总和与图片数量
    user_data = DataVolume.objects.values('user')
        .order_by('user')
        .annotate(
            imgs=Sum('imgs'),
            total_bytes=Sum('user_bytes')
        )
    
    # 定义字节转换函数
    def convert_bytes(size):
        units = ['B', 'KB', 'MB', 'GB', 'TB', 'PB', 'EB', 'ZB', 'YB']
        index = 0
        # 循环转换单位，直到数值小于1024或达到最大单位
        while size >= 1024 and index < len(units) - 1:
            size /= 1024
            index += 1
        return f"{size:.2f} {units[index]}"
    
    # 遍历结果，添加转换后的size字段
    for item in user_data:
        item['size'] = convert_bytes(item['total_bytes'])
    
    context['user'] = user_data
    return context

优缺点 #

• 优点：Python代码逻辑清晰，容易调试和扩展，不依赖数据库特性；
• 缺点：如果数据量极大，遍历所有结果会占用一定内存，但绝大多数业务场景下完全够用。

方案二：用数据库函数在SQL层面完成转换 #

如果你的数据量特别大，不想在Python中遍历处理，可以用Django的Case/When等数据库函数，在SQL层面实现单位转换。

示例代码：

from django.db.models import Sum, Case, When, F, Value, FloatField, CharField

def get_context_data(self, *args, **kwargs):
    context = super().get_context_data(*args, **kwargs)
 
    # 数据库层面完成聚合、单位判断与数值转换
    user_data = DataVolume.objects.values('user')
        .order_by('user')
        .annotate(
            imgs=Sum('imgs'),
            total_bytes=Sum('user_bytes'),
            # 判断应该使用的单位索引（对应YB到B的顺序）
            unit_index=Case(
                When(total_bytes__gte=1024**8, then=Value(8)),
                When(total_bytes__gte=1024**7, then=Value(7)),
                When(total_bytes__gte=1024**6, then=Value(6)),
                When(total_bytes__gte=1024**5, then=Value(5)),
                When(total_bytes__gte=1024**4, then=Value(4)),
                When(total_bytes__gte=1024**3, then=Value(3)),
                When(total_bytes__gte=1024**2, then=Value(2)),
                When(total_bytes__gte=1024, then=Value(1)),
                default=Value(0),
                output_field=FloatField()
            ),
            # 计算转换后的数值
            size_value=F('total_bytes') / (1024 ** F('unit_index')),
            # 匹配对应的单位字符串
            unit=Case(
                When(unit_index=8, then=Value('YB')),
                When(unit_index=7, then=Value('ZB')),
                When(unit_index=6, then=Value('EB')),
                When(unit_index=5, then=Value('PB')),
                When(unit_index=4, then=Value('TB')),
                When(unit_index=3, then=Value('GB')),
                When(unit_index=2, then=Value('MB')),
                When(unit_index=1, then=Value('KB')),
                default=Value('B'),
                output_field=CharField()
            )
        )
    
    # 合并数值与单位为一个字段（也可以直接在模板中拼接）
    for item in user_data:
        item['size'] = f"{item['size_value']:.2f} {item['unit']}"
    
    context['user'] = user_data
    return context

优缺点 #

• 优点：转换逻辑在数据库执行，返回结果直接可用，减少Python端的内存占用；
• 缺点：SQL层面的逻辑写法繁琐，依赖数据库支持的数学运算，不同数据库（如MySQL、PostgreSQL）可能需要调整语法，精度控制也不如Python灵活。

内容的提问来源于stack exchange，提问作者user1463152