SaleclerkStatusCode.CLERK_NOT_FOUND

Brightcells 7 lat temu
rodzic
commit
703c9d8042
1 zmienionych plików z 7 dodań i 2 usunięć
  1. 7 2
      page/sale_views.py

+ 7 - 2
page/sale_views.py

@@ -8,7 +8,7 @@ from django.shortcuts import render
8 8
 from django_logit import logit
9 9
 from django_response import response
10 10
 
11
-from account.models import SaleclerkInfo
11
+from account.models import SaleclerkInfo, UserInfo
12 12
 from integral.models import SaleclerkIntegralIncomeExpensesInfo, SaleclerkSubmitLogInfo
13 13
 from product.models import ProductCodeSubmitLogInfo, ProductInfo, ProductModelInfo
14 14
 from utils.error.errno_utils import ProductModelStatusCode, ProductStatusCode, SaleclerkStatusCode
@@ -48,7 +48,12 @@ def clerk_sale_submit_api(request):
48 48
     file_path = request.POST.get('file_path', '')
49 49
 
50 50
     try:
51
-        clerk = SaleclerkInfo.objects.get(clerk_id=user_id, status=True)
51
+        user = UserInfo.objects.get(user_id=user_id, status=True)
52
+    except UserInfo.DoesNotExist:
53
+        return response(SaleclerkStatusCode.CLERK_NOT_FOUND)
54
+
55
+    try:
56
+        clerk = SaleclerkInfo.objects.get(unionid=user.unionid, status=True)
52 57
     except SaleclerkInfo.DoesNotExist:
53 58
         return response(SaleclerkStatusCode.CLERK_NOT_FOUND)
54 59