locky 4 лет назад
Родитель
Сommit
8b0569ae23
1 измененных файлов с 3 добавлено и 2 удалено
  1. 3 2
      Controller/CompanyController.py

+ 3 - 2
Controller/CompanyController.py

@@ -136,8 +136,8 @@ class CompanyView(View):
         if not all([id, quantity]):
             return response.json(444)
 
-        company = CompanyModel.objects.get(id=id)
-        if not company.exists():
+        company_qs = CompanyModel.objects.filter(id=id)
+        if not company_qs.exists():
             return response.json(444)
 
         p2p_sum_Serial = SerialNumberModel.objects.filter(p2p=p2p).count()
@@ -147,6 +147,7 @@ class CompanyView(View):
             return response.json(10041)
 
         try:
+            company = company_qs[0]
             start_1 = p2p_sum_Serial_company
             end_1 = int(start_1) + int(quantity)
             serial_qs = SerialNumberModel.objects.filter(p2p=p2p)[start_1:end_1]