运维帮助

apps/tenant/operation_help/serializer.py

@@ -15,3 +15,13 @@ class OperationHelpSerializer(serializers.ModelSerializer):
     class Meta:
         model = OperationHelp
         fields = '__all__'
+
+    def create(self, validated_data):
+        validated_data['create_user'] = self.context['request'].user
+        validated_data['tenant'] = self.context['request'].user.employee.tenant
+        instance = super(OperationHelpSerializer, self).create(validated_data)
+
+        return instance
+
+
+

apps/tenant/operation_help/views.py

@@ -7,12 +7,10 @@ import json
 from .models import OperationHelp
 from utils.permission import IsAdministratorUser
 from .serializer import OperationHelpSerializer
+from apps.tenant import tenant_log
 from apps.log.models import BizLog
 from .filters import OperationHelpFilter
 from django.db import transaction
-from utils.exceptions import CustomError
-from utils import response_ok, response_error
-
 
 class OperationHelpViewSet(CustomModelViewSet):
     permission_classes = [IsAdministratorUser, ]
@@ -24,36 +22,16 @@ class OperationHelpViewSet(CustomModelViewSet):
         f = OperationHelpFilter(self.request.GET, queryset=queryset)
         return f.qs
 
-    def create(self, request, *args, **kwargs):
-        data = json.loads(request.POST.get('data'))
-        banner_img = request.FILES.get('banner_img')
-        try:
-            data['create_user'] = request.user
-            tenant = request.user.employee.tenant
-            data['tenant'] = tenant
-            operation = OperationHelp.objects.create(**data)
-            operation._add_img(banner_img, request.user)
-            BizLog.objects.addnew(tenant, request.user, BizLog.INSERT, u'添加运维帮助信息', data)
-            return response_ok()
-        except CustomError as e:
-            return response_error(e.get_error_msg())
-        except Exception as e:
-            return response_error(str(e))
-
-    @action(methods=['post'], detail=True)
-    def update_poster(self, request, pk):
-        data = json.loads(request.POST.get('data'))
-        banner_img = request.FILES.get('banner_img')
-        try:
-            if data['enable']:
-                OperationHelp.objects.filter(id=pk).update(**data)
-                poster = self.get_object()
-                if banner_img:
-                    poster.del_images()
-                    poster._add_img(banner_img, request.user)
-        except CustomError as e:
-            return response_error(e.get_error_msg())
-        except Exception as e:
-            return response_error(str(e))
-        return response_ok()
-
+    def perform_create(self, serializer):
+        super(OperationHelpViewSet,self).perform_create(serializer)
+        instance = serializer.instance
+        validated_data = serializer.validated_data
+        tenant_log(self.request.user.employee, BizLog.INSERT, u'添加运维帮助[%s]，id=%d' % (instance.name, instance.id),
+                   validated_data)
+
+    def perform_update(self, serializer):
+        super(OperationHelpViewSet,self).perform_update(serializer)
+        instance = serializer.instance
+        validated_data = serializer.validated_data
+        tenant_log(self.request.user.employee, BizLog.INSERT, u'修改运维帮助[%s]，id=%d' % (instance.name, instance.id),
+                   validated_data)