Forráskód Böngészése

修改企业生成序列号_2

locky 4 éve
szülő
commit
8b0569ae23
1 módosított fájl, 3 hozzáadás és 2 törlés
  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]