门店管理

apps/agent/serializers.py

@@ -29,6 +29,7 @@ class StoreComboboxSerializer(serializers.ModelSerializer):
 
 class StoreSerializer(serializers.ModelSerializer):
     create_user_text = serializers.CharField(source='create_user.name', read_only=True)
+    status_text = serializers.CharField(source='get_status_display', read_only=True)
 
     class Meta:
         model = Store

apps/agent/views.py

@@ -2,9 +2,12 @@
 from utils.custom_modelviewset import CustomModelViewSet
 from django.db.models import Q
 from django.utils import timezone
+from rest_framework.decorators import action
 from rest_framework.views import APIView
+from django.conf import settings
 
 from utils import response_ok, response_error
+from utils.exceptions import CustomError
 from .models import Agent,Store
 from .serializers import AgentSerializer,StoreSerializer
 from .filters import AgentFilter,StoreFilter
@@ -51,6 +54,27 @@ class StoreViewSet(CustomModelViewSet):
         super(StoreViewSet, self).perform_destroy(instance)
         return response_ok()
 
+    @action(methods=['post'], detail=False)
+    def check(self, request, pk):
+        # 审核
+        report_status = request.POST.get('report_status')
+        try:
+            instance = Store.objects.filter(id=pk).first()
+            if not instance:
+                raise CustomError('当前门店信息有误！')
+            if instance.report_status > settings.NOT_CHECKED :
+                raise CustomError('当前企业已审核！')
+
+            instance.check_user = self.request.user
+            instance.report_status = report_status
+            instance.check_time = timezone.now()
+            instance.save()
+        except CustomError as e:
+            return response_error(e.get_error_msg())
+        except Exception as e:
+            return response_error(str(e))
+        return response_ok('审核完成！')
+
 class AgentDictView(APIView):
     permission_classes = [isLogin, ]
 

apps/customer/models.py

@@ -60,12 +60,4 @@ class ReportCustomer(models.Model):
     #         self.check_time = timezone.now()
     #         self.save()
     #     # if self.report_status == Customer.CHECKED:
-    #     #     # 撞单
-
-
-    def check_order(self, user):
-        # 审核
-        self.check_user = user
-        self.check_time = timezone.now()
-        self.check_status = self.PASS
-        self.save()
+    #     #     # 撞单