Оптимизация цикла Django Queryset Many to Many for loop с помощью prefetch_related

У меня есть функция в представлении, которая выполняет 64 запроса за 4,5 секунды, поэтому мне нужно оптимизировать ее для лучшей производительности.

Проблема в том, что prefetch_related не работает так, как я ожидал, поэтому у меня есть новый набор запросов для каждой итерации цикла for.

Здесь представлена модель.

введите описание изображения здесь

class Carrera(models.Model):
'''Carreras de cada sede'''
codigo = models.CharField(max_length=5, unique=True)
nombre = models.CharField(max_length=300)

def __str__(self):
    return f"{self.codigo} - {self.nombre}"


class EstandarProducto(models.Model):
''''''
costo_unitario_uf = models.DecimalField(max_digits=20, decimal_places=5, default=0)
cantidad = models.IntegerField(default=0)
total_uf = models.DecimalField(max_digits=20, decimal_places=5, default=0)
recinto = models.ForeignKey(Recinto, related_name='estandares_producto', on_delete=models.CASCADE)
producto = models.ForeignKey(
    Producto, related_name='estandares_producto', on_delete=models.PROTECT)
proveedor = models.ForeignKey(
    Proveedor, related_name='estandares_producto', on_delete=models.CASCADE, blank=True, null=True)
carreras = models.ManyToManyField(Carrera, related_name="estandares_productos", blank=True)

И вот мои взгляды:

class getEstandarPorRecinto(APIView):

@query_debugger
def get(self, request, format=None):
    id_recinto = request.GET.get('recinto')
    id_sede = request.GET.get('sede')
    estandares = EstandarProducto.objects.select_related('recinto','producto','proveedor').prefetch_related(Prefetch('carreras')).filter(recinto=id_recinto)
    inventario = InventarioProducto.objects.select_related('inventario').filter(inventario__sede=id_sede)
    num_salas = SedeRecinto.objects.select_related('sede').filter(sede=id_sede, recinto=id_recinto)[0].numero_salas
    productos_en_inventario = inventario.values() 
    lista_enviar = []

    for estandarProducto in estandares:
        carreras = []
        for carrera in estandarProducto.carreras.values():
            print("Codigo: " ,carrera.get("codigo"))
            carreras.append(carrera.get("codigo"))
        cantidad_inventario = 0
        año_compra = ""
        id_inventario_producto = ""
        print("Carreras: ",carreras)
        for producto in productos_en_inventario: 
            if producto.get("id") == estandarProducto.producto.id: 
                cantidad_inventario = producto.get("cantidad") 
                año_compra = producto.get("ultimo_año_reposicion") 
                id_inventario_producto = producto.get("id") 

        
        estandar_json = {
            "id_producto": estandarProducto.producto.id, 
            "id_estandar_producto": estandarProducto.id, 
            "codigo": estandarProducto.producto.codigo, 
            "carreras": carreras, 
            "categoria": estandarProducto.producto.categoria, 
            "nombre": estandarProducto.producto.nombre, 
            "descripcion": estandarProducto.producto.descripcion, 
            "cantidad_esperada": estandarProducto.cantidad,
            "proveedor_id": estandarProducto.proveedor.id, 
            "proveedor": estandarProducto.proveedor.nombre, 
            "frecuencia_reposicion": estandarProducto.producto.frecuencia_reposicion, 
            "inventario_producto_id": id_inventario_producto, 
            "costo_un_uf": estandarProducto.costo_unitario_uf, 
            "total_uf": estandarProducto.total_uf, 
            "tipo_presupuesto": estandarProducto.producto.tipo_presupuesto, 
            "inventario": cantidad_inventario, 
            "ultimo_año_compra": año_compra, 
            "numero_salas": num_salas, 
            "a_comprar": estandarProducto.cantidad*num_salas 
        }
        
        lista_enviar.append(estandar_json)
    data = lista_enviar
    return Response(data, status.HTTP_200_OK)

Проблема заключается в следующем:

estandares = EstandarProducto.objects.select_related('recinto','producto','proveedor').prefetch_related(Prefetch('carreras')).filter(recinto=id_recinto)

и

for carrera in estandarProducto.carreras.values():
        print("Codigo: " ,carrera.get("codigo"))
        carreras.append(carrera.get("codigo"))

как я могу получить все carreras.codigo одним запросом, чтобы затем добавить в список carreras[]?

Вы можете использовать values_list и установить flat True для вывода списка

например: 

carreras  = estandares.values_list("carreras__codigo", flat=True)

Как объясняется в docs:

все последующие цепочки методов, которые подразумевают другой запрос к базе данных, будут игнорировать ранее кэшированные результаты и получать данные, используя свежий запрос к базе данных.

Это означает, что вызов values() по сути делает вашу предварительную выборку непригодной для использования и заставит ваш цикл for обращаться к базе данных, чтобы удовлетворить вызов values(). В этом случае просто используйте all() и получите codigo из самого экземпляра следующим образом:

    for estandarProducto in estandares:
        carreras = []
        for carrera in estandarProducto.carreras.all():
            carreras.append(carrera.codigo)

Но, похоже, вам нужны только сами codigo, поэтому ответ @pouria farhadi будет работать хорошо.

Вернуться на верх