|
@@ -13,10 +13,9 @@ from apps.log.models import BizLog
|
|
from .models import GeneralAgent, Agent, Store
|
|
from .models import GeneralAgent, Agent, Store
|
|
from .serializers import GeneralAgentSerializer, AgentSerializer, StoreSerializer
|
|
from .serializers import GeneralAgentSerializer, AgentSerializer, StoreSerializer
|
|
from .filters import GeneralAgentFilter, AgentFilter, StoreFilter
|
|
from .filters import GeneralAgentFilter, AgentFilter, StoreFilter
|
|
-from utils.permission import isLogin, permission_required
|
|
|
|
|
|
+from utils.permission import isLogin, check_permission, permission_required
|
|
from apps.agent.serializers import AgentComboboxSerializer, StoreComboboxSerializer, GeneralAgentComboboxSerializer
|
|
from apps.agent.serializers import AgentComboboxSerializer, StoreComboboxSerializer, GeneralAgentComboboxSerializer
|
|
|
|
|
|
-
|
|
|
|
class GeneralAgentViewSet(CustomModelViewSet):
|
|
class GeneralAgentViewSet(CustomModelViewSet):
|
|
permission_classes = [isLogin, ]
|
|
permission_classes = [isLogin, ]
|
|
queryset = GeneralAgent.objects.filter()
|
|
queryset = GeneralAgent.objects.filter()
|
|
@@ -24,7 +23,9 @@ class GeneralAgentViewSet(CustomModelViewSet):
|
|
|
|
|
|
@permission_required('agent.view_agent')
|
|
@permission_required('agent.view_agent')
|
|
def filter_queryset(self, queryset):
|
|
def filter_queryset(self, queryset):
|
|
- queryset = queryset.filter(Q(id=self.request.user.general_agent_id) | Q(create_user=self.request.user))
|
|
|
|
|
|
+ queryset = queryset.filter(create_user=self.request.user)
|
|
|
|
+ if self.request.user.general_agent:
|
|
|
|
+ queryset = queryset.filter(id=self.request.user.general_agent.id)
|
|
f = GeneralAgentFilter(self.request.GET, queryset=queryset)
|
|
f = GeneralAgentFilter(self.request.GET, queryset=queryset)
|
|
return f.qs
|
|
return f.qs
|
|
|
|
|
|
@@ -101,10 +102,10 @@ class StoreViewSet(CustomModelViewSet):
|
|
super(StoreViewSet, self).perform_destroy(instance)
|
|
super(StoreViewSet, self).perform_destroy(instance)
|
|
return response_ok()
|
|
return response_ok()
|
|
|
|
|
|
- @permission_required('agent.check_store')
|
|
|
|
@action(methods=['post'], detail=True)
|
|
@action(methods=['post'], detail=True)
|
|
def check(self, request, pk):
|
|
def check(self, request, pk):
|
|
# 审核
|
|
# 审核
|
|
|
|
+ check_permission(request, 'agent.check_store')
|
|
status = request.POST.get('status')
|
|
status = request.POST.get('status')
|
|
try:
|
|
try:
|
|
with transaction.atomic():
|
|
with transaction.atomic():
|
|
@@ -119,7 +120,7 @@ class StoreViewSet(CustomModelViewSet):
|
|
instance.check_time = timezone.now()
|
|
instance.check_time = timezone.now()
|
|
instance.save()
|
|
instance.save()
|
|
BizLog.objects.addnew(request.user, BizLog.INSERT,
|
|
BizLog.objects.addnew(request.user, BizLog.INSERT,
|
|
- u'审核门店[%s]状态为[%s],id=%d' % (instance.name, status, instance.id))
|
|
|
|
|
|
+ u'审核门店[%s]状态为[%s],id=%d' % (instance.name, status, instance.id))
|
|
except CustomError as e:
|
|
except CustomError as e:
|
|
return response_error(e.get_error_msg())
|
|
return response_error(e.get_error_msg())
|
|
except Exception as e:
|
|
except Exception as e:
|