from pprint import pprint

from django.db import connections
from django.http import JsonResponse

from systextil.queries.deposito.total_modelo import totais_modelos_depositos


def estoque_depositos_modelo(request, modelo):
    cursor = connections['so'].cursor()
    data = {
        'modelo': modelo,
    }

    try:
        totais = totais_modelos_depositos(
            cursor, ('101', '102', '103', '122', '231'))
        total_est = totais[modelo]
        data.update({
            'result': 'OK',
            'total_est': total_est,
        })

    except Exception as e:
        data.update({
            'result': 'ERR',
            'descricao_erro': 'Erro ao buscar estoque nos depósitos',
        })

    return JsonResponse(data, safe=False)

@@ -413,16 +413,33 @@
 '231'))%0A
+        try:%0A    
         
@@ -465,16 +465,67 @@
 modelo%5D%0A
+        except KeyError:%0A            total_est = 0%0A
         

se totais não calculou o modelo é porque a quantidade é zero